Spartan/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-22 08:57:20 -06:00
Code Issues Packages Projects Releases Wiki Activity

Bug fixes (#14092)

Browse Source 
* Fix path of selected GPU

* Fix selection bug

* Simplify
This commit is contained in:
Nicolas Mowen 2024-10-01 07:54:27 -06:00 committed by GitHub
parent 15fa55c223
commit 3cff0df0ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/ffmpeg_presets.py
View File

@ -32,7 +32,7 @@ class LibvaGpuSelector:
devices = list(filter(lambda d: d.startswith("render"), os.listdir("/dev/dri")))
if len(devices) < 2:
self._selected_gpu = devices[0]
self._selected_gpu = f"/dev/dri/{devices[0]}"
return self._selected_gpu
for device in devices:

6
frigate/video.py
View File

@ -734,7 +734,7 @@ def process_frames(
object_tracker.update_frame_times(frame_time)
# group the attribute detections based on what label they apply to
attribute_detections: dict[str, ObjectAttribute] = {}
attribute_detections: dict[str, list[ObjectAttribute]] = {}
for label, attribute_labels in model_config.attributes_map.items():
attribute_detections[label] = [
ObjectAttribute(d)
@ -752,7 +752,9 @@ def process_frames(
for attributes in attribute_detections.values():
for attribute in attributes:
filtered_objects = filter(
lambda o: o["label"] in attribute_detections.keys(), all_objects
lambda o: attribute.label
in model_config.attributes_map.get(o["label"], []),
all_objects,
)
selected_object_id = attribute.find_best_object(filtered_objects)