Skip to content
forked from wojciechmo/yolo2

Train YOLOv2 object detector from scratch using Tensorflow.

Notifications You must be signed in to change notification settings

kybelvody/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%