Mask RCNN是目标分割检测框架--扩展到人体关键点检测
对于原理不清晰的同学,建议你去看一下Kaming He的论文:https://arxiv.org/pdf/1703.06870.pdf
我的博客里也有论文的翻译版:Mask R-CNN 论文翻译
对于视频中的多人进行姿态估计,预训练权重,可以直接下载:链接: mask_rcnn_humanpose.h5 密码: qx8f
我已将Mask rcnn扩展到多人姿态估计的代码上传到我的github上了:Keypoints-of-humanpose-with-Mask-R-CNN
如果觉得对你项目有用的话,点个star吧!
1、安装环境:
1、电脑环境:
Mask R-CNN是基于Python3,Keras,TensorFlow。
- Python 3.4+
- TensorFlow 1.3+
- Keras 2.0.8+
- Jupyter Notebook
- Numpy, skimage, scipy, Pillow, cython, h5py
- opencv 2.0
2、预训练权重下载:链接: mask_rcnn_humanpose.h5 密码: qx8f
3、如果需要在COCO数据集上训练或测试,需要安装pycocotools
, clone
下来,make
生成对应的文件,make之后将生成的pycocotools文件夹复制到项目文件夹中:
Linux: