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, we call the directory
PASCAL_VOC
mkdir data cd data wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar tar -xvzf VOCtrainval_06-Nov-2007.tar cd .. ln -s $PASCAL_VOC$ data/pascal_voc
-
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