Upload model

configuration_rf_detr.py

@@ -72,7 +72,7 @@ class RFDetrOnnxConfig(ViTOnnxConfig):
         return OrderedDict(
             {
                 "pixel_values": {0: "batch_size", 1: "num_channels", 2: "height", 3: "width"},
-                "pixel_mask": {0: "batch_size", 2: "height", 3: "width"},
+                "pixel_mask": {0: "batch_size", 1: "height", 2: "width"},
             }
         )
 

modeling_rf_detr.py

@@ -118,7 +118,7 @@ class RFDetrModelForObjectDetection(PreTrainedModel):
                 label["labels"] = label["labels"].to(self.config.device)
 
     def forward(self, pixel_values: torch.Tensor, pixel_mask: torch.Tensor, labels=None, **kwargs) -> ModelOutput:
-        resize = Resize((self.config.resolution, self.config.resolution)) 
+        resize = Resize((self.config.resolution, self.config.resolution), antialias=False) # antialias set to False for onnx export
 
         if labels is not None:
             self.validate_labels(labels)