diff --git a/setup.py b/setup.py new file mode 100644 index 00000000..5c8f93f3 --- /dev/null +++ b/setup.py @@ -0,0 +1,31 @@ +from setuptools import setup, find_packages +import os + +# collect requirements from requirements.txt +lib_dir = os.path.dirname(os.path.realpath(__file__)) +requirements_path = lib_dir + "/requirements.txt" +install_requires = [] +if os.path.isfile(requirements_path): + with open(requirements_path) as f: + install_requires = f.read().splitlines() + +# remove comments +_install_requires = [] +for req in install_requires: + idx = req.find("#") + req = req[:idx] + if req: + _install_requires.append(req) +install_requires = _install_requires + +setup( + name='yolov6', + version='0.1.0', + packages=find_packages(), + url='https://github.com/meituan/YOLOv6', + license='GNU General Public License v3.0', + author='meituan', + author_email='', + description='', + install_requires=install_requires +) \ No newline at end of file diff --git a/yolov6/__init__.py b/yolov6/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/core/__init__.py b/yolov6/core/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/data/__init__.py b/yolov6/data/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/layers/__init__.py b/yolov6/layers/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/models/__init__.py b/yolov6/models/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/solver/__init__.py b/yolov6/solver/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/yolov6/utils/__init__.py b/yolov6/utils/__init__.py new file mode 100644 index 00000000..e69de29b