Update script.py

script.py

@@ -26,21 +26,16 @@ class ONNXWorker:
         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)
+        new_size = (384, 384) 
+        return np.array(Image.fromarray(image).resize(new_size))
+    
 
     def predict_image(self, image: np.ndarray) -> list():
-        """Run inference using ONNX runtime.
-        :param image: Input image as numpy array.
-        :return: A list with logits and confidences.
-        """
 
-        logits, _ = self.ort_session.run(None, {"input": image.astype(dtype=np.uint8)})
+        """Run inference using ONNX runtime."""
+        resized_image = self._resize_image(image)
+        logits = self.ort_session.run(None, {"input": resized_image})
 
         return logits.tolist()
 
@@ -56,7 +51,7 @@ def make_submission(test_metadata, model_path, output_csv_path="./submission.csv
         image_path = os.path.join(images_root_path, row.filename)
 
         test_image = Image.open(image_path).convert("RGB")
-        test_image_resized = np.asarray(test_image.resize((256, 256)))
+        test_image_resized = np.asarray(test_image.resize((384, 384)))
 
         logits = model.predict_image(test_image_resized)