This commit is contained in:
Nicolas Mowen 2024-10-01 07:51:56 -06:00
parent 2ab14d0ad6
commit b9044cbfc6

View File

@ -752,8 +752,7 @@ def process_frames(
for attributes in attribute_detections.values():
for attribute in attributes:
filtered_objects = filter(
lambda o: o["label"] in attribute_detections.keys()
and attribute.label
lambda o: attribute.label
in model_config.attributes_map.get(o["label"], []),
all_objects,
)