14ashray / ssd_model

SSD model for hand detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ssd_model

The programs in this repository train and use a Single Shot MultiBox Detector to take an image and draw bounding boxes around hands contained in the image. The network is based on the VGG-16 model and uses the approach described in this https://arxiv.org/abs/1512.02325 paper by Wei Liu et al.

Hand DataSets can be downloaded from here -
1)http://www.robots.ox.ac.uk/~vgg/data/hands/
2)http://vision.soic.indiana.edu/projects/egohands/

You then need to preprocess the dataset before you can train the model on it.
./process_dataset.py

You can then train the whole thing. It will take around 150 to 200 epochs to get good results.
./train.py

If you want to make detection basing on the inference model, check out:
./detect.py

About

SSD model for hand detection


Languages

Language:Python 100.0%