diff --git a/frigate/detectors/detector_config.py b/frigate/detectors/detector_config.py index 0ac93d858..be12e7fcc 100644 --- a/frigate/detectors/detector_config.py +++ b/frigate/detectors/detector_config.py @@ -40,6 +40,7 @@ class ModelTypeEnum(str, Enum): dfine = "dfine" yologeneric = "yolo-generic" + class ModelConfig(BaseModel): path: Optional[str] = Field(None, title="Custom Object detection model path.") labelmap_path: Optional[str] = Field(