From fb073198316c48ed2fe28e44a97ce6e69ff9571f Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 26 Sep 2024 11:19:31 -0600 Subject: [PATCH] Fix ROCm input name --- frigate/detectors/plugins/rocm.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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],