Skip to content

Rfdetr exp

RFDetrExperimentalModel

Bases: InferenceExpObjectDetectionModelAdapter

Adapter for RF-DETR using inference_exp AutoModel backend.

This class wraps an inference_exp AutoModel to present the same interface as legacy models in the inference server.

Source code in inference/models/rfdetr/rfdetr_exp.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
class RFDetrExperimentalModel(InferenceExpObjectDetectionModelAdapter):
    """Adapter for RF-DETR using inference_exp AutoModel backend.

    This class wraps an inference_exp AutoModel to present the same interface
    as legacy models in the inference server.
    """

    def map_inference_kwargs(self, kwargs: dict) -> dict:
        return {
            "threshold": kwargs.get("confidence"),
        }