增加了单帧入侵判断及yoloV10

2025-04-11 17:02:39 +08:00
parent 798c596acc
commit e044c85a04
197 changed files with 1863 additions and 997 deletions
--- a/ultralytics/nn/tasks.py
+++ b/ultralytics/nn/tasks.py
@ -730,7 +730,12 @@ def torch_safe_load(weight):
                "ultralytics.yolo.data": "ultralytics.data",
            }
        ):  # for legacy 8.0 Classify and Pose models
-            ckpt = torch.load(file, map_location="cpu")
+            if torch.__version__ >= '2.6':
+                ckpt = torch.load(file, map_location="cpu", weights_only=False)
+            else:
+                ckpt = torch.load(file, map_location="cpu")
+                
+            

    except ModuleNotFoundError as e:  # e.name is missing module name
        if e.name == "models":