Tensorflow implementation of YOLO, including training and test phase.
-
Clone yolo_tensorflow repository
$ git clone https://github.com/hizhangp/yolo_tensorflow.git $ cd yolo_tensorflow -
Download Pascal VOC dataset, and create correct directories
$ ./download_data.sh
-
Download YOLO_small weight file and put it in
data/weight -
Modify configuration in
yolo/config.py -
Training
$ python train.py
-
Test
$ python test.py
-
Tensorflow
-
OpenCV