Tensorflow implementation of YOLO
- python 3.6
- tensorlayer 1.10.0
- tensorflow 1.9.0
- matplotlib 2.2.3
- itchat 1.3.10
- numpy 1.15.0
- tqdm 4.25.0
-
Clone yolo_tensorflow repository
$ git clone https://github.com/TooDifficultForMe/tensorflow-YOLO.git $ cd tensorflow-YOLO
-
Download Pascal VOC dataset, and create correct directories
$ ./download_data.sh
-
Modify configuration in
config.py
-
Training
$ python train.py
-
Move trained weight file in
data/weight
-
Test
open
test.ipynb
with Jupyter Notebook