Skip to content

RTDETRModel

sahi.models.rtdetr

Classes

RTDetrDetectionModel

Bases: UltralyticsDetectionModel

Source code in sahi/models/rtdetr.py 
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
class RTDetrDetectionModel(UltralyticsDetectionModel):
    def __init__(self, *args, **kwargs):
        self.required_packages = list(getattr(self, "required_packages", [])) + ["ultralytics"]
        super().__init__(*args, **kwargs)

    def load_model(self):
        """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 rtdet model path: ", e)
Functions
load_model()

Detection model is initialized and set to self.model.

Source code in sahi/models/rtdetr.py 
 9
10
11
12
13
14
15
16
17
18
19
def load_model(self):
    """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 rtdet model path: ", e)