Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-04-19 01:16:11 +02:00
Code Issues Projects Releases Wiki Activity

Fixes (#13990)

Browse Source 
* Fix ROCm input name

* Fix incorrect parsing of None
This commit is contained in:
Nicolas Mowen 2024-09-26 11:42:21 -06:00 committed by GitHub
parent a5595189ed
commit a65aaab849
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/detectors/plugins/rocm.py
View File

@ -125,8 +125,9 @@ class ROCmDetector(DetectionApi):
def detect_raw(self, tensor_input): def detect_raw(self, tensor_input):
model_input_name = self.model.get_parameter_names()[0] model_input_name = self.model.get_parameter_names()[0]
model_input_name = self.model.get_inputs()[0].name model_input_shape = tuple(
model_input_shape = self.model.get_inputs()[0].shape self.model.get_parameter_shapes()[model_input_name].lens()
)
tensor_input = cv2.dnn.blobFromImage( tensor_input = cv2.dnn.blobFromImage(
tensor_input[0], tensor_input[0],

5
frigate/events/maintainer.py
View File

@ -163,6 +163,7 @@ class EventProcessor(threading.Thread):
) )
) )
try:
attributes = [ attributes = [
( (
None None
@ -179,6 +180,10 @@ class EventProcessor(threading.Thread):
) )
for a in event_data["snapshot"]["attributes"] for a in event_data["snapshot"]["attributes"]
] ]
except TypeError:
logger.warning(
f"Failed to parse attributes of event data, event data is {event_data}"
)
# keep these from being set back to false because the event # keep these from being set back to false because the event
# may have started while recordings and snapshots were enabled # may have started while recordings and snapshots were enabled