mirror of
https://github.com/ultralytics/ultralytics
synced 2026-05-24 09:38:39 +00:00
ultralytics 8.3.93 Fix TorchScript Model with NMS loading error (#19747)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
a85185b0ae
commit
d97689fbf3
3 changed files with 4 additions and 2 deletions
|
|
@ -10,7 +10,7 @@ keywords: YOLOE, open-vocabulary detection, real-time object detection, instance
|
|||
|
||||

|
||||
|
||||
[**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 🚧"
|
||||
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
||||
|
||||
__version__ = "8.3.92"
|
||||
__version__ = "8.3.93"
|
||||
|
||||
import os
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in a new issue