app update

app.py

@@ -13,6 +13,7 @@ csv.field_size_limit(sys.maxsize)
 
 import gc
 import json
+import random
 from collections import Counter
 from pathlib import Path
 
@@ -204,6 +205,20 @@ def color_set(cls_num):
     return color_list
 
 
+# 随机生成浅色系或者深色系
+def random_color(cls_num, is_light=True):
+    color_list = []
+    for i in range(cls_num):
+        color = (
+            random.randint(0, 127) + int(is_light) * 128,
+            random.randint(0, 127) + int(is_light) * 128,
+            random.randint(0, 127) + int(is_light) * 128,
+        )
+        color_list.append(color)
+
+    return color_list
+
+
 # 检测绘制
 def pil_draw(img, score_l, bbox_l, cls_l, cls_index_l, textFont, color_list, opt):
 
@@ -227,7 +242,7 @@ def pil_draw(img, score_l, bbox_l, cls_l, cls_index_l, textFont, color_list, opt
             img_pil.multiline_text(
                 (xmin, ymin),
                 countdown_msg,
-                fill=(255, 255, 255),
+                fill=(0, 0, 0),
                 font=textFont,
                 align="center",
             )
@@ -280,7 +295,7 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
     model.max_det = int(max_num)  # 最大检测框数
     model.classes = model_cls  # 模型类别
 
-    color_list = color_set(len(model_cls_name_cp))  # 设置颜色
+    color_list = random_color(len(model_cls_name_cp), True)
 
     img_size = img.size  # 帧尺寸