Merge pull request #1667 from roboflow/fix/training-resolution

Fix issue with inference resolution not denoted correctly in RFDetr

Author: PawelPeczek-Roboflow

inference_experimental/inference_exp/models/rfdetr/rfdetr_instance_segmentation_pytorch.py

@@ -124,6 +124,7 @@ def from_pretrained(
         model_config = CONFIG_FOR_MODEL_TYPE[model_type](device=device)
         checkpoint_num_classes = weights_dict["class_embed.bias"].shape[0]
         model_config.num_classes = checkpoint_num_classes - 1
+        model_config.resolution = inference_config.network_input.training_input_size.height
         model = build_model(config=model_config)
         model.load_state_dict(weights_dict)
         model = model.eval().to(device)

inference_experimental/inference_exp/models/rfdetr/rfdetr_object_detection_pytorch.py

@@ -130,6 +130,7 @@ def from_pretrained(
         model_config = CONFIG_FOR_MODEL_TYPE[model_type](device=device)
         checkpoint_num_classes = weights_dict["class_embed.bias"].shape[0]
         model_config.num_classes = checkpoint_num_classes - 1
+        model_config.resolution = inference_config.network_input.training_input_size.height
         model = build_model(config=model_config)
         model.load_state_dict(weights_dict)
         model = model.eval().to(device)

inference_experimental/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "inference-exp"
-version = "0.15.4"
+version = "0.15.5"
 description = "Experimental vresion of inference package which is supposed to evolve into inference 1.0"
 readme = "README.md"
 requires-python = ">=3.10,<3.13"