其中传入的pred_bboxes格式为3维的数组的list格式,也就是说每个list都是一个3维数组(有batch的考量),为一个样本的所有bbox。
其他的同理如pred_labels 同理。
list化可以参考下面代码
pred_bboxes, pred_labels, pred_scores = list(), list(), list()
gt_bboxes, gt_labels, gt_difficults = list(), list(), list()
bbox1=np.expand_dims(bbox, axis=0)
label1=np.expand_dims(label, axis=0)
labels1=np.expand_dims(labels, axis=0)
bounding1=np.expand_dims(bounding, axis=0)
confidence1=np.expand_dims(confidence, axis=0)
gt_bboxes += list(bbox1)
gt_labels += list(labels1)
# gt_difficults += list(gt_difficults_.numpy())
pred_bboxes += list(bounding1)
pred_labels += list(label1)
pred_scores += list(confidence1)
下面就是本文所要使用的,其中bbox_iou函数参考:https://blog.csdn.net/a362682954/article/details/82896242