参考博客:https://blog.csdn.net/mrjiale/article/details/82078601
1、需要的模块就不说了(运行没有模块的话就下载该模块)
2、教程原话:https://pjreddie.com/media/files/yolov3.weights
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
python yolo_video.py [OPTIONS...] --image, for image detection mode, OR
python yolo_video.py [video_path] [output_path (optional)]
3、步骤解读:
3.1 下载权重文件 https://pjreddie.com/media/files/yolov3.weights,放在根目录下
3.2 进入控制台执行该语句: python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5 我用的pycharm的terminal直接执行的该语句。
3.3 运行程序:如果要测试图片,在终端输入python yolo_video.py --image(因为在这个py文件中才有main,其他的都没有,所有不是运行yolo.py文件),之后输入要检测的图片路径,建议放在当前的根目录下,这样简单点。
如果要测试视频:先修改文件yolo.py中的detect_video函数,将参数改为视频路径则测试视频,将参数改为0则测试摄像头视频。输入命令 python yolo_video.py 直接运行
同时如果是CPU运行的话,则在yolo.py中修改gpu参数(本人也不知道不改会不会出错,因为我只是小菜鸟一枚,读者可以尝试先不改)。如果对该算法的理解够透彻的话,同时其他的参数也可以更改从而提高性能。