-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
45 lines (43 loc) · 1.4 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
38
39
40
41
42
43
44
45
from setuptools import find_packages, setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
# pip install -e . --user
# using setup.py develop somehow fails at "open3d==0.14.1" saying it can't find module
setup(name='vcn',
version='1.0',
description='Complete the surface of incomplete lidar point clouds',
author='Darren Tsai',
author_email='d.tsai@acfr.usyd.edu.au',
license='MIT',
packages=find_packages(exclude=['scripts','experiments']),
cmdclass={'build_ext': BuildExtension},
install_requires=[
'argparse',
'easydict',
'h5py',
'matplotlib',
'numpy',
'opencv-python',
'pyyaml',
'scipy',
'tensorboardX',
'timm==0.4.5 ',
'tqdm>=4.51.0',
'open3d==0.14.1',
'transforms3d'
],
ext_modules=[
CUDAExtension(
name='chamfer',
sources=[
'extensions/chamfer_dist/chamfer_cuda.cpp',
'extensions/chamfer_dist/chamfer.cu',
]),
CUDAExtension(
name='iou3d_nms_cuda',
sources=[
'extensions/iou3d_nms/src/iou3d_cpu.cpp',
'extensions/iou3d_nms/src/iou3d_nms_api.cpp',
'extensions/iou3d_nms/src/iou3d_nms.cpp',
'extensions/iou3d_nms/src/iou3d_nms_kernel.cu',
]),
])