diff --git a/get_dr_txt.py b/get_dr_txt.py index ae354a21030fa7dd9841da49de16cfa07f5c47e4..1235ef357b05cbfcbc23c26e22535852ccd0332a 100644 --- a/get_dr_txt.py +++ b/get_dr_txt.py @@ -22,6 +22,7 @@ class mAP_Yolo(YOLO): #---------------------------------------------------# def detect_image(self,image_id,image): self.confidence = 0.01 + self.iou = 0.5 f = open("./input/detection-results/"+image_id+".txt","w") image_shape = np.array(np.shape(image)[0:2])