feat: 新增标签过滤器类

utils.py

@@ -0,0 +1,49 @@
+from enum import Enum
+
+# 根据标签修改
+class tlabel(Enum):
+    BBLOCK   = 1    # 蓝色方块
+    RBLOCK   = 2    # 红色方块
+    HOSPITAL = 3    # 医院
+    BBALL    = 4    # 蓝球
+    YBALL    = 5    # 黄球
+    TOWER    = 6    # 通信塔
+    RBALL    = 7    # 红球
+    BASKET   = 8    # 球筐
+    MARKL    = 9    # 指向标
+    MARKR    = 10   # 指向标
+    SPILLAR  = 11   # 小柱体 (红色)
+    MPILLAR  = 12   # 中柱体 (蓝色)
+    LPILLAR  = 13   # 大柱体 (红色)
+    SIGN     = 14   # 文字标牌
+    TARGET   = 15   # 目标靶
+    SHELTER  = 16   # 停车区
+    BASE     = 17   # 基地
+
+class label_filter:
+    def __init__(self, list_src):
+        self.num = 0
+        self.pos = []
+        self.list = list_src # 获取目标检测输出的接口 (含标签，位置，置信度)
+        # TODO 添加置信度阈值
+        pass
+    def get(self, tlabel):
+        # TODO 循环查找匹配的标签值
+        # TODO 返回对应标签的个数，以及坐标列表
+        return self.num, self.pos
+    def find(self, tlabel):
+        # TODO 遍历返回的列表，有对应标签则返回 True
+        return False
+    def aim_left(self, tlabel):
+        # TODO 如果标签存在，则返回列表中位置最靠左的目标框和中心的偏移值
+        error = 0
+        return error
+    def aim_right(self, tlabel):
+        # TODO 如果标签存在，则返回列表中位置最靠右的目标框和中心的偏移值
+        error = 0
+        return error
+    def aim_near(self, tlabel):
+        # TODO 如果标签存在，则返回列表中位置最近的目标框和中心的偏移值
+        error = 0
+        return error
+