用masker_rcnn原始权重只检测想要的类别

本文档介绍了如何使用Mask R-CNN的原始权重进行人检测。参考特定教程并解决其中的bug,通过修改几行代码即可实现目标类别(人)的检测。在Ubuntu环境下,下载权重文件并安装pycocotools后,使用jupyter打开并修改samples/demo.py中的类别索引。
摘要由CSDN通过智能技术生成

记录一下怎么用原始权重只检测人 参考的是:https://blog.csdn.net/qq_15969343/article/details/80568579,因为有一些bug,所以自己在记录一下。谢谢原博主的代码!

https://github.com/matterport/Mask_RCNN github地址 只需要修改几行代码就可以只检测人,或者你想要的类别。

下载权重:https://github.com/matterport/Mask_RCNN/releases 这里我没有下载2.1的 而是下载2.0的mask_rcnn_coco.h5

然后要安装pycocotools,https://github.com/waleedka/coco,git后放在任何路径,我是在Ubuntu下,所以直接make就是,然后把coco-master/PythonAPI/pycocotools,下的pycocotools这个文件夹放在Mask_RCNN目录下就好。

然后用jupyter打开samples下的demo.py,只需修改几行就好。只用改Run Object Detection下的就好。如下:


                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值