From d97689fbf325e3f4a9a87e5c703c0d330697a90c Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Wed, 19 Mar 2025 06:15:29 +0600 Subject: [PATCH] `ultralytics 8.3.93` Fix TorchScript Model with NMS loading error (#19747) Signed-off-by: Glenn Jocher Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- docs/en/models/yoloe.md | 2 +- ultralytics/__init__.py | 2 +- ultralytics/nn/autobackend.py | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/docs/en/models/yoloe.md b/docs/en/models/yoloe.md index 131d19b9e6..4ca8d463a9 100644 --- a/docs/en/models/yoloe.md +++ b/docs/en/models/yoloe.md @@ -10,7 +10,7 @@ keywords: YOLOE, open-vocabulary detection, real-time object detection, instance ![YOLOE Prompting Options](https://github.com/THU-MIG/yoloe/raw/main/figures/visualization.svg) -[**YOLOE** (You Only Look Once **E**nhanced)](https://arxiv.org/html/2503.07465v1) is a new advancement in zero-shot, promptable YOLO models, designed for **open-vocabulary** detection and segmentation. Unlike previous YOLO models limited to fixed categories, YOLOE uses text, image, or internal vocabulary prompts, enabling real-time detection of any object class. Built upon YOLOv10 and inspired by [YOLO-World](https://docs.ultralytics.com/models/yolo-world/), YOLOE achieves **state-of-the-art zero-shot performance** with minimal impact on speed and accuracy. +[YOLOE (Real-Time Seeing Anything)](https://arxiv.org/html/2503.07465v1) is a new advancement in zero-shot, promptable YOLO models, designed for **open-vocabulary** detection and segmentation. Unlike previous YOLO models limited to fixed categories, YOLOE uses text, image, or internal vocabulary prompts, enabling real-time detection of any object class. Built upon YOLOv10 and inspired by [YOLO-World](https://docs.ultralytics.com/models/yolo-world/), YOLOE achieves **state-of-the-art zero-shot performance** with minimal impact on speed and accuracy. !!! note "Ultralytics Integration Status 🚧" diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index b8e99fc180..d94da95f67 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license -__version__ = "8.3.92" +__version__ = "8.3.93" import os diff --git a/ultralytics/nn/autobackend.py b/ultralytics/nn/autobackend.py index a7bf7cf732..29b58d2541 100644 --- a/ultralytics/nn/autobackend.py +++ b/ultralytics/nn/autobackend.py @@ -185,6 +185,8 @@ class AutoBackend(nn.Module): # TorchScript elif jit: + import torchvision # noqa - https://github.com/ultralytics/ultralytics/pull/19747 + LOGGER.info(f"Loading {w} for TorchScript inference...") extra_files = {"config.txt": ""} # model metadata model = torch.jit.load(w, _extra_files=extra_files, map_location=device)