yolov8 obb cuda后处理是否错误

[e1e1t7t7]

在postprocess.cu中的decode_kernel_obb(float* predict, int num_bboxes, float confidence_threshold, float* parray, int max_objects) 中如下代码：

float* pout_item = parray + 1 + index * bbox_element;
*pout_item++ = cx;                                        //0
*pout_item++ = cy;                                        //1
*pout_item++ = width;                                     //2
*pout_item++ = height;                                    //3
*pout_item++ = confidence;                                //4
*pout_item++ = label;                                     //5
*pout_item++ = 1;  // 1 = keep, 0 = ignore                6
*pout_item++ = angle;                                    //7

bbox_element在types.h中，其值为7，但obb的输出element应该是8吧（left, top, right, bottom, confidence, class, keepflag, angle）,上述代码会导致decode的结果中每个结果的angle被覆盖

[wang-xinyu]

@lindsayshuo 看下呢

[LorenzoSun-V]

我也遇到了类似的问题，yolov8 obb cuda后处理结果是不对的

[LorenzoSun-V]

但是这个keepflag实际不是模型输出的结果，而是在做NMS时给出的结果，符合阈值且满足NMS筛选过后的框，keepflag=1.

[stale]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.