فهرست منبع

修改fasterrcnn报错问题

zhy 2 هفته پیش
والد
کامیت
3a5163dd89
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      watermark_verify/inference/rcnn_inference.py

+ 1 - 1
watermark_verify/inference/rcnn_inference.py

@@ -39,7 +39,7 @@ class FasterRCNNInference:
         # ---------------------------------------------------------#
         if not (len(np.shape(image)) == 3 and np.shape(image)[2] == 3):
             image = image.convert('RGB')
-        image_data = resize_image(image, self.input_size, False)
+        image_data = resize_image(image, self.input_size, True)
         image_data = np.array(image_data, dtype='float32')
         image_data = image_data / 255.0
         image_data = np.expand_dims(np.transpose(image_data, self.swap).copy(), 0)