未验证 提交 042d4a49 编写于 作者: B Bubbliiiing 提交者: GitHub

Update get_dr_txt.py

上级 19384f63
...@@ -28,7 +28,7 @@ class mAP_Yolo(YOLO): ...@@ -28,7 +28,7 @@ class mAP_Yolo(YOLO):
f = open("./input/detection-results/"+image_id+".txt","w") f = open("./input/detection-results/"+image_id+".txt","w")
image_shape = np.array(np.shape(image)[0:2]) image_shape = np.array(np.shape(image)[0:2])
crop_img = np.array(letterbox_image(image, (self.model_image_size[0],self.model_image_size[1]))) crop_img = np.array(letterbox_image(image, (self.model_image_size[1],self.model_image_size[0])))
photo = np.array(crop_img,dtype = np.float32) photo = np.array(crop_img,dtype = np.float32)
photo /= 255.0 photo /= 255.0
photo = np.transpose(photo, (2, 0, 1)) photo = np.transpose(photo, (2, 0, 1))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册