Upload inference.py

inference.py

@@ -314,3 +314,21 @@ class ModelInference:
             class_names[class_id_str] = class_name
 
         return class_names
+
+    def get_tensor(self, crop: Image.Image):
+        """Preprocess a crop into a numpy array for batch inference."""
+        img = np.array(crop)
+        img = cv2.resize(img, (self.img_size, self.img_size))
+        return img
+
+    def classify_batch(self, batch):
+        """Run inference on a batch of preprocessed numpy arrays."""
+        probs = self.model.predict(batch, verbose=0)
+        results = []
+        for p in probs:
+            classifications = [
+                [self.class_ids[i], float(p[i])]
+                for i in range(len(self.class_ids))
+            ]
+            results.append(classifications)
+        return results