|
|
@@ -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, True)
|
|
|
+ image_data = resize_image(image, self.input_size, False)
|
|
|
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)
|
|
|
@@ -89,4 +89,4 @@ def resize_image(image, size, letterbox_image):
|
|
|
new_image.paste(image, ((w - nw) // 2, (h - nh) // 2))
|
|
|
else:
|
|
|
new_image = image.resize((w, h), Image.BICUBIC)
|
|
|
- return new_image
|
|
|
+ return new_image
|
