Skip to content

JY-112553/yolov1-keras-voc

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

yolov1-keras-voc

Keras implementation of YOLOv1 (Tensorflow backend) inspired by allanzelener/YAD2K and qqwweee/keras-yolo3.

Download Yolo weights

Download tiny-yolov1 weights from here

wget http://pjreddie.com/media/files/yolov1/tiny-yolov1.weights

Convert the model

Convert the Darknet model to a Keras model

python convert.py cfg/yolov1-tiny.cfg tiny-yolov1.weights tiny-yolov1.hdf5

Train your datasets

I use pascal-v0c2012 to train, you can download the datasets from here

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar
tar xvf VOCtrainval_11-May-2012.tar

use data/build datasets.py to build datasets

python data/build datasets.py

use train.py to train by your datasets

About

Keras implementation of YOLOv1 (Tensorflow backend)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages