From be994728372999541af9317c0b3f06f64b51367f Mon Sep 17 00:00:00 2001
From: martin <1486756632@qq.com>
Date: Wed, 8 Apr 2026 10:40:13 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dbug=EF=BC=8C=E8=BF=87?=
 =?UTF-8?q?=E6=BB=A4=E4=B8=8D=E5=85=81=E8=AE=B8=E7=9A=84=E7=B1=BB=E5=88=AB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 yolo/detect/multi_yolo_trt_detect_track_trt10_yolo11.py | 4 ++++
 1 file changed, 4 insertions(+)

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,