search

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
BSD 3-Clause "New" or "Revised" License
875 stars 287 forks source link

Add SSD to benchmark model repo #151

Open bertmaher opened 4 years ago

bertmaher commented 4 years ago

This is in pytorch/hub: https://pytorch.org/hub/nvidia_deeplearningexamples_ssd/.

There's some evidence that this can be made much faster (up to 3x) with some graph optimizations: https://paulbridger.com/posts/video-analytics-deepstream-pipeline/

desertfire commented 3 years ago

https://pytorch.org/hub/nvidia_deeplearningexamples_ssd/ is highly coupled with Nvidia's DeepStream and TensorRT. I decide to pick a more general implementation from https://github.com/sgrvinod/a-PyTorch-Tutorial-to-Object-Detection.git .