动作识别-mmskeleton使用
一、源码链接:https://github.com/open-mmlab/mmskeleton
二、由于该项目环境涉及openpose等,环境配置耗费大量时间,于是乎想到有没有别人配置好的环境,在docker hub搜索,有一个,非常开心,先把代码跑起来再说,拉取镜像
docker pull ioir123ju/mmskeleton
以上是别人提供的镜像
一下是自己在上面镜像的基础上配置的自己的镜像环境,GPU为RTX2080Ti
docker pull liuwenhua/mmskeleton
三、详细记录了在docker中配置环境遇到的问题
1.在安装第三方包lazy_import 报错,无法进行安装,想到用源码安装,源码安装还在报错,编码方式错误,更改setup.py 打开文件的编码方式为utf-8就可以了,
2.测试代码
python mmskl.py pose_demo --gpus 1
报错如下:
For friends who may run into the same issue "ImportError: mmskeleton.utils.third_party attempted to use a functionality that requires module mmdet.apis, but it couldn't be loaded. Please install mmdet and retry."
各种找原因,最终在参来链接 https://github.com/open-mmlab/mmskeleton/issues/326
解决方案
pip uninstall mmcv, then $ pip install mmcv==0.4.3 fixed this issue. 解决了问题,好心累
于是乎各种测试,都可以正常运行
1.数据准备代码
mmskl configs/utils/build_dataset_example.yaml --gpus 1
2.训练阶段
mmskl configs/recognition/st_gcn/dataset_example/train.yaml
3.测试阶段
mmskl configs/recognition/st_gcn/dataset_example/test.yaml --checkpoint /mmskeleton/work_dir/recognition/st_gcn/dataset_example/latest.pt
四,准备自己的数据集,开始训练