administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-05 04:57:42 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Improve typing

Browse Source
This commit is contained in:
Nicolas Mowen 2025-04-17 11:18:44 -06:00
parent 7b851cca48
commit 955b483cf5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/detection_api.py
View File

@ -4,7 +4,7 @@ from typing import List
import numpy as np import numpy as np
from frigate.detectors.detector_config import ModelTypeEnum from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -14,7 +14,7 @@ class DetectionApi(ABC):
supported_models: List[ModelTypeEnum] supported_models: List[ModelTypeEnum]
@abstractmethod @abstractmethod
def __init__(self, detector_config): def __init__(self, detector_config: BaseDetectorConfig):
self.detector_config = detector_config self.detector_config = detector_config
self.thresh = 0.5 self.thresh = 0.5
self.height = detector_config.model.height self.height = detector_config.model.height