动作识别-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

四,准备自己的数据集,开始训练

原文地址:https://www.cnblogs.com/liuwenhua/p/13739810.html