zishuo/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-26 19:41:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Correctly cast to enum when loading plus config (#20031)

Browse Source
This commit is contained in:
Nicolas Mowen
2025-09-11 16:39:34 -06:00
committed by GitHub
parent fad28a764c
commit ba650af6f2
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/detectors/detector_config.py
View File

@@ -154,12 +154,12 @@ class ModelConfig(BaseModel):
self.width = model_info["width"]
self.height = model_info["height"]
self.input_tensor = model_info["inputShape"]
self.input_pixel_format = model_info["pixelFormat"]
self.model_type = model_info["type"]
self.input_tensor = InputTensorEnum(model_info["inputShape"])
self.input_pixel_format = PixelFormatEnum(model_info["pixelFormat"])
self.model_type = ModelTypeEnum(model_info["type"])
if model_info.get("inputDataType"):
self.input_dtype = model_info["inputDataType"]
self.input_dtype = InputDTypeEnum(model_info["inputDataType"])
# generate list of attribute labels
self.attributes_map = {