IndexError: index 69 is out of bounds for axis 1 with size 52 yolov3数据加载报错

IndexError: index 69 is out of bounds for axis 1 with size 52

生成0矩阵的代码:

 labels[feature_size] = np.zeros(shape=(feature_size, feature_size, 3, 5 + cfg.CLASS_NUM))

中心点计算:

                cx_offset, cx_index = math.modf(cx * feature_size / cfg.IMG_WIDTH)
                cy_offset, cy_index = math.modf(cy * feature_size / cfg.IMG_WIDTH)

写入标签:

    labels[feature_size][int(cy_index), int(cx_index), i] = np.array(
                        [iou, cx_offset, cy_offset, np.log(p_w), np.log(p_h), *one_hot(cfg.CLASS_NUM, int(cls))])

在计算中心点整数部分的索引,写入标签时cx_index超出了0矩阵大小featuresize*featuresize。

你可能感兴趣的:(深度学习,python,pytorch)