Skip to content

🚖 Object Detection (YOLOv1) implentation in tensorflow, with training, testing and video features.

Notifications You must be signed in to change notification settings

dshahrokhian/YOLO_tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Tensorflow implementation of You Only Look Once

Video

An improvement of the implementation from @gliese581gg, with added training, testing and video parsing. We also used the VOC tools to parse the VOC dataset from @mprat.

Paper: https://arxiv.org/abs/1506.02640

Note: the code needs cleaning (I didn't write it myself from scratch). I could not find the time and honestly, at this point, there are multiple better Open Source implementations of YOLOv2, so I don't see the point of doing it anymore.

Installation

Requirenments

  • Tensorflow
  • OpenCV2

If you want to train the network yourself

cd data_parsing

python data_parsing/setup.py install

If you want to use pre-trained weights

Download YOLO weight file from: [https://drive.google.com/file/d/0B2JbaJSrWLpza08yS2FSUnV2dlE/view?usp=sharing].

Put the 'YOLO_small.ckpt' in the 'weight' folder of downloaded code.

Uninstall (training)

cd data_parsing

python setup.py develop --uninstall

Usage

Images

python network/YOLO_small_tf.py -fromfile "name of input file" -tofile_img "name of output file"

Videos

python network/YOLO_small_tf.py -video "name of input file" -tofile_vid "name of output file"

License

Refeer to the LICENSE files of both data_parsing and network.

About

🚖 Object Detection (YOLOv1) implentation in tensorflow, with training, testing and video features.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages