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