Skip to content

Train YOLOv2 object detector from scratch using Tensorflow.

Notifications You must be signed in to change notification settings

wojciechmo/yolo2

Repository files navigation

You Only Look Once

  1. 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
  1. Train YOLOv2 object detector from scratch with Tensorflow.
python train.py
  1. Evaluate YOLOv2 model trained with COCO dataset using Tensorflow. Conversion from Darknet to Tensorflow framework done with darkflow project.
python eval.py

About

Train YOLOv2 object detector from scratch using Tensorflow.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages