-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.py
81 lines (63 loc) · 3.36 KB
/
setup.py
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
"""
Copyright (C) 2022 Fraunhofer Institute for Microelectronic Circuits and Systems.
All rights reserved.
AIfES-Converter is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
from setuptools import setup, find_packages
import pathlib
here = pathlib.Path(__file__).parent.resolve()
long_description = (here / "README.md").read_text(encoding="utf-8")
setup(
name="AIfES-Converter",
version="1.0.0",
description="This is a convert tool to create AIfES models for direct use in the Arduino IDE or other IDEs",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://aifes.ai/",
author="Fraunhofer IMS",
author_email="[email protected]",
classifiers=[
"Development Status :: 5 - Production/Stable",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Code Generators",
"Topic :: Software Development :: Embedded Systems",
"License :: OSI Approved :: GNU Affero General Public License v3",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3 :: Only",
],
package_dir={"": "src"},
packages=find_packages(where="src"),
python_requires=">=3.7, <3.10",
install_requires=[
"numpy>=1.19",
"packaging"
],
extras_require={
"tensorflow": ["tensorflow>=2.4"],
"pytorch": ["torch>=1.8"],
"both": ["tensorflow>=2.4", "torch>=1.8"]
},
include_package_data=True,
# data_files=[('./src/aifes/aifes_code_generator/templates/aifes', ['src/aifes/aifes_code_generator/templates/aifes/aifes_f32_fnn.h', 'src/aifes/aifes_code_generator/templates/aifes/aifes_f32_weights.h', 'src/aifes/aifes_code_generator/templates/aifes/aifes_q7_fnn.h', 'src/aifes/aifes_code_generator/templates/aifes/aifes_q7_weights.h']),
# ('./src/aifes/aifes_code_generator/templates/aifes_express',['src/aifes/aifes_code_generator/templates/aifes_express/aifes_e_f32_fnn.h', 'src/aifes/aifes_code_generator/templates/aifes_express/aifes_e_f32_weights.h','src/aifes/aifes_code_generator/templates/aifes_express/aifes_e_q7_fnn.h', 'src/aifes/aifes_code_generator/templates/aifes_express/aifes_e_q7_weights.h'])],
project_urls={
"Source": "https://github.com/Fraunhofer-IMS/AIfES-Converter/",
"Bug Reports": "https://github.com/Fraunhofer-IMS/AIfES-Converter/issues",
"Documentation": "https://fraunhofer-ims.github.io/AIfES-Converter/#index"
},
)