1、下载YOLOv3权重:https://pjreddie.com/media/files/yolov3.weights
2、下载YOLOv3代码:https://github.com/qqwweee/keras-yolo3
3、转换权重.weights文件为.h5文件:
将.weights文件放到代码文件夹中的convert.py文件的同级目录,运行
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
转换后的.h5模型在model_data文件夹下。
4、找张图片来test,运行代码:
python yolo_video.py --image #测试图片 其余可以用pthon yolo_video.py -h显示帮助
按照提示输入image_name即可。
出现错误:
RuntimeWarning: compiletime version 3.5 of module 'tensorflow.python.framework.fast_tensor_util' does not match runtime version 3.6
解决方法:
在https://github.com/lakshayg/tensorflow-build下载对应的tensorflow,使用以下命令安装:
pip install --ignore-installed --upgrade tensorflow-1.4.0-cp36-cp36m-macosx_10_12_x86_64.whl
后期打算复现为project,不在命令行中运行,批量测试图片。