setup.py
2.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import io
import os
import re
import setuptools
def get_long_description():
base_dir = os.path.abspath(os.path.dirname(__file__))
with io.open(os.path.join(base_dir, "README.md"), encoding="utf-8") as f:
return f.read()
def get_requirements():
with open("requirements.txt") as f:
return f.read().splitlines()
def get_version():
current_dir = os.path.abspath(os.path.dirname(__file__))
version_file = os.path.join(current_dir, "yolov5", "__init__.py")
with io.open(version_file, encoding="utf-8") as f:
return re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]', f.read(), re.M).group(1)
setuptools.setup(
name="yolov5",
version=get_version(),
author="",
license="GPL",
description="Packaged version of the Yolov5 object detector",
long_description=get_long_description(),
long_description_content_type="text/markdown",
url="https://github.com/fcakyon/yolov5-pip",
packages=setuptools.find_packages(exclude=["tests"]),
python_requires=">=3.6",
install_requires=get_requirements(),
extras_require={"tests": ["pytest"]},
include_package_data=True,
options={'bdist_wheel':{'python_tag':'py36.py37.py38'}},
classifiers=[
"Development Status :: 5 - Production/Stable",
"License :: OSI Approved :: GNU General Public License (GPL)",
"Operating System :: OS Independent",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules",
"Topic :: Education",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Image Recognition",
],
keywords="machine-learning, deep-learning, ml, pytorch, YOLO, object-detection, vision, YOLOv3, YOLOv4, YOLOv5",
entry_points={'console_scripts': [
'yolo_train=yolov5.train:main',
'yolo_test=yolov5.test:main',
'yolo_detect=yolov5.detect:main',
'yolo_export=yolov5.models.export:main'
],
}
)