trainer.py 文件源码

def get_image_detections(path):
    tb = TB()

    #cv2.namedWindow("outputs", cv2.WINDOW_NORMAL)
    sample = cv2.imread(path)[:, :, :3]
    sample = cv2.resize(sample, (300,300))

    boxes_, confidences_ = tb.single_image(sample)

    return boxes_, confidences_