Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues 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"))) devices = list(filter(lambda d: d.startswith("render"), os.listdir("/dev/dri")))
if len(devices) < 2: if len(devices) < 2:
self._selected_gpu = devices[0] self._selected_gpu = f"/dev/dri/{devices[0]}"
return self._selected_gpu return self._selected_gpu
for device in devices: for device in devices:

6
frigate/video.py
View File

@ -734,7 +734,7 @@ def process_frames(
object_tracker.update_frame_times(frame_time) object_tracker.update_frame_times(frame_time)
# group the attribute detections based on what label they apply to # 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(): for label, attribute_labels in model_config.attributes_map.items():
attribute_detections[label] = [ attribute_detections[label] = [
ObjectAttribute(d) ObjectAttribute(d)
@ -752,7 +752,9 @@ def process_frames(
for attributes in attribute_detections.values(): for attributes in attribute_detections.values():
for attribute in attributes: for attribute in attributes:
filtered_objects = filter( 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) selected_object_id = attribute.find_best_object(filtered_objects)