pushkar/yolo_tensorflow

Tensorflow implementation of YOLO, including training and test phase

Python

YOLO_tensorflow

Tensorflow implementation of YOLO, including training and test phase.

Installation

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
```
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
```

Requirements

Tensorflow
OpenCV