原论文代码地址：https://github.com/IsshikiHugh/HSMR

原论文地址：https://isshikihugh.github.io/HSMR/

环境

1. wsl2 ubuntu24
2. python3.10
3. cuda 12.9（显卡5070Ti）
4. venv

步骤

1. 克隆

   git clone https://github.com/IsshikiHugh/HSMR --recursive
   

2. 创建虚拟环境并激活

3. 安装conda12.9

   pip3 install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu129
   

4. 进入克隆下来的文件夹

5. 安装所需要的包

   pip install -r requirements.txt
   

6. 更新构建工具

   pip install --upgrade setuptools wheel
   

7. 根据这个文档从源码安装detection2(关键步骤)

   注意：

   1. 因为使用venv，所以需要隔离，在末尾加--no-build-isolation
   2. torch版本要和detectron2一致，下面的版本是支持torch = 2.8.0+cu129
   3. 版本具体看这里

   pip install "git+https://github.com/facebookresearch/detectron2.git@a1ce2f956a1d2212ad672e3c47d53405c2fe4312" --no-build-isolation
   

8. pip install "git+https://github.com/mattloper/chumpy" --no-build-isolation
   

9. pip install -e .
   

10. pip install -e thirdparty/SKEL --no-deps
    

11. 进入data_inputs文件夹

    cd data_inputs
    

12. 创建released_models文件夹

    mkdir -p released_models
    

13. 下载资源

    wget -c 'https://huggingface.co/IsshikiHugh/HSMR-data_inputs/resolve/main/body_models.tar.gz' -O body_models.tar.gz
    

14. wget -c 'https://huggingface.co/IsshikiHugh/HSMR-data_inputs/resolve/main/released_models/HSMR-ViTH-r1d1.tar.gz' -O released_models/HSMR-ViTH-r1d1.tar.gz
    

15. 解压

    tar -xzf body_models.tar.gz
    

16. 解压

    tar -xzf released_models/HSMR-ViTH-r1d1.tar.gz -C released_models/
    

17. 回到HSMR文件夹

18. 图片测试

    python exp/run_demo.py --input_path "data_inputs/demo/example_imgs"
    

19. 视频测试

    python exp/run_demo.py --input_path "data_inputs/demo/example_video/vide_name.mp4"