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:

• Hakuyume/chainer-ssd
• amdegroot/ssd.pytorch