最近在学习yolo v3,下载了yolo的keras版本,按照工程里readme中将权重下载,运行Convert后,可以进行yolo的图像检测,
图像检测命令:
python yolo_video.py --image,然后在提示下输入图片路径即可实现本地图片检测。
本地视频检测:
python yolo_video.py --input xxx.mp4即可实时检测本地视频。
然后,想用笔记本的自带摄像头进行实时检测,在网上搜了好久没有搜到,只看到了python读取摄像头的命令,然后经过尝试,终于实现了自带摄像头进行实时检测,其实,修改很简单,只需要yolo.py的下面vid改下,改成vid = cv2.VideoCapture(0),如下图,然后就可以使用了,使用时,命令是python yolo_video.py。(会运行detect_video部分),在笔记本上测试非tiny版本实时检测速率FPS为1.
有的朋友不知道视频打开后怎么关闭,其实看下代码就知道,yolo.py的倒数2、3行,如下图键盘的按键q会打断检测,因此记得是按q键,不是点界面的X:
最后,再介绍下保存检测的视频到本地的方法:
一种是运行程序时加参数 --output 文件名
另一种直接修改yolo.py里的代码,将def detect_video(yolo, video_path, output_path=""):下的两处output处的代码修改:
改为:
第二处是:
改为:
亲测了下,保存本地视频效果没问题,但是保存笔记本摄像头视频时,保存的视频打开后格式有问题,这个后续再研究。
网友评论