-
Notifications
You must be signed in to change notification settings - Fork 6
/
setup.py
37 lines (32 loc) · 1.05 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
34
35
36
37
import os
import sys
from setuptools import Extension
from setuptools import setup, find_packages
__version__ = '0.0.0'
py_dir = '/usr/local/lib/python{}/dist-packages/'.format(sys.version[:3])
# lidar BEV cpp extension
lidar_bev_ext_module = Extension(
name = 'lidardet.ops.lidar_bev.bev',
sources = ['lidardet/ops/lidar_bev/bev.cpp'],
include_dirs = ['/usr/include/eigen3', py_dir + 'pybind11/include'],
language='c++'
)
if __name__ == '__main__':
setup(
name='lidardet',
version="0.0.0",
description='LidarDet is a general codebase for learning based perception task',
install_requires=[
'numpy',
'torch>=1.1',
'numba',
'tensorboardX',
'easydict',
'pyyaml'
],
author='jiaolong',
author_email='jiaolongxu@gmail.com',
license='Apache License 2.0',
packages=find_packages(exclude=['tools', 'data', 'output', 'cache', 'ros', 'docker', 'config']),
ext_modules=[lidar_bev_ext_module]
)