mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 21:36:30 +00:00
Alex Bezdieniezhnykh
c0f8dd792d
fix same files problem in python different libs correct command logging in command handler
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from setuptools import setup, Extension
|
|
from Cython.Build import cythonize
|
|
import numpy as np
|
|
|
|
extensions = [
|
|
Extension('constants_inf', ['constants_inf.pyx']),
|
|
Extension('file_data', ['file_data.pyx']),
|
|
Extension('remote_command_inf', ['remote_command_inf.pyx']),
|
|
Extension('remote_command_handler_inf', ['remote_command_handler_inf.pyx']),
|
|
Extension('annotation', ['annotation.pyx']),
|
|
Extension('loader_client', ['loader_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_inference', ['main_inference.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
|
|
) |