administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-08 12:15:25 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update detector_config.py

Browse Source
This commit is contained in:
alec-groff 2024-01-26 14:25:00 -06:00 committed by GitHub
parent 64a91f552f
commit eee625bef2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/detectors/detector_config.py
View File

@ -51,6 +51,9 @@ class ModelConfig(BaseModel):
model_type: ModelTypeEnum = Field( model_type: ModelTypeEnum = Field(
default=ModelTypeEnum.ssd, title="Object Detection Model Type" default=ModelTypeEnum.ssd, title="Object Detection Model Type"
) )
order: list[int] = Field(
default=[0,1,2,3], title="Order Output Tensors [0=boxes,1=scores,2=class_ids,3=count]"
)
_merged_labelmap: Optional[Dict[int, str]] = PrivateAttr() _merged_labelmap: Optional[Dict[int, str]] = PrivateAttr()
_colormap: Dict[int, Tuple[int, int, int]] = PrivateAttr() _colormap: Dict[int, Tuple[int, int, int]] = PrivateAttr()
_model_hash: str = PrivateAttr() _model_hash: str = PrivateAttr()