|
@@ -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)
|