Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-14 17:52:10 +02:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats 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.width = model_info["width"]
self.height = model_info["height"] self.height = model_info["height"]
self.input_tensor = model_info["inputShape"] self.input_tensor = InputTensorEnum(model_info["inputShape"])
self.input_pixel_format = model_info["pixelFormat"] self.input_pixel_format = PixelFormatEnum(model_info["pixelFormat"])
self.model_type = model_info["type"] self.model_type = ModelTypeEnum(model_info["type"])
if model_info.get("inputDataType"): if model_info.get("inputDataType"):
self.input_dtype = model_info["inputDataType"] self.input_dtype = InputDTypeEnum(model_info["inputDataType"])
# generate list of attribute labels # generate list of attribute labels
self.attributes_map = { self.attributes_map = {