PlantWatcher-SSD

This is an implementation of Single Shot MultiBox Detector in PyTorch trained on plant images, done as a part of an academic project, Design Practicum. Below poster represents a gist of the motive behind project.

Use pretrained VGG16 model

I do not recommend training SSD from scratch. Using 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