69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
import os
|
|
from glob import glob
|
|
from setuptools import find_packages, setup
|
|
|
|
package_name = 'vision_detect'
|
|
|
|
openvino_files = []
|
|
for model_dir in glob("checkpoints/*_openvino_model"):
|
|
model_name = os.path.basename(model_dir)
|
|
model_files = glob(os.path.join(model_dir, '*'))
|
|
openvino_files.append(
|
|
('share/' + package_name + '/checkpoints/' + model_name, model_files)
|
|
)
|
|
|
|
data_files = [
|
|
('share/ament_index/resource_index/packages', ['resource/' + package_name]),
|
|
('share/' + package_name, ['package.xml']),
|
|
('share/' + package_name + '/launch', glob('launch/*.launch.py')),
|
|
('share/' + package_name + '/configs', glob('configs/*.json')),
|
|
('share/' + package_name + '/configs/flexiv_configs',
|
|
glob('configs/flexiv_configs/*.json')),
|
|
('share/' + package_name + '/configs/hand_eye_mat', glob('configs/hand_eye_mat/*.json')),
|
|
('share/' + package_name + '/configs/launch_configs',
|
|
glob('configs/launch_configs/*.json')),
|
|
('share/' + package_name + '/configs/error_configs', glob('configs/error_configs/*.json')),
|
|
|
|
('share/' + package_name + '/checkpoints', glob('checkpoints/*.pt')),
|
|
('share/' + package_name + '/checkpoints', glob('checkpoints/*.onnx')),
|
|
('share/' + package_name + '/checkpoints', glob('checkpoints/*.engine')),
|
|
('share/' + package_name + '/pointclouds', glob('pointclouds/*.pcd')),
|
|
|
|
('lib/python3.10/site-packages/' + package_name + '/VisionDetect/net/pointnet2/pointnet2',
|
|
glob('vision_detect/VisionDetect/net/pointnet2/pointnet2/*.so')),
|
|
('lib/python3.10/site-packages/' + package_name + '/VisionDetect/net/knn/knn_pytorch',
|
|
glob('vision_detect/VisionDetect/net/knn/knn_pytorch/*.so'))
|
|
]
|
|
|
|
data_files.extend(openvino_files)
|
|
|
|
setup(
|
|
name=package_name,
|
|
version='0.0.0',
|
|
packages=find_packages(exclude=['test']),
|
|
data_files=data_files,
|
|
install_requires=['setuptools'],
|
|
zip_safe=True,
|
|
include_package_data=True,
|
|
maintainer='lyx',
|
|
maintainer_email='lyx@todo.todo',
|
|
description='TODO: Package description',
|
|
license='TODO: License declaration',
|
|
tests_require=['pytest'],
|
|
entry_points={
|
|
'console_scripts': [
|
|
# 'red_detect_topic_node = vision_detect.detect_red_topic:main',
|
|
# 'red_detect_service_node = vision_detect.detect_red_service:main',
|
|
|
|
'flexiv_detect_topic_node = vision_detect.flexivaidk_detect_topic:main',
|
|
'flexiv_detect_service_node = vision_detect.flexivaidk_detect_service:main',
|
|
|
|
'sub_pose_node = vision_detect.sub_pose:main',
|
|
'service_client_node = vision_detect.service_client:main',
|
|
'get_camera_pose_node = vision_detect.get_camera_pose:main',
|
|
|
|
'detect_node = vision_detect.detect_node:main',
|
|
],
|
|
},
|
|
)
|