56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
from setuptools import find_packages, setup
|
|
import os
|
|
from glob import glob
|
|
|
|
package_name = 'camera_kl520'
|
|
|
|
models_paths = []
|
|
directories= glob('models/')+glob('models/*/')+glob('models/*/*/')
|
|
for directory in directories:
|
|
models_paths.append((os.path.join('share',package_name,directory),glob(f'{directory}/*.*')))
|
|
|
|
setup(
|
|
name=package_name,
|
|
version='0.0.0',
|
|
packages=find_packages(exclude=['test']),
|
|
data_files=[
|
|
('share/ament_index/resource_index/packages',
|
|
['resource/' + package_name]),
|
|
('share/' + package_name, ['package.xml']),
|
|
# get all the utils python files
|
|
(os.path.join('share', package_name, 'utils'),glob('utils/*.py')),
|
|
# get the firmware image and models
|
|
(os.path.join('share', package_name, 'res','firmware','KL520'),glob('res/firmware/KL520/*.bin')),
|
|
(os.path.join('share', package_name, 'res','images'),glob('res/images/*.*')),
|
|
(os.path.join('share', package_name, 'res','models','KL520', 'yolov5-noupsample_w640h640_kn-model-zoo'),glob('res/models/KL520/yolov5-noupsample_w640h640_kn-model-zoo/*.nef')),
|
|
]+models_paths,
|
|
|
|
install_requires=[
|
|
'setuptools',
|
|
'torch',
|
|
'torchvision',
|
|
],
|
|
zip_safe=True,
|
|
maintainer='jack',
|
|
maintainer_email='jack23162329@gmail.com',
|
|
description='TODO: Package description',
|
|
license='TODO: License declaration',
|
|
# tests_require=['pytest'],
|
|
entry_points={
|
|
'console_scripts': [
|
|
"detection_pub = camera_kl520.object_detection_pub:main",
|
|
"detection_sub = camera_kl520.object_detection_sub:main",
|
|
'yolo_inference_pub = camera_kl520.camera_yolo5l_pub:main',
|
|
'yolo_inference_sub = camera_kl520.camera_yolo5l_sub:main',
|
|
"testing_camera = camera_kl520.testing_camera:main",
|
|
"scan_kl520 = camera_kl520.Scan_kl520:main",
|
|
"detection_kl520_pub = camera_kl520.object_detection_kl520_pub:main",
|
|
"detection_kl520_sub = camera_kl520.object_detection_kl520_sub:main",
|
|
"yolo5s_kl520_pub = camera_kl520.camera_yolo5v_kl520_pub:main",
|
|
"yolo5s_kl520_sub = camera_kl520.camera_yolo5v_kl520_sub:main",
|
|
"yolo5s_kl720_pub = camera_kl520.camera_yolo5v_kl720_pub:main",
|
|
"yolo5s_kl720_sub = camera_kl520.camera_yolo5v_kl720_sub:main",
|
|
],
|
|
},
|
|
)
|