多人姿態估計與跟蹤,復現成功經驗。
1、下載
git clone -b pytorch https://github.com/MVIG-SJTU/AlphaPose.git
2、下載配置
pip install -r requirements.txt
因為我的配置跟他的有些出入,所以沒有安裝文件中的版本,直接修改requirement.txt把版本號刪掉安裝就可以了。
3、參考地址:https://blog.csdn.net/qq_38522972/article/details/82993216
下載兩個模型: duc_se.pth 和yolov3-spp.weights,分別將他們放在./models/sppe and ./models/yolo文件夾里
鏈接:https://pan.baidu.com/s/1JOQlOlMUzJeHFi5f5jmOlg
提取碼:c5ob
4、測試,我直接用的攝像頭
測試攝像頭
python3 webcam_demo.py --webcam 0 --outdir examples/res --vis
其他代碼:
測試圖像: python3 demo.py --indir ${img_directory} --outdir examples/res
測試視頻並保存 python3 video_demo.py --video ${path to video} --outdir examples/res --save_video
在examples文件夾里添加一個res文件夾即可。
5、效果:
就攝像頭效果而言檢測的效果不錯,你做啥怪異的動作都能跟着走,也不會有很強的調頻現象,視覺觀感非常好。
但是唯一的問題就是檢測視頻卡頓,我在服務器上運行都會有大概10秒左右的延遲,可以說我都做完動作回來了它才顯示。
我的電腦有2塊GPU,都出現這個情況。