/tensorflow-yolo

tensorflow implementation of 'YOLO : Real-Time Object Detection'(train and test)

Primary LanguagePython

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

  1. 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

  2. Extract all of these tars into one directory named VOCdevkit

    tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar

  3. It should have this basic structure

    $VOCdevkit/                           # development kit
$VOCdevkit/VOCcode/                   # VOC utility code
$VOCdevkit/VOC2007                    # image sets, annotations, etc.
# ... and several other directories ...

  4. 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

  1. transform your training data to text_record file(the format reference to pascal_voc)

  2. write your own train-configure file

  3. train (python tools/train.py -c $your_configure_file)

test demo

python demo.py