yolov7-pose遇到的问题

1.plots.py中的问题，添加这些东西就好了。否则会报错，pos1 = (int(kpts[(sk[0]-1)*steps]), int(kpts[(sk[0]-1)*steps+1]))，IndexError: index 0 is out of bounds for axis 1 with size 0（plot.py）

skeleton = [[1, 2], [2, 3],  [3, 4]]
pose_limb_color = palette[[9, 9, 9, 9, 7, 7, 7, 0, 0, 0, 0, 0, 16, 16, 16, 16, 16, 16, 16]]
pose_kpt_color = palette
radius = 5
num_kpts = len(kpts) // steps
skeleton = skeleton[:(num_kpts+2)]

2. loss中的修改，

anchors, shape = self.anchors[i], p[i].shape

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))

训练过程中将如下设置：

'--kpt-label', default='True',

https://blog.csdn.net/small_wu/article/details/127299912

https://blog.csdn.net/qq_37048420/article/details/133863621?csdn_share_tail=%7B%22type%22%3A%22blog%22%2C%22rType%22%3A%22article%22%2C%22rId%22%3A%22133863621%22%2C%22source%22%3A%22qq_37048420%22%7D


分类|网络|人工智能|_基于yoloV7-pose添加任意关键点 ＋ 多类别分类网络修改
 

YOLOv8-pose关键点检测自制数据集-CSDN博客

Yolo-pose训练和测试，多分类修改，COCO数据集(17关键点)-CSDN博客

YOLOPOSE 1.自定义关键点/修改label脚本/训练数据可视化_yolo训练可视化-CSDN博客

 

dataset一直加载不进入关键点信息，因此我将数据集中的0，1，2显示关键点的信息全部删除了。因此能够得到加载进入labels里面了。

另外，

File "/home/young/yolov7-pose/utils/plots.py", line 136, in plot_skeleton_kpts
    cv2.line(im, pos1, pos2, (int(r), int(g), int(b)), thickness=2)

skeleton = [[0, 1], [1, 2],  [2, 3]]  要从0开始设置，要不然就会出现

pos2 = (int(kpts[(sk[1]-1)*steps]), int(kpts[(sk[1]-1)*steps+1]))编译不出来！

2023.10.21

参考：Yolo-pose训练和测试，多分类修改，COCO数据集(17关键点)-CSDN博客

detect.py文件错误，126行修改为如下所示：

for *xyxy, conf, cls in det:

2023.11.28

训练中总是显示错误

File "/home/young/yolov7-pose/utils/plots.py", line 133, in plot_skeleton_kpts pos2 = (int(kpts[(sk[1])*steps]), int(kpts[(sk[1])*steps+1])) IndexError: index 7 is out of bounds for axis 0 with size 7

在plot.py中找到下面这一句，将steps=2，写上后就好了，不知道为啥啊！还没时间研究！

if kpt_label:   # 这里的step必须填写2不知道为什么？
    plot_one_box(box, mosaic, label=label, color=color, line_thickness=tl, kpt_label=kpt_label, kpts=kpts[:,j], steps=2, orig_shape=orig_shape)

2023.11.29

如果出现bias问题，

将test.py中 half_precision 由true变为true

如果无法画图，则将test.py中的所有关于plot的地方加上相应的关键点信息即可！