安裝 Simple-HRNet

git clone https://github.com/stefanopini/simple-HRNet.git

cd simple-HRNet

conda env remove -n hrnet

conda create -n hrnet python=3.9

conda activate hrnet

pip install -r requirements.txt

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113

# demo

python scripts/live-demo.py --camera_id 0 --single_person --device cuda

python scripts/live-demo.py --filename video.mp4 --single_person --disable_vidgear --save_video