Skip to content

Tensorflow implementation of YOLO, including training and test phase.

License

Notifications You must be signed in to change notification settings

marcbelmont/yolo_tensorflow

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLO_tensorflow

Tensorflow implementation of YOLO, including training and test phase.

Installation

  1. Clone yolo_tensorflow repository

    $ git clone https://github.com/hizhangp/yolo_tensorflow.git
    $ cd yolo_tensorflow
  2. Download Pascal VOC dataset, and create correct directories

    $ ./download_data.sh
  3. Download YOLO_small weight file and put it in data/weight

  4. Modify configuration in yolo/config.py

  5. Training

    $ python train.py
  6. Test

    $ python test.py

Requirements

  1. Tensorflow

  2. OpenCV

About

Tensorflow implementation of YOLO, including training and test phase.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.1%
  • Shell 0.9%