Skip to content

jokiness/yolo2

 
 

Repository files navigation

You Only Look Once

1.Train YOLOv2 object detector from scratch with Tensorflow.

Usage

Prepare two files:

data.csv (three columns: filenames, rois, classes - each row contains image filepath, list of rois (each [x,y,w,h]), list of classes) and anchors.txt (each row contains width and height of one anchor).

python make_tfrecord.py
python train.py
python eval.py

2.Evaluate YOLOv2 model trained with COCO dataset using Tensorflow. Conversion from Darknet to Tensorflow framework done with darkflow project.

About

Train YOLOv2 object detector from scratch using Tensorflow.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%