Skip to content

RTDETR Model

sahi.models.rtdetr

RT-DETR detection model wrapper for SAHI.

Provides integration with Ultralytics RT-DETR real-time detection transformer models.

Classes

RTDetrDetectionModel

Bases: UltralyticsDetectionModel

RT-DETR object detection model.

Wraps Ultralytics RT-DETR for real-time detection inference.

Source code in sahi/models/rtdetr.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class RTDetrDetectionModel(UltralyticsDetectionModel):
    """RT-DETR object detection model.

    Wraps Ultralytics RT-DETR for real-time detection inference.
    """

    def load_model(self) -> None:
        """Detection model is initialized and set to self.model."""
        from ultralytics import RTDETR

        try:
            model_source = self.model_path or "rtdetr-l.pt"
            model = RTDETR(model_source)
            model.to(self.device)
            self.set_model(model)
        except Exception as e:
            raise TypeError("model_path is not a valid rtdetr model path: ", e)
Functions
load_model()

Detection model is initialized and set to self.model.

Source code in sahi/models/rtdetr.py 
17
18
19
20
21
22
23
24
25
26
27
def load_model(self) -> None:
    """Detection model is initialized and set to self.model."""
    from ultralytics import RTDETR

    try:
        model_source = self.model_path or "rtdetr-l.pt"
        model = RTDETR(model_source)
        model.to(self.device)
        self.set_model(model)
    except Exception as e:
        raise TypeError("model_path is not a valid rtdetr model path: ", e)