YOLOv3: Inplement in Tensorflow 2.0
- Please download the VOC2012 dataset and put it into
data
folder.
$ mkdir data
$ wget http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar -O ./data/VOCtrainval_11-May-2012.tar
$ tar xvf ./data/VOCtrainval_11-May-2012.tar --directory ./data
- Split dataset and transfer to
tfrecord
.
# train
$ python3 voc2012.py \
--data_dir ./data/VOCdevkit/VOC2012/ \
--split train \
--output_file ./data/voc2012_train.tfrecord
# val
$ python3 voc2012.py \
--data_dir ./data/VOCdevkit/VOC2012/ \
--split val \
--output_file ./data/voc2012_val.tfrecord
Download pre-trained Darknet weights
$ wget https://pjreddie.com/media/files/yolov3.weights -O model_data/yolov3.weights