Código: https://github.com/WildflowerSchools/AlphaPose
O ambiente aqui é diferente do yolov3, preste atenção.
1. Meio Ambiente
cuda101
torch==1.6.0
torchvision==0.3.0
opencv-python==4.1.0.25
tqdm
natsort
Cython
pycocotools
python setup.py build develop --user
apt-get update
apt-get -y install python3.6-tk
Para python setup.py build development --user, o processo de instalação da biblioteca Python é lento, use:
pip --default-time=1000 install -i https://pypi.tuna.tsinghua.edu.cn/simple ***
2. Preparação de modelo e preparação de dados
Referência: https://blog.csdn.net/qq_35975447/article/details/114654804
A diferença é que o download é yolov4.weigts . Ou aqui: https://github.com/AlexeyAB/darknet
3.demo - rastreamento
python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_2x-dcn.yaml --checkpoint pretrained_models/fast_dcn_res50_256x192.pth --indir examples/demo/ --vis --showbox --save_img --pose_track --sp
Trajetória de rastreamento pode se referir à trajetória da pintura chinesa deep-sort-yolov3. Ou consulte https://xiaorun.blog.csdn.net/article/details/80572523
Para rastrear qual ponto de junta, adicione as coordenadas do ponto de junta correspondentes ao deque.
keypoints中的0-17分别代表
0 noise
1 left_eye
2 right_eye
3 left_ear
4 right_ear
5 left_shoulder
6 right_shoulder
7 left_elbow
8 right_elbow
9 left_wrist
10 right_wrist
11 left_hip
12 right_hip
13 left_knee
14 right_knee
15 left_ankle
16 right_ankle
Consulte aqui: https://zhuanlan.zhihu.com/p/121452714