diff --git a/frigate/detectors/plugins/rocm.py b/frigate/detectors/plugins/rocm.py index 5203934e9..655973dfd 100644 --- a/frigate/detectors/plugins/rocm.py +++ b/frigate/detectors/plugins/rocm.py @@ -125,8 +125,9 @@ class ROCmDetector(DetectionApi): def detect_raw(self, tensor_input): model_input_name = self.model.get_parameter_names()[0] - model_input_name = self.model.get_inputs()[0].name - model_input_shape = self.model.get_inputs()[0].shape + model_input_shape = tuple( + self.model.get_parameter_shapes()[model_input_name].lens() + ) tensor_input = cv2.dnn.blobFromImage( tensor_input[0], diff --git a/frigate/events/maintainer.py b/frigate/events/maintainer.py index 6707bfb41..10cdb6b43 100644 --- a/frigate/events/maintainer.py +++ b/frigate/events/maintainer.py @@ -163,22 +163,27 @@ class EventProcessor(threading.Thread): ) ) - attributes = [ - ( - None - if event_data["snapshot"] is None - else { - "box": to_relative_box( - width, - height, - a["box"], - ), - "label": a["label"], - "score": a["score"], - } + try: + attributes = [ + ( + None + if event_data["snapshot"] is None + else { + "box": to_relative_box( + width, + height, + a["box"], + ), + "label": a["label"], + "score": a["score"], + } + ) + for a in event_data["snapshot"]["attributes"] + ] + except TypeError: + logger.warning( + f"Failed to parse attributes of event data, event data is {event_data}" ) - for a in event_data["snapshot"]["attributes"] - ] # keep these from being set back to false because the event # may have started while recordings and snapshots were enabled