tensorflow1.4.0+keras2.0.5 yolov3复现

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,不在命令行中运行,批量测试图片。

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值