SSD: Single Shot MultiBox Object Detector, in PyTorch
A PyTorch implementation of Single Shot MultiBox Detector from the 2016 paper by Wei Liu, Dragomir Anguelov, Dumitru Erhan, Christian Szegedy, Scott Reed, Cheng-Yang, and Alexander C. Berg. The official and original Caffe code can be found here.
Table of Contents
Installation
Install PyTorch by selecting your environment on the website and running the appropriate command.
Clone this repository.
Note: We currently only support Python 3+.
Then download the dataset by following the instructions below.
We now support Visdom for real-time loss visualization during training!
To use Visdom in the browser:
# First install Python server and client
pip installvisdom
# Start the serve