Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-01-31 00:18:55 +01: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):
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],

35
frigate/events/maintainer.py
View File

@ -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