YOLOv7-Pose运行记录

推理

https://github.com/WongKinYiu/yolov7/tree/pose
YOLOv7-Pose分支少两个函数，可以在YOLOv7对应文件中找到
utils/general non_max_suppression_kpt
utils/plots 中的output_to_keypoint

训练

train：
python train.py --data ./data/coco_kpts.yaml --cfg cfg/yolov7-w6-pose.yaml --weights ‘./weights/yolov7-w6-person.pt’ --batch-size 4 --kpt-label

测试

python test.py --data data/coco_kpts.yaml --img 960 --conf 0.001 --iou 0.65 --weights yolov7-w6-pose.pt --kpt-label --batch-size 8