Tensorflow implementation of You Only Look Once
An improvement of the implementation from @gliese581gg, with added training, testing and video parsing. We also used the VOC tools to parse the VOC dataset from @mprat.
Note: the code needs cleaning (I didn't write it myself from scratch). I could not find the time and honestly, at this point, there are multiple better Open Source implementations of YOLOv2, so I don't see the point of doing it anymore.
If you want to train the network yourself
python data_parsing/setup.py install
If you want to use pre-trained weights
Download YOLO weight file from: [https://drive.google.com/file/d/0B2JbaJSrWLpza08yS2FSUnV2dlE/view?usp=sharing].
Put the 'YOLO_small.ckpt' in the 'weight' folder of downloaded code.
python setup.py develop --uninstall
python network/YOLO_small_tf.py -fromfile "name of input file" -tofile_img "name of output file"
python network/YOLO_small_tf.py -video "name of input file" -tofile_vid "name of output file"
Refeer to the LICENSE files of both data_parsing and network.