From e096edcd114edf3862bf337567694fea48ac84a6 Mon Sep 17 00:00:00 2001 From: Ayush Chaurasia Date: Mon, 29 Jan 2024 04:59:59 +0530 Subject: [PATCH] `ultralytics 8.1.7` `USER_CONFIG_DIR` Explorer ops (#7861) Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/data/explorer/explorer.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 464ebd10..f7b77615 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.1.6" +__version__ = "8.1.7" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO diff --git a/ultralytics/data/explorer/explorer.py b/ultralytics/data/explorer/explorer.py index b77b8414..6dba618d 100644 --- a/ultralytics/data/explorer/explorer.py +++ b/ultralytics/data/explorer/explorer.py @@ -16,7 +16,7 @@ from ultralytics.data.augment import Format from ultralytics.data.dataset import YOLODataset from ultralytics.data.utils import check_det_dataset from ultralytics.models.yolo.model import YOLO -from ultralytics.utils import LOGGER, IterableSimpleNamespace, checks +from ultralytics.utils import LOGGER, IterableSimpleNamespace, checks, USER_CONFIG_DIR from .utils import get_sim_index_schema, get_table_schema, plot_query_result, prompt_sql_query, sanitize_batch @@ -54,7 +54,10 @@ class ExplorerDataset(YOLODataset): class Explorer: def __init__( - self, data: Union[str, Path] = "coco128.yaml", model: str = "yolov8n.pt", uri: str = "~/ultralytics/explorer" + self, + data: Union[str, Path] = "coco128.yaml", + model: str = "yolov8n.pt", + uri: str = USER_CONFIG_DIR / "explorer", ) -> None: checks.check_requirements(["lancedb>=0.4.3", "duckdb"]) import lancedb