VOC_process_util.py 文件源码

def proc_image(img, bboxes=None, border_size=1, font_size=0.3, xconfidence=0.5):
    # img = cv2.resize(img, (512, 512))
    if bboxes is not None:
        for bbox in bboxes:
            if bbox.shape[0] == 5:
                c, xmin, ymin, xmax, ymax = [int(x) for x in bbox]
                mx = int((xmin+xmax)/2)
                my = int((ymin+ymax)/2)
                cv2.putText(img, "%d: %s"%(c, rcmap[c]), (mx, my), cv2.FONT_HERSHEY_SIMPLEX, font_size, (0,255,155), border_size)
                cv2.rectangle(img,(xmin, ymax),(xmax, ymin),(255,15,5),border_size)
            else:
                c, xmin, ymin, xmax, ymax = [int(x) for x in bbox[:5]]
                confidence = bbox[5]
                if confidence > xconfidence:
                    mx = int((xmin+xmax)/2)
                    my = int((ymin+ymax)/2)
                    cv2.putText(img, str(confidence), (mx, my), cv2.FONT_HERSHEY_SIMPLEX, font_size, (255,0,155), border_size)
                    cv2.putText(img, "%d: %s"%(c, rcmap[c]), (mx, my + 15), cv2.FONT_HERSHEY_SIMPLEX, font_size, (255,0,155), border_size)
                    cv2.rectangle(img,(xmin, ymax),(xmax, ymin),(0,255,155),border_size)
    return img