Repository: https://github.com/sgrvinod/a-PyTorch-Tutorial-to-Object-Detection
You can run object detection in video using webcam
SSD: Single Shot MultiBox Detector ECCV 2016
Wei Liu 1, Dragomir Anguelov 2, Dumitru Erhan 3, Christian Szegedy 3, Scott Reed 4,
Cheng-Yang Fu 1, Alexander C. Berg 1
1 UNC Chapel Hill, 2 Zoox Inc., 3 Google Inc., 4 University of Michigan, Ann-Arbor
- [Python 3.6]
- [PyTorch 1.1.0]
$ git clone https://github.com/jinyoungHan/pytorch-ssd.git
$ cd pytorch-ssd/
VOC2012 test
VOC2007 train
VOC2007 test
Extract both the VOC 2007 trainval and 2007 test data to the same location
data/VOC2007/train
data/VOC2007/test
After downloading datasets
$ python data2json.py
On a GTX 1060, each epoch of training required about 30 minutes.
$ python train.py
i: image , image path or no image path(default path)
w: webcam, no image path
$ python detect.py <i or w> <image path>