window python tensorflow keras yolo3 测试

研究了好几天的yolo,尝试的跑了一下window版的yolo v3,特此写下此篇以防忘记,也供大家参考学习下:
本人电脑配置是:

window10,GTX850M,python35,pycharm,cuda9.0,cudann9.0

安装的包主要有:
这里写图片描述

这里写图片描述
至于keras-yolo3源码可以到github查找下载。
这里写图片描述
(里面的图片是自己从darknet官网下载来的darknet源码包内的测试图片,yolov3_weights文件是也是官网下载的。)
这里写图片描述
(data目录下有官网测试图片。)

下面开始测试:
最主要的是keras-yolo3源码包下的yolo.py和yolo_video.py ,刚开始我也没了解github主qqwweee的keras-yolo3怎么进行测试,当然他已经有介绍了用法,但是难免有一些简略(特别是我这种菜鸟【坏笑】)
这里写图片描述
第一步如果你已经下载好yolov3_weights文件并且放在keras-yolo3源码包下你就不用了;
主要第二步,是把yolov3.cfg yolov3.weights转换成yolo.h5文件放在model_data/下。

开始测试:
有人说直接运行yolo.py就可以了,反正我的不行:
这里写图片描述

要运行的是yolo_video.py,当然直接运行也是不是很好的办法:

这里写图片描述

我们在运行yolo_video.py需要传入参数(他们都已经封装好很多东西,你需要运行程序时候传入参数,正如官方所言,其实我是看不懂啦,后来我认真看了代码才知道):
这里写图片描述

操作:
这里写图片描述

如果你想用摄像头来测试呢,怎么办?
如果像图片测试那样的话,python yolo_video.py --input 就可以了,但事实是运行后会提示,找不到视频文件:
这里写图片描述
Couldn't open webcam or video
(不过你可以尝试拿个视频放在该目录下,我也看不懂官网的添加要读取文件的路径,网上参阅也很少)

解决的办法是:既然没找到摄像头的话,看yolo.py的代码,可以发现有那么一段函数定义的video,所以改代码:
这里写图片描述
然后可以直接运行yolo_video.py就可以摄像头测试yolo3了。

最后能行了是最很开心的啦。至于yolo3的解读,网上有很多就不啰嗦了。

创作不易,如需转载,请标明作者和出处。

猜你喜欢

转载自blog.csdn.net/mrjiale/article/details/82078601
今日推荐