PyTorch-SSD has been deprecated. Please see torchcv, which includes an implementation of SSD300/SSD512.
Single Shot MultiBox Detector in PyTorch.
Test
Use pretrained VGG16 model
I do not recommend training SSD from scratch. Use pretrained VGG model helps a lot to achieve lower losses.
I use the pretrained pytorch/vision VGG16 model from PyTorch model zoo.
Credit
This implementation is initially inspired by: