Skip to content

基于yolo v2的目标检测(附带训练和测试程序)

Notifications You must be signed in to change notification settings

jason4521/yolo2-tensorflow

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 coco_eval.py

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

About

基于yolo v2的目标检测(附带训练和测试程序)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages