pytorch-ssd icon indicating copy to clipboard operation
pytorch-ssd copied to clipboard

Published 20 hours ago • verified publisher icon dusty-nv
→

Metadata

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

Results 3 pytorch-ssd issues
Sort by recently updated
recently updated
newest added

Added Nesterov momentum to SGD optimizer

1 comment

georgekasa avatar georgekasa

Fix size mismatch if the prediction is empty

Currently, if the model output empty predictions like `(tensor([]), tensor([]), tensor([]))`, the evaluation script would output errors like ```bash Traceback (most recent call last): File "eval_ssd.py", line 243, in mean_ap,...

ZhiliangWu avatar ZhiliangWu

Update train_ssd.py to support multiple GPUs

6 comment

Hello @dusty-nv, According to your suggestion here [Support multiple GPU](https://github.com/dusty-nv/jetson-inference/issues/1171#issuecomment-899585136) and [the issue referenced here](https://github.com/qfgaohao/pytorch-ssd/issues/161) @Mystique-orca and I have enabled multiple GPUs to support training SSD-based Object Detection Model using...

NISHANTSHRIVASTAV avatar NISHANTSHRIVASTAV

← Metadata

49
Stars
68
Forks
Watchers

Owner

verified publisher icon dusty-nv

Metadata

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

Back