Refactor model assets structure, for rfdetr 1.4.3+ (#6)

* Bump `rfdetr` dependency to `1.4.3` for compatibility
* Add MD5 hashes for RF-DETR+ XLarge and 2XLarge model weights
* Add inference tests for RF-DETR+ XLarge and 2XLarge models with parameterized resolutions

Author: Borda

.github/workflows/ci-tests-cpu.yml

@@ -48,7 +48,7 @@ jobs:
         run: uv pip install -e . --group tests
 
       - name: Run the Test
-        run: uv run pytest -m "not gpu" --cov=rfdetr_plus --cov-report=xml
+        run: pytest -m "not gpu" --cov=rfdetr_plus --cov-report=xml
 
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v5

pyproject.toml

@@ -35,7 +35,7 @@ classifiers = [
 keywords = ["machine-learning", "deep-learning", "vision", "ML", "DL", "AI", "DETR", "RF-DETR", "Roboflow"]
 
 dependencies = [
-    "rfdetr>=1.4.1,<2",
+    "rfdetr>=1.4.3,<2",
 ]
 
 [project.optional-dependencies]

tests/test_version.py src/rfdetr_plus/assets/__init__.pytests/test_version.py renamed to src/rfdetr_plus/assets/__init__.py

@@ -3,12 +3,9 @@
 # Copyright (c) 2026 Roboflow, Inc. All Rights Reserved.
 # Licensed under the Platform Model License 1.0 [see LICENSE for details]
 # ------------------------------------------------------------------------
-"""Test basic package functionality."""
 
+"""RF-DETR+ model assets module."""
 
-def test_version() -> None:
-    """Test that the package version is accessible."""
-    from rfdetr_plus import __version__
+from rfdetr_plus.assets.model_weights import ModelWeights
 
-    assert isinstance(__version__, str)
-    assert len(__version__) > 0
+__all__ = ["ModelWeights"]

src/rfdetr_plus/assets/model_weights.py

@@ -0,0 +1,54 @@
+# ------------------------------------------------------------------------
+# RF-DETR+
+# Copyright (c) 2026 Roboflow, Inc. All Rights Reserved.
+# Licensed under the Platform Model License 1.0 [see LICENSE for details]
+# ------------------------------------------------------------------------
+
+"""
+RF-DETR+ Model weights registry.
+
+Provides ModelWeights enum for platform-licensed large-scale models,
+compatible with rf-detr's asset structure introduced in version 1.4.3.
+"""
+
+from rfdetr.assets.model_weights import ModelWeightAsset, ModelWeightsBase
+
+
+class ModelWeights(ModelWeightsBase):
+    """
+    Enumeration of RF-DETR+ platform-licensed model assets.
+
+    Inherits from rf-detr's ModelWeightsBase to ensure compatibility.
+
+    Each enum member's value is a ModelWeightAsset instance containing:
+    - filename: The local filename for the model weights
+    - url: The download URL
+    - md5_hash: The expected MD5 hash for integrity validation
+
+    Example:
+        >>> from rfdetr_plus.assets import ModelWeights
+        >>> asset = ModelWeights.RF_DETR_XLARGE
+        >>> asset.filename
+        'rf-detr-xlarge.pth'
+        >>> asset.url
+        'https://storage.googleapis.com/rfdetr/platform-licensed/rf-detr-xlarge.pth'
+    """
+
+    # Platform-Licensed Detection Models (XLarge and 2XLarge)
+    # These models are subject to the Platform Model License 1.0
+    RF_DETR_XLARGE = ModelWeightAsset(
+        "rf-detr-xlarge.pth",
+        "https://storage.googleapis.com/rfdetr/platform-licensed/rf-detr-xlarge.pth",
+        "6ddf834f2bc5bed3214a82f9b0aaeed7",
+    )
+    RF_DETR_XXLARGE = ModelWeightAsset(
+        "rf-detr-xxlarge.pth",
+        "https://storage.googleapis.com/rfdetr/platform-licensed/rf-detr-xxlarge.pth",
+        "e3204689c1f0280427e4c33e6a2ac6cd",
+    )
+
+    # All methods inherited from ModelWeightsBase:
+    # - from_filename(filename) -> Optional[ModelWeightAsset]
+    # - get_url(filename) -> Optional[str]
+    # - get_md5(filename) -> Optional[str]
+    # - list_models() -> list[str]

src/rfdetr_plus/models/downloads.py

@@ -4,7 +4,15 @@
 # Licensed under the Platform Model License 1.0 [see LICENSE for details]
 # ------------------------------------------------------------------------
 
-PLATFORM_MODELS = {
-    "rf-detr-xlarge.pth": "https://storage.googleapis.com/rfdetr/platform-licensed/rf-detr-xlarge.pth",
-    "rf-detr-xxlarge.pth": "https://storage.googleapis.com/rfdetr/platform-licensed/rf-detr-xxlarge.pth",
-}
+"""
+Legacy model downloads dictionary.
+
+DEPRECATED: Use rfdetr_plus.assets.ModelWeights instead.
+This dictionary is maintained for backward compatibility only.
+"""
+
+from rfdetr_plus.assets import ModelWeights
+
+# Legacy dictionary for backward compatibility
+# New code should use rfdetr_plus.assets.ModelWeights
+PLATFORM_MODELS = {asset.filename: asset.url for asset in ModelWeights}

tests/test_inference.py

@@ -0,0 +1,31 @@
+# ------------------------------------------------------------------------
+# RF-DETR+
+# Copyright (c) 2026 Roboflow, Inc. All Rights Reserved.
+# Licensed under the Platform Model License 1.0 [see LICENSE for details]
+# ------------------------------------------------------------------------
+"""Test basic package functionality."""
+
+import numpy as np
+import pytest
+
+from rfdetr_plus import RFDETR2XLarge, RFDETRXLarge
+
+
+@pytest.mark.parametrize(
+    ("model_class", "resolution"),
+    [
+        (RFDETRXLarge, 700),
+        (RFDETR2XLarge, 880),
+    ],
+)
+def test_model_inference(model_class, resolution) -> None:
+    """Test that we can instantiate RF-DETR+ models and run inference."""
+    # Instantiate and run inference
+    rf_detr = model_class()
+    dummy_image = np.random.randint(0, 255, (resolution, resolution, 3), dtype=np.uint8)
+
+    # Run inference - this verifies the model can be instantiated and used
+    predictions = rf_detr.predict(dummy_image, conf_threshold=0.1)
+
+    # Verify predictions were returned
+    assert predictions is not None