/ssd-torch7

SSD : Single Shot MultiBox Detector for torch7

Primary LanguageLua

SSD : Single Shot MultiBox Detector for Torch7

This is an experimental Torch7 implementation of SSD. This code is not implemented normalization and data augmentation. (I used 1x1 convolution instead of normalization)

Requirements

Torch7, caffe and

luarocks install loadcaffe
luarocks install nninit
luarocks install optnet

Usage

download PascalVOC and caffemodel, and convert data

./DataDownloadPreprocess.sh

To train

th main.lua

If you don't have time to train your model, you can download a pre-trained model

wget https://www.dropbox.com/s/r9b2t8oxab8a3d8/model120000iteration.t7

and detection (you need to change model and image path)

th demo.lua

example of output

test1 test3 test4 test2