You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
2.5 KiB
54 lines
2.5 KiB
# Ultralytics YOLO 🚀, GPL-3.0 license |
|
|
|
import re |
|
from pathlib import Path |
|
|
|
import pkg_resources as pkg |
|
from setuptools import find_packages, setup |
|
|
|
# Settings |
|
FILE = Path(__file__).resolve() |
|
PARENT = FILE.parent # root directory |
|
README = (PARENT / "README.md").read_text(encoding="utf-8") |
|
REQUIREMENTS = [f'{x.name}{x.specifier}' for x in pkg.parse_requirements((PARENT / 'requirements.txt').read_text())] |
|
PKG_REQUIREMENTS = ['sentry_sdk'] # pip-only requirements |
|
|
|
|
|
def get_version(): |
|
file = PARENT / 'ultralytics/__init__.py' |
|
return re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]', file.read_text(encoding="utf-8"), re.M)[1] |
|
|
|
|
|
setup( |
|
name="ultralytics", # name of pypi package |
|
version=get_version(), # version of pypi package |
|
python_requires=">=3.7,<=3.11", |
|
license='GPL-3.0', |
|
description='Ultralytics YOLOv8', |
|
long_description=README, |
|
long_description_content_type="text/markdown", |
|
url="https://github.com/ultralytics/ultralytics", |
|
project_urls={ |
|
'Bug Reports': 'https://github.com/ultralytics/ultralytics/issues', |
|
'Funding': 'https://ultralytics.com', |
|
'Source': 'https://github.com/ultralytics/ultralytics'}, |
|
author="Ultralytics", |
|
author_email='hello@ultralytics.com', |
|
packages=find_packages(), # required |
|
include_package_data=True, |
|
install_requires=REQUIREMENTS + PKG_REQUIREMENTS, |
|
extras_require={ |
|
'dev': |
|
['check-manifest', 'pytest', 'pytest-cov', 'coverage', 'mkdocs', 'mkdocstrings[python]', 'mkdocs-material']}, |
|
classifiers=[ |
|
"Intended Audience :: Developers", "Intended Audience :: Science/Research", |
|
"License :: OSI Approved :: GNU General Public License v3 (GPLv3)", "Programming Language :: Python :: 3", |
|
"Programming Language :: Python :: 3.7", "Programming Language :: Python :: 3.8", |
|
"Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", |
|
"Topic :: Software Development", "Topic :: Scientific/Engineering", |
|
"Topic :: Scientific/Engineering :: Artificial Intelligence", |
|
"Topic :: Scientific/Engineering :: Image Recognition", "Operating System :: POSIX :: Linux", |
|
"Operating System :: MacOS", "Operating System :: Microsoft :: Windows"], |
|
keywords="machine-learning, deep-learning, vision, ML, DL, AI, YOLO, YOLOv3, YOLOv5, YOLOv8, HUB, Ultralytics", |
|
entry_points={ |
|
'console_scripts': ['yolo = ultralytics.yolo.cfg:entrypoint', 'ultralytics = ultralytics.yolo.cfg:entrypoint']})
|
|
|