/yolov3-tf2

Implemented YOLOv3 with Tensorflow 2.0

Primary LanguageJupyter NotebookMIT LicenseMIT

YOLOv3: Inplement in Tensorflow 2.0

Download Dataset

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

Step-by-Step

  • yolov3.ipynb

Acknowledgments