From 052ad58a0eb70894af2cb0b77094a806feaa20df Mon Sep 17 00:00:00 2001 From: "ckmessi@163.com" Date: Mon, 27 May 2024 17:34:30 +0800 Subject: [PATCH] chore: update pyproject.toml and project version info --- pyproject.toml | 22 +++++++++------------- ultralytics/__init__.py | 2 +- ultralytics/utils/torch_utils.py | 2 +- 3 files changed, 11 insertions(+), 15 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index d42c3805..813e10c2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,4 +1,4 @@ -# Ultralytics YOLO 🚀, AGPL-3.0 license +# THU-MIG YOLO v10 🚀, AGPL-3.0 license # Overview: # This pyproject.toml file manages the build, packaging, and distribution of the Ultralytics library. @@ -24,22 +24,18 @@ build-backend = "setuptools.build_meta" # Project settings ----------------------------------------------------------------------------------------------------- [project] -name = "ultralytics" +name = "yolov10" dynamic = ["version"] -description = "Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification." +description = "THU-MIG YOLOv10 for SOTA Real-Time End-to-End Object Detection, based on Ultralytics YOLOv8." readme = "README.md" -requires-python = ">=3.8" +requires-python = ">=3.9" license = { "text" = "AGPL-3.0" } keywords = ["machine-learning", "deep-learning", "computer-vision", "ML", "DL", "AI", "YOLO", "YOLOv3", "YOLOv5", "YOLOv8", "HUB", "Ultralytics"] authors = [ - { name = "Glenn Jocher" }, - { name = "Ayush Chaurasia" }, - { name = "Jing Qiu" } + { name = "Ao Wang" }, ] maintainers = [ - { name = "Glenn Jocher" }, - { name = "Ayush Chaurasia" }, - { name = "Jing Qiu" } + { name = "Ao Wang" }, ] classifiers = [ "Development Status :: 4 - Beta", @@ -124,9 +120,9 @@ extra = [ ] [project.urls] -"Bug Reports" = "https://github.com/ultralytics/ultralytics/issues" -"Funding" = "https://ultralytics.com" -"Source" = "https://github.com/ultralytics/ultralytics/" +"Bug Reports" = "https://github.com/THU-MIG/yolov10/issues" +# "Funding" = "https://ultralytics.com" +"Source" = "https://github.com/THU-MIG/yolov10/" [project.scripts] yolo = "ultralytics.cfg:entrypoint" diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 8ff1b4fb..1cd5eb5d 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.1.34" +__version__ = "10.1.1" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld, YOLOv10 diff --git a/ultralytics/utils/torch_utils.py b/ultralytics/utils/torch_utils.py index d476e1f8..90a792a9 100644 --- a/ultralytics/utils/torch_utils.py +++ b/ultralytics/utils/torch_utils.py @@ -104,7 +104,7 @@ def select_device(device="", batch=0, newline=False, verbose=True): if isinstance(device, torch.device): return device - s = f"Ultralytics YOLOv{__version__} 🚀 Python-{PYTHON_VERSION} torch-{torch.__version__} " + s = f"THU-MIG YOLOv{__version__} 🚀 Python-{PYTHON_VERSION} torch-{torch.__version__} " device = str(device).lower() for remove in "cuda:", "none", "(", ")", "[", "]", "'", " ": device = device.replace(remove, "") # to string, 'cuda:0' -> '0' and '(0, 1)' -> '0,1'