tensorflow + python + keras-yolov3目标检测

keras-yolo3源码地址:

https://github.com/qqwweee/keras-yolo3

要跑通这个项目,首先,在官网下载yolo模型参数(https://pjreddie.com/darknet/yolo/),然后把yolov3.cfg yolov3.weights转换成yolo.h5文件放在model_data/下,然后运行convert.py 第一个参数为yolov3.cfg yolov3.weights模型参数文件,第二个参数为转换后的模型文件的保存路径model_data/yolo.h5

转换后即可运行yolo_video.py

参数:

--image  对图片进行目标检测

--input [视频地址]  对视频进行目标检测 视频格式为mp4或者avi

图片运行效果如下:

猜你喜欢

转载自blog.csdn.net/Polaris47/article/details/84035921