记录一下怎么用原始权重只检测人 参考的是: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下的就好。如下: