yolo_v2入门实现目标检测

python开发环境,安装了tensorflow-gpukeras和其他常用机器学习库。该过程使用yolo_v2进行测试。

获取YAD2K源码

yolo有专有的深度学习网络darknet,YAD2K是python使用yolo_v2网络的一种接口方法。
克隆或下载https://github.com/allanzelener/YAD2K源码。

获取模型、权重文件

https://pjreddie.com/darknet/yolo/下载对应的cfg文件和weights文件。如下图(后两个是链接):
这里写图片描述

模型转换

在YAD2K源码文件夹中执行如下命令:

python .\yad2k.py .\my_config\yolov2.cfg .\my_config\yolov2.weights .\model_data\yolo.h5

其中my_config为步骤2的两个文件所在文件夹。该操作将 Darknet YOLO_v2 model转换为了Keras model,在model_data文件夹生成了两个文件:
这里写图片描述

执行测试

python .\test_yolo.py .\model_data\yolo.h5

该操作使用images文件夹中的图片,生成图片到images/out文件夹。

这里写图片描述
这里写图片描述

该过程精简修正代码(不是我的哈哈,0.0)在GitHub上。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值