From a43a141fc261c7f624580c977e536ed52a2ba2ee Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Tue, 10 Sep 2024 05:52:57 +0800 Subject: [PATCH] `ultralytics 8.2.91` fix `v10DetectLoss` module rename for YOLOv10 (#16148) Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/nn/tasks.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index b45d97ce59..5db8585676 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.90" +__version__ = "8.2.91" import os diff --git a/ultralytics/nn/tasks.py b/ultralytics/nn/tasks.py index 07a040d90d..46974620a6 100644 --- a/ultralytics/nn/tasks.py +++ b/ultralytics/nn/tasks.py @@ -776,6 +776,7 @@ def torch_safe_load(weight): attributes={ "ultralytics.nn.modules.block.Silence": "torch.nn.Identity", # YOLOv9e "ultralytics.nn.tasks.YOLOv10DetectionModel": "ultralytics.nn.tasks.DetectionModel", # YOLOv10 + "ultralytics.utils.loss.v10DetectLoss": "ultralytics.utils.loss.E2EDetectLoss", # YOLOv10 }, ): ckpt = torch.load(file, map_location="cpu")