mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 21:36:30 +00:00
Alex Bezdieniezhnykh
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from setuptools import setup, Extension
|
|
from Cython.Build import cythonize
|
|
import numpy as np
|
|
|
|
extensions = [
|
|
Extension('constants', ['constants.pyx']),
|
|
Extension('annotation', ['annotation.pyx']),
|
|
Extension('credentials', ['credentials.pyx']),
|
|
Extension('file_data', ['file_data.pyx']),
|
|
Extension('hardware_service', ['hardware_service.pyx'], extra_compile_args=["-g"], extra_link_args=["-g"]),
|
|
Extension('security', ['security.pyx']),
|
|
Extension('remote_command', ['remote_command.pyx']),
|
|
Extension('remote_command_handler', ['remote_command_handler.pyx']),
|
|
Extension('user', ['user.pyx']),
|
|
Extension('cdn_manager', ['cdn_manager.pyx']),
|
|
Extension('api_client', ['api_client.pyx']),
|
|
Extension('ai_config', ['ai_config.pyx']),
|
|
Extension('tensorrt_engine', ['tensorrt_engine.pyx'], include_dirs=[np.get_include()]),
|
|
Extension('onnx_engine', ['onnx_engine.pyx'], include_dirs=[np.get_include()]),
|
|
Extension('inference_engine', ['inference_engine.pyx'], include_dirs=[np.get_include()]),
|
|
Extension('inference', ['inference.pyx'], include_dirs=[np.get_include()]),
|
|
Extension('main', ['main.pyx']),
|
|
]
|
|
|
|
setup(
|
|
name="azaion.ai",
|
|
ext_modules=cythonize(
|
|
extensions,
|
|
compiler_directives={
|
|
"language_level": 3,
|
|
"emit_code_comments" : False,
|
|
"binding": True,
|
|
'boundscheck': False,
|
|
'wraparound': False
|
|
}
|
|
),
|
|
install_requires=[
|
|
'ultralytics>=8.0.0',
|
|
'pywin32; platform_system=="Windows"'
|
|
],
|
|
zip_safe=False
|
|
) |