YOLO_V1_Tensorflow

YOLO_V1 Tensorflow version

The folder includes train, test phase, and detailed comments in yolo_v1.py.

Installation

1. Clone yolo_v1_tensorflow repository

   $ git clone https://github.com/zjcanjux/yolo_v1_tensorflow.git
   $ cd yolo_v1_tensorflow

2. Download Pascal VOC dataset, and create correct directories

   $ ./download_data.sh

3. Download YOLO_small weight file and put it in data/weight

4. Modify configuration in yolo/config.py

5. Training

   $ python yolo_v1train.py

6. Test

   $ python yolo_v1test.py

Requirements

1. Tensorflow

2. OpenCV