先贴上yolo3源码链接:
https://github.com/qqwweee/keras-yolo3
里面虽然有使用教程,但是过于简洁。
在测试时需要用到以下两个py文件:
yolo.py (需要修改模型文件的路径和类别标签等)
yolo_video.py
测试图片
在当前文件夹下运行:
python yolo_video.py --image
运行之后会提示输入你将要测试的图片名称,例如1.jpg 回车即可。
测试一段视频
为了方便查看,可以先创建两个文件夹,一个存储待测视频(例如input),另外一个存储检测之后的视频(例如output)。
在项目文件下运行:
python yolo_video.py --input=input/test.mp4 --output=output/test.mp4
即可。