ncnn

1
import sys
2
from setuptools import setup, find_packages
3

4
try:
5
    from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
6

7
    class bdist_wheel(_bdist_wheel):
8
        def finalize_options(self):
9
            _bdist_wheel.finalize_options(self)
10
            self.root_is_pure = False
11

12

13
except ImportError:
14
    bdist_wheel = None
15

16
if sys.version_info < (3, 0):
17
    sys.exit("Sorry, Python < 3.0 is not supported")
18

19
requirements = ["numpy", "tqdm", "requests", "portalocker", "opencv-python"]
20

21
setup(
22
    name="ncnn",
23
    version="${PACKAGE_VERSION}",
24
    author="nihui",
25
    author_email="nihuini@tencent.com",
26
    maintainer="caishanli",
27
    maintainer_email="caishanli25@gmail.com",
28
    description="ncnn is a high-performance neural network inference framework optimized for the mobile platform",
29
    url="https://github.com/Tencent/ncnn",
30
    classifiers=[
31
        "Programming Language :: C++",
32
        "Programming Language :: Python :: 3",
33
        "Programming Language :: Python :: 3.6",
34
        "Programming Language :: Python :: 3.7",
35
        "Programming Language :: Python :: 3.8",
36
        "Programming Language :: Python :: 3.9",
37
        "Programming Language :: Python :: 3.10",
38
        "License :: OSI Approved :: BSD License",
39
        "Operating System :: OS Independent",
40
    ],
41
    license="BSD-3",
42
    python_requires=">=3.5",
43
    packages=find_packages(),
44
    package_dir={"": "."},
45
    package_data={"ncnn": ["ncnn${PYTHON_MODULE_PREFIX}${PYTHON_MODULE_EXTENSION}"]},
46
    install_requires=requirements,
47
    cmdclass={"bdist_wheel": bdist_wheel},
48
)
49