feat: resizing before inference

script.py

@@ -21,8 +21,20 @@ class ONNXWorker:
             providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
         else:
             providers = ["CPUExecutionProvider"]
+
+        print(f"Using {providers}")
         self.ort_session = ort.InferenceSession(onnx_path, providers=providers)
 
+    def _resize_image(self, image: np.ndarray) -> np.ndarray:
+        """
+
+        :param image:
+        :return:
+        """
+
+        newsize = (300, 300)
+        im1 = im1.resize(newsize)
+
     def predict_image(self, image: np.ndarray) -> list():
         """Run inference using ONNX runtime.
 
@@ -44,8 +56,11 @@ def make_submission(test_metadata, model_path, output_csv_path="./submission.csv
 
     for _, row in tqdm(test_metadata.iterrows(), total=len(test_metadata)):
         image_path = os.path.join(images_root_path, row.filename)
-        test_image = np.asarray(Image.open(image_path).convert("RGB"))
-        logits = model.predict_image(test_image)
+
+        test_image = Image.open(image_path).convert("RGB")
+        test_image_resized = np.asarray(test_image.resize((256, 256)))
+
+        logits = model.predict_image(test_image_resized)
 
         predictions.append(np.argmax(logits))