app update

app.py

@@ -34,6 +34,8 @@ SimSun = font_manager.FontProperties(fname=SimSun_path, size=12)
 # 新罗马字体
 TimesNesRoman = font_manager.FontProperties(fname=TimesNesRoman_path, size=12)
 
+from copy import deepcopy
+
 import yaml
 from PIL import Image, ImageDraw, ImageFont
 
@@ -161,26 +163,14 @@ def check_online():
 def model_loading(img_path, conf, iou, infer_size, yolo_model="yolov8n.pt"):
     model = YOLO(yolo_model)
 
-    results = model(img_path, imgsz=infer_size, conf=conf, iou=iou)
-    return results[0].tolist()
-
-
-# 检测信息
-def export_json(results, img_size):
-
-    return [[{
-        "ID": i,
-        "CLASS": int(result[i][5]),
-        "CLASS_NAME": model_cls_name_cp[int(result[i][5])],
-        "BOUNDING_BOX": {
-            "XMIN": round(result[i][:4].tolist()[0], 6),
-            "YMIN": round(result[i][:4].tolist()[1], 6),
-            "XMAX": round(result[i][:4].tolist()[2], 6),
-            "YMAX": round(result[i][:4].tolist()[3], 6),},
-        "CONF": round(float(result[i][4]), 2),
-        "FPS": round(1000 / float(results.t[1]), 2),
-        "IMG_WIDTH": img_size[0],
-        "IMG_HEIGHT": img_size[1],} for i in range(len(result))] for result in results.xyxyn]
+    results = model.predict(source=img_path, imgsz=infer_size, conf=conf, iou=iou, return_outputs=True)
+    det = []
+    for i in results:
+        det.append(deepcopy(i))
+
+    det_list = det[0]["det"].tolist()
+
+    return det_list
 
 
 # 标签和边界框颜色设置

requirements.txt

@@ -1,4 +1,4 @@
-ultralytics==8.0.3
+ultralytics
 opencv-python
 matplotlib
 numpy