forked from TakanoTaiga/ros2ai
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
33 lines (31 loc) · 1.21 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
from setuptools import find_packages, setup
package_name = 'ros2ai'
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']),
],
install_requires=['setuptools'],
zip_safe=True,
maintainer='taiga',
maintainer_email='[email protected]',
description='TODO: Package description',
license='TODO: License declaration',
tests_require=['pytest'],
entry_points={
'console_scripts': [
'blip_image_captioning_base = ros2ai.blip_image_captioning_base:main',
'blip_image_captioning_large = ros2ai.blip_image_captioning_large:main',
'pix2struct_textcaps_base = ros2ai.pix2struct_textcaps_base:main',
'vit_gpt2_image_captioning = ros2ai.vit_gpt2_image_captioning:main',
'resnet_50 = ros2ai.resnet_50:main',
'vilt_b32_finetuned_vqa = ros2ai.vilt_b32_finetuned_vqa:main',
'owlvit_base_patch16 = ros2ai.owlvit_base_patch16:main',
'owlvit_base_patch32 = ros2ai.owlvit_base_patch32:main',
],
},
)