xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: update pyproject.toml and project version info

Browse Source
This commit is contained in:
ckmessi@163.com 2024-05-27 17:34:30 +08:00
parent 9f73bc7768
commit 052ad58a0e
3 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pyproject.toml
View File

@ -1,4 +1,4 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # THU-MIG YOLO v10 🚀, AGPL-3.0 license
# Overview: # Overview:
# This pyproject.toml file manages the build, packaging, and distribution of the Ultralytics library. # 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 settings -----------------------------------------------------------------------------------------------------
[project] [project]
name = "ultralytics" name = "yolov10"
dynamic = ["version"] 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" readme = "README.md"
requires-python = ">=3.8" requires-python = ">=3.9"
license = { "text" = "AGPL-3.0" } license = { "text" = "AGPL-3.0" }
keywords = ["machine-learning", "deep-learning", "computer-vision", "ML", "DL", "AI", "YOLO", "YOLOv3", "YOLOv5", "YOLOv8", "HUB", "Ultralytics"] keywords = ["machine-learning", "deep-learning", "computer-vision", "ML", "DL", "AI", "YOLO", "YOLOv3", "YOLOv5", "YOLOv8", "HUB", "Ultralytics"]
authors = [ authors = [
{ name = "Glenn Jocher" }, { name = "Ao Wang" },
{ name = "Ayush Chaurasia" },
{ name = "Jing Qiu" }
] ]
maintainers = [ maintainers = [
{ name = "Glenn Jocher" }, { name = "Ao Wang" },
{ name = "Ayush Chaurasia" },
{ name = "Jing Qiu" }
] ]
classifiers = [ classifiers = [
"Development Status :: 4 - Beta", "Development Status :: 4 - Beta",
@ -124,9 +120,9 @@ extra = [
] ]
[project.urls] [project.urls]
"Bug Reports" = "https://github.com/ultralytics/ultralytics/issues" "Bug Reports" = "https://github.com/THU-MIG/yolov10/issues"
"Funding" = "https://ultralytics.com" # "Funding" = "https://ultralytics.com"
"Source" = "https://github.com/ultralytics/ultralytics/" "Source" = "https://github.com/THU-MIG/yolov10/"
[project.scripts] [project.scripts]
yolo = "ultralytics.cfg:entrypoint" yolo = "ultralytics.cfg:entrypoint"

2
ultralytics/__init__.py
View File

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.1.34" __version__ = "10.1.1"
from ultralytics.data.explorer.explorer import Explorer from ultralytics.data.explorer.explorer import Explorer
from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld, YOLOv10 from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld, YOLOv10

2
ultralytics/utils/torch_utils.py
View File

@ -104,7 +104,7 @@ def select_device(device="", batch=0, newline=False, verbose=True):
if isinstance(device, torch.device): if isinstance(device, torch.device):
return 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() device = str(device).lower()
for remove in "cuda:", "none", "(", ")", "[", "]", "'", " ": for remove in "cuda:", "none", "(", ")", "[", "]", "'", " ":
device = device.replace(remove, "") # to string, 'cuda:0' -> '0' and '(0, 1)' -> '0,1' device = device.replace(remove, "") # to string, 'cuda:0' -> '0' and '(0, 1)' -> '0,1'