Upload processor

image_processing.py

@@ -6,11 +6,11 @@ from transformers import BaseImageProcessorFast
 
 
 class LSPDetrImageProcessor(BaseImageProcessorFast):
+    image_processor_type = BaseImageProcessorFast
     image_mean = [0.485, 0.456, 0.406]
     image_std = [0.229, 0.224, 0.225]
     do_rescale = True
     do_normalize = True
-    return_tensors = "pt"
 
     def post_process(self, outputs: dict[str, Tensor]) -> list[dict[str, Tensor]]:
         """Converts the raw output into polygons.
@@ -59,11 +59,11 @@ class LSPDetrImageProcessor(BaseImageProcessorFast):
                 device=result["polygons"].device,
             )
 
-            for i, polygon in enumerate(result["polygons"]):
-                img = Image.fromarray(masks[i].cpu().numpy())
+            for j, polygon in enumerate(result["polygons"]):
+                img = Image.fromarray(masks[j].cpu().numpy())
                 canvas = ImageDraw.Draw(img)
                 canvas.polygon(xy=polygon.flatten().tolist(), outline=1, fill=1)
-                masks[i] = torch.from_numpy(np.asarray(img))
+                masks[j] = torch.tensor(np.asarray(img))
 
             results[i]["masks"] = masks
 

preprocessor_config.json

@@ -1,4 +1,7 @@
 {
+  "auto_map": {
+    "AutoImageProcessor": "image_processing.LSPDetrImageProcessor"
+  },
   "crop_size": null,
   "data_format": "channels_first",
   "default_to_square": true,