v04 update

app.py

@@ -133,16 +133,22 @@ def yaml_csv(file_path, file_tag):
 
 
 # model loading
-def model_loading(model_name, device, opt=["refresh_yolov5"]):
-
-    # load model
-    model = torch.hub.load(
-        model_path, 
-        model_name, 
-        force_reload=[True if "refresh_yolov5" in opt else False][0], 
-        device=device, 
-        _verbose=False
-    )
+def model_loading(model_name, device, opt=[]):
+
+    # 加载本地模型
+    try:
+        # load model
+        model = torch.hub.load(
+            model_path, 
+            model_name, 
+            force_reload=[True if "refresh_yolov5" in opt else False][0], 
+            device=device, 
+            _verbose=False
+        )
+    except Exception as e:
+        print(e)
+    else:
+        print(f"🚀 welcome to {GYD_VERSION}，{model_name} loaded successfully!")
 
     return model
 
@@ -218,9 +224,15 @@ def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_
     if model_name_tmp != model_name:
         # Model judgment to avoid repeated loading
         model_name_tmp = model_name
+        print(f"Loading model {model_name_tmp}......")
         model = model_loading(model_name_tmp, device, opt)
     elif device_tmp != device:
+        # Device judgment to avoid repeated loading
         device_tmp = device
+        print(f"Loading model {model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+    else:
+        print(f"Loading model {model_name_tmp}......")
         model = model_loading(model_name_tmp, device, opt)
 
     # -------------Model tuning -------------
@@ -363,11 +375,17 @@ def yolo_det_video(video, device, model_name, infer_size, conf, iou, max_num, mo
     if model_name_tmp != model_name:
         # Model judgment to avoid repeated loading
         model_name_tmp = model_name
+        print(f"Loading model {model_name_tmp}......")
         model = model_loading(model_name_tmp, device, opt)
     elif device_tmp != device:
+        # Device judgment to avoid repeated loading
         device_tmp = device
+        print(f"Loading model {model_name_tmp}......")
         model = model_loading(model_name_tmp, device, opt)
-
+    else:
+        print(f"Loading model {model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+        
     # -------------Model tuning -------------
     model.conf = conf  # NMS confidence threshold
     model.iou = iou  # NMS IOU threshold