richardhahahaha / pytorch-ssd

Single Shot MultiBox Detector in PyTorch

PyTorch-SSD

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