YOLOv7可否同时进行目标检测和姿态估计?

Question

YOLOv7可否同时进行目标检测和姿态估计?比如我要同时检测出汽车这个目标，还有视频中出现的人体的关键点。是分别用两个权重模型还是一个权重模型可以搞定？例如，只用yolov7-w6-pose.pt能否同时检测关键点和其他目标物体？有具体的案例代码吗？谢谢！

Answer 1

yolov7本身就有pose分支啊，可以检测出来人和人体姿势关键点，其他类别的话好像不行，我没仔细研究，不知道其他类别结果是被丢弃了没绘制出来还是本身就不检测。如果是本身不检测的话，那就比较麻烦了，需要两个模型，或者得修改原来的网络结构，键检测头的类别加回来。如果只是代码本身对结果非person类不做处理，那还好做一些，将不做处理的部分加上去就可以达到你想要的目标了。

https://github.com/WongKinYiu/yolov7/tree/pose

我又大概研究了下，在yaml文件里面，只有一个类别，你可以试一下将这个nc改成你的类重新训练一下模型看看，然后后处理的nms中判断一下如果是非person类，跳过后面的keypoint处理应该就可以了