mmskeleton基于人体关键点的实时跟踪与识别代码复现

1. 环境搭建

conda create -n mmskeleton_zhw
conda activate mmskeleton_zhw
cd /mnt/hdd2/home_old/home/lcm/Code/mygit/mmskeleton/
pip3 install -r requirements.txt
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

*cuda版本为11.4，没有11.4对应的pytorch，故安装cu11.3

问题一：conda命令不存在

解决：

vim ~/.bashrc
#按e进入内容，按a编辑，在最后一行加入下一句
export PATH=~/home/anaconda3/bin:$PATH
#按ESC退出编辑，shift + :wq保存并退出
source ~/.bashrc

问题二：conda搭建完环境activate时提示错误

CommandNotFoundError: Your shell has not been properly configured to use 'conda activate'.

解决：当root前出现(base)表示可以激活conda虚拟环境

conda init bash

问题三：根据requirement下载torch库太慢

解决：寻找对应cuda版本的torch下载指令https://pytorch.org/get-started/previous-versions/

问题四：下载opencv时太慢

解决：采用清华源

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

2. 运行代码

以./mycode/multi_fusion.py为例

python setup.py install
sh setup.sh

cd mycode #一定需要进入该路径
python multi_fusion.py

需要显示图像则需要调用./mycode/render/myrender.