ru/models/yoloe/

[giscus[bot]]

ru/models/yoloe/

YOLOE - это модель обнаружения и сегментации с открытым словарем в режиме реального времени, которая расширяет YOLO с помощью текстовых, графических или внутренних словарных подсказок, позволяя обнаруживать любые классы объектов с самой современной производительностью без выстрелов.

https://docs.ultralytics.com/ru/models/yoloe/

[Mariegri]

Hello there!

Trying to run the model, I faced the following error:
"NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'torchvision::nms' is only available for these backends: [CPU, Meta, QuantizedCPU, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMTIA, AutogradMeta, Tracer, AutocastCPU, AutocastMTIA, AutocastXPU, AutocastMPS, AutocastCUDA, FuncTorchBatched, BatchedNestedTensor, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PreDispatch, PythonDispatcher]".

Even if I switch device to cpu (results = model.predict(path, device = "cpu", conf = 0.1, iou = 0.7)) it returns
"Ultralytics 8.3.148 Python-3.12.7 torch-2.7.0+cu128 CUDA:0 (NVIDIA GeForce GTX 1650 Ti, 4096MiB)
YOLOe-11l-seg summary (fused): 227 layers, 35,117,862 parameters, 2,254,374 gradients, 144.1 GFLOPs"
and then the same error.

params:
windows 11
python 3.12.7 | packaged by Anaconda, Inc. | (main, Oct 4 2024, 13:17:27) [MSC v.1929 64 bit (AMD64)]
cuda 12.8
cuddn 9.10
torch==2.7.0+cu128 torchvision==0.22.0 torchaudio==2.7.0+cu128 --index-url https://download.pytorch.org/whl/cu128
tensorboard==2.19.0 tensorboard-data-server== 0.7.2 tensorflow==2.19.0

What can be the problem?
Thanks a lot in advance!

[UltralyticsAssistant]

👋 Hello, thank you for your interest in Ultralytics 🚀! We recommend checking out the Docs for extensive information—including many Python and CLI usage examples—which might already answer many common questions.

If this is a 🐛 Bug Report, please provide a minimum reproducible example to help us debug the issue.

If this is a custom training ❓ Question, please include as much detail as possible, such as dataset image examples and training logs, and make sure you’re following our Tips for Best Training Results.

Join the Ultralytics community wherever you prefer: for real-time chat, visit Discord 🎧. For in-depth discussions, check out Discourse. Or share knowledge on our Subreddit.

Upgrade

Upgrade to the latest ultralytics package (and all requirements) in a Python>=3.8 environment with PyTorch>=1.8 to be sure your issue isn’t already resolved in the latest version:

pip install -U ultralytics

Environments

YOLO can be run in these up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python, and PyTorch preinstalled):

• Notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all Ultralytics CI tests are currently passing. CI tests verify correct operation of all YOLO Modes and Tasks on macOS, Windows, and Ubuntu every 24 hours and on every commit.

---

This is an automated response 🤖. An Ultralytics engineer will review your discussion and assist you soon!