administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix ROCm input name

Browse Source
This commit is contained in:
Nicolas Mowen 2024-09-26 11:19:31 -06:00
parent a5595189ed
commit fb07319831
1 changed files with 3 additions and 2 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):
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],