'YOLO:实时对象检测'(train and test)的Tensorflow实现
'YOLO:实时对象检测'(train and test)的Tensorflow实现
Python 机器学习
共795Star
详细介绍
tensorflow-yolo
Require
tensorflow-1.0
download pretrained model
yolo_tiny: https://drive.google.com/file/d/0B-yiAeTLLamRekxqVE01Yi1RRlk/view?usp=sharing
mv yolo_tiny.ckpt models/pretrain/
Train
Train on pascal-voc2007 data
Download pascal-Voc2007 data
-
Download the training, validation and test data
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
-
Extract all of these tars into one directory named
VOCdevkit
tar xvf VOCtrainval_06-Nov-2007.tar tar xvf VOCtest_06-Nov-2007.tar
-
It should have this basic structure
$VOCdevkit/ # development kit $VOCdevkit/VOCcode/ # VOC utility code $VOCdevkit/VOC2007 # image sets, annotations, etc. # ... and several other directories ...
-
Create symlinks for the PASCAL VOC dataset
cd $YOLO_ROOT/data ln -s $VOCdevkit VOCdevkit2007
Using symlinks is a good idea because you will likely want to share the same PASCAL dataset installation between multiple projects.
convert the Pascal-voc data to text_record file
python tools/preprocess_pascal_voc.py
train
python tools/train.py -c conf/train.cfg
Train your customer data
-
transform your training data to text_record file(the format reference to pascal_voc)
-
write your own train-configure file
-
train (python tools/train.py -c $your_configure_file)
test demo
python demo.py
-
564 Star
-
2888 Star
-
812 Star
-
3948 Star
-
363 Star
-
5657 Star
-
158 Star