diff --git a/yolo/detect/multi_yolo_trt_detect_track_trt10_yolo11.py b/yolo/detect/multi_yolo_trt_detect_track_trt10_yolo11.py index d9ea9b8..4326f6f 100644 --- a/yolo/detect/multi_yolo_trt_detect_track_trt10_yolo11.py +++ b/yolo/detect/multi_yolo_trt_detect_track_trt10_yolo11.py @@ -723,6 +723,10 @@ class MultiYoloTrtDetectorTrackId_TRT10_YOLO11: elif isinstance(model_class_names, dict): class_name = model_class_names.get(class_id, "unknown") + # 过滤不允许的类别 + if allowed and class_name not in allowed: + continue + # 创建DetectionResult对象 detection = DetectionResult( bbox=box,