YOLO_tensorflow-master
所属分类:图形图像处理
开发工具:Python
文件大小:13071KB
下载次数:91
上传日期:2017-08-10 10:57:51
上 传 者:
sunli123
说明: 基于TensorFlow平台下的YOLO算法,可用于目标检测
(YOLO algorithm based on TensorFlow platform, which can be used for target detection)
文件列表:
LICENSE (641, 2017-02-20)
test\person.jpg (113880, 2017-02-20)
weights\put_weight_file_here.txt (0, 2017-02-20)
YOLO_face_tf.py (10222, 2017-02-20)
YOLO_small_tf.py (10628, 2017-02-20)
YOLO_tiny_tf.py (9863, 2017-02-20)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\alexnet.cfg (815, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\darknet.cfg (999, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\extraction.cfg (1786, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\extraction.conv.cfg (1689, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\jnet-conv.cfg (1081, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\msr_152.cfg (14537, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\msr_34.cfg (3606, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\msr_50.cfg (4986, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\strided.cfg (1778, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\vgg-16.cfg (1408, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\vgg-conv.cfg (1114, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\writing.cfg (400, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\yolo-coco.cfg (2342, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\yolo-small.cfg (2334, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\yolo-tiny.cfg (1355, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\cfg\yolo.cfg (2293, 2016-02-09)
YOLO_weight_extractor\YOLO_weight_extractor\darknet (410619, 2016-02-15)
YOLO_weight_extractor\YOLO_weight_extractor\data\1.JPG (1867494, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\2.JPG (188092, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\3.JPG (3035507, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\4.JPG (1929259, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\5.JPG (195900, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\camera_image_follower.jpeg (2594, 2015-12-23)
YOLO_weight_extractor\YOLO_weight_extractor\data\dog.jpg (162407, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\eagle.jpg (141886, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\horses.jpg (133495, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\inet.labels.list (10000, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\labels\aeroplane.png (713, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\labels\airplane.png (629, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\labels\apple.png (537, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\labels\backpack.png (610, 2016-01-04)
YOLO_weight_extractor\YOLO_weight_extractor\data\labels\banana.png (494, 2016-01-04)
... ...
#YOLO_tensorflow
(Version 0.3, Last updated :2017.02.21)
###1.Introduction
This is tensorflow implementation of the YOLO:Real-Time Object Detection
It can only do predictions using pretrained YOLO_small & YOLO_tiny network for now.
(+ YOLO_face detector from https://github.com/quanhua92/darknet )
I extracted weight values from darknet's (.weight) files.
My code does not support training. Use darknet for training.
Original code(C implementation) & paper : http://pjreddie.com/darknet/yolo/
###2.Install
(1) Download code
(2) Download YOLO weight file from
YOLO_small : https://drive.google.com/file/d/0B2JbaJSrWLpza08yS2FSUnV2dlE/view?usp=sharing
YOLO_tiny : https://drive.google.com/file/d/0B2JbaJSrWLpza0FtQlc3ejhMTTA/view?usp=sharing
YOLO_face : https://drive.google.com/file/d/0B2JbaJSrWLpzMzR5eURGN2dMTk0/view?usp=sharing
(3) Put the 'YOLO_(version).ckpt' in the 'weight' folder of downloaded code
###3.Usage
(1) direct usage with default settings (display on console, show output image, no output file writing)
python YOLO_(small or tiny)_tf.py -fromfile (input image filename)
(2) direct usage with custom settings
python YOLO_(small or tiny)_tf.py argvs
where argvs are
-fromfile (input image filename) : input image file
-disp_console (0 or 1) : whether display results on terminal or not
-imshow (0 or 1) : whether display result image or not
-tofile_img (output image filename) : output image file
-tofile_txt (output txt filename) : output text file (contains class, x, y, w, h, probability)
(3) import on other scripts
import YOLO_(small or tiny)_tf
yolo = YOLO_(small or tiny)_tf.YOLO_TF()
yolo.disp_console = (True or False, default = True)
yolo.imshow = (True or False, default = True)
yolo.tofile_img = (output image filename)
yolo.tofile_txt = (output txt filename)
yolo.filewrite_img = (True or False, default = False)
yolo.filewrite_txt = (True of False, default = False)
yolo.detect_from_file(filename)
yolo.detect_from_cvmat(cvmat)
###4.Requirements
- Tensorflow
- Opencv2
###5.Copyright
According to the LICENSE file of the original code,
- Me and original author hold no liability for any damages
- Do not use this on commercial!
###6.Changelog
2016/02/15 : First upload!
2016/02/16 : Added YOLO_tiny, Fixed bug that ignores one of the boxes in grid when both boxes detected valid objects
2016/08/26 : Uploaded weight file converter! (darknet weight -> tensorflow ckpt)
2017/02/21 : Added YOLO_face (Thanks https://github.com/quanhua92/darknet)
近期下载者:
相关文件:
收藏者: