【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试

前言 0

作为yolo的入门级选手,在前面的两篇文章中,我介绍了最新发布的yolov3的使用情况。追本溯源,在这里想深入的学习一下,所以,在这,将yolo_v1也同样学习一下。 

所需材料

清单:

python3.6

tensorflow-gpu  1.4.0,

cuda_8.0.61_win10

cudnn-8.0-windows10-x64-v6.0

TensorFlow 版本的源码 :https://github.com/hizhangp/yolo_tensorflow

YOLO_small权重文件:https://pan.baidu.com/s/1SShEp8C0rTSbWNNKSgRKLg (提取码:gph6) 

开始运行

step1:将下载的TensorFlow 版本的源码解压,在该目录下新建一个data\weights目录,将下载的YOLO_small权重文件拷贝到此,如下:

【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第1张图片

step2:打开test.py,在 line206处,输入自己的测试图片,

【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第2张图片

运行即可。效果如下:

 

【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第3张图片 原图1
【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第4张图片 效果图1

 

【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第5张图片 原图2
【YOLO初探】 yolo_tensorflow_v1使用YOLO_small.ckpt进行测试_第6张图片 效果图2

 

能看的出来,这个版本的效果,比v3还是差了很多的!


★finishe by songpl,2019.1.12

你可能感兴趣的:(YOLOV1)