Skip to content

kaka-lin/yolov2-tf2

Repository files navigation

YOLOv2: Inplement in Tensorflow 2

Step-by-Step

yolov2.ipynb

  1. Jupyter Notebook

    $ jupyter notebook
  2. Colab

    Open in Colab

Prepare the Dataset for training

  1. Please download the kangaroo dataset and put it into data folder. I already write a script for doing this, as flow:

    $ ./downloda_data.sh
    
  2. parse voc and transfer to tfrecord.

    $ python3 voc_to_tfrecord.py

Download pre-trained Darknet weights

$ wget https://pjreddie.com/media/files/yolov2.weights -O model_data/yolov2.weights

Acknowledgments

Releases

No releases published

Packages

No packages published

Languages