Unable to Determine the Confidence Values from Output Results of MobileNetV2 Backbone with ATSS Head Model
Content Type: Troubleshooting | Article ID: 000098366 | Last Reviewed: 04/13/2024
for i, detection in enumerate(detections):
class_id = labels[i]
xmin, ymin, xmax, ymax, confidence = detection
if confidence > 0.5:
label = int(labels[class_id])
xmin = int(xmin * w)
ymin = int(ymin * h)
xmax = int(xmax * w)
ymax = int(ymax * h)
log.info(f'Found: label = (label), confidence = (confidence:.2f), ' f'coords = ((xmin), (ymin)), ((xmax), (ymax)) ')