deep_learning/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics.git synced 2025-09-15 15:48:41 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Revert "Re-enable TensorRT export in GPU tests" (#22078)

Browse Source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-09-15 00:21:11 +02:00 committed by GitHub
parent a7a56595f8
commit 1aa3688613
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/en/datasets/explorer/api.md
View File

@ -150,9 +150,10 @@ Note: This works using LLMs under the hood so the results are probabilistic and
!!! example "Ask AI" !!! example "Ask AI"
```python ```python
from ultralytics import Explorer
from ultralytics.data.explorer import plot_query_result from ultralytics.data.explorer import plot_query_result
from ultralytics import Explorer
# create an Explorer object # create an Explorer object
exp = Explorer(data="coco128.yaml", model="yolo11n.pt") exp = Explorer(data="coco128.yaml", model="yolo11n.pt")
exp.create_embeddings_table() exp.create_embeddings_table()

1
docs/en/datasets/explorer/explorer.md
View File

@ -116,7 +116,6 @@ Image.fromarray(plt)
```python ```python
# plot # plot
from PIL import Image from PIL import Image
from ultralytics.data.explorer import plot_query_result from ultralytics.data.explorer import plot_query_result
plt = plot_query_result(exp.ask_ai("show me 10 images containing exactly 2 persons")) plt = plot_query_result(exp.ask_ai("show me 10 images containing exactly 2 persons"))

3
tests/test_cuda.py
View File

@ -68,7 +68,7 @@ def test_export_onnx_matrix(task, dynamic, int8, half, batch, simplify, nms):
half=half, half=half,
batch=batch, batch=batch,
simplify=simplify, simplify=simplify,
nms=nms, nms=nms and task != "obb", # WARNING: Failing NMS with OBB
device=DEVICES[0], device=DEVICES[0],
) )
YOLO(file)([SOURCE] * batch, imgsz=64 if dynamic else 32, device=DEVICES[0]) # exported model inference YOLO(file)([SOURCE] * batch, imgsz=64 if dynamic else 32, device=DEVICES[0]) # exported model inference
@ -76,6 +76,7 @@ def test_export_onnx_matrix(task, dynamic, int8, half, batch, simplify, nms):
@pytest.mark.slow @pytest.mark.slow
@pytest.mark.skipif(True, reason="WARNING: Failing TensorRT export tests")
@pytest.mark.skipif(not DEVICES, reason="No CUDA devices available") @pytest.mark.skipif(not DEVICES, reason="No CUDA devices available")
@pytest.mark.parametrize( @pytest.mark.parametrize(
"task, dynamic, int8, half, batch", "task, dynamic, int8, half, batch",