Upload folder using huggingface_hub

inference.py

@@ -1,19 +1,34 @@
 from PIL import Image
 import torch
 import numpy as np
-from transformers import PreTrainedModel
 import onnxruntime
+import os
 
 class FasterRCNNInference:
     def __init__(self, model_path):
         # Load ONNX model
         self.ort_session = onnxruntime.InferenceSession(f"{model_path}/model.onnx")
-        self.processor = torch.load(f"{model_path}/processor.bin")
+        
+        # Try to load Hugging Face image processor
+        try:
+            from transformers import AutoImageProcessor
+            self.image_processor = AutoImageProcessor.from_pretrained(model_path)
+            self.use_hf_processor = True
+            print("Using Hugging Face image processor")
+        except Exception as e:
+            print(f"Could not load Hugging Face image processor: {e}")
+            print("Falling back to custom processor")
+            self.processor = torch.load(f"{model_path}/processor.bin")
+            self.use_hf_processor = False
         
     def predict(self, images, threshold=0.5):
         # Preprocess
-        inputs = self.processor(images)
-        pixel_values = inputs["pixel_values"].numpy()
+        if self.use_hf_processor:
+            inputs = self.image_processor(images=images, return_tensors="pt")
+            pixel_values = inputs["pixel_values"].numpy()
+        else:
+            inputs = self.processor(images)
+            pixel_values = inputs["pixel_values"].numpy()
         
         # Run inference
         outputs = self.ort_session.run(None, {"pixel_values": pixel_values})

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:456470547ad0365bb4229e2d1c81ff99015984487753ca26e694f7ba8847e37d
+oid sha256:83785c773e3567b4244528396f7f553de877cc3e04724a383b723b647f2fc670
 size 331069979

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:38a17a2591648db50e1b701341cf3ef261db1cec76e5131975a82f6600769455
+oid sha256:a184954eea223d4339ca5e728ed03ef966bcbdb3fd9a499c1a1af6c60c282bd2
 size 331180436

preprocessor_config.json

@@ -0,0 +1,26 @@
+{
+  "do_convert_annotations": true,
+  "do_normalize": true,
+  "do_pad": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "format": "coco_detection",
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "DetrImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "pad_size": null,
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "longest_edge": 1333,
+    "shortest_edge": 800
+  }
+}