wuwenhai
/
bev-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Minor] Fix some bugs.

This commit is contained in:
Haotian Tang 2022-06-09 15:29:52 +00:00
parent 86b077bea0
commit 6bce86a53c
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -19,6 +19,7 @@ def make_cuda_ext(
"-D__CUDA_NO_HALF_OPERATORS__", "-D__CUDA_NO_HALF_OPERATORS__",
"-D__CUDA_NO_HALF_CONVERSIONS__", "-D__CUDA_NO_HALF_CONVERSIONS__",
"-D__CUDA_NO_HALF2_OPERATORS__", "-D__CUDA_NO_HALF2_OPERATORS__",
"-gencode=arch=compute_70,code=sm_70",
"-gencode=arch=compute_75,code=sm_75", "-gencode=arch=compute_75,code=sm_75",
"-gencode=arch=compute_86,code=sm_86", "-gencode=arch=compute_86,code=sm_86",
] ]

4
tools/create_data.py
View File

@ -1,7 +1,7 @@
import argparse import argparse
from tools.data_converter import nuscenes_converter as nuscenes_converter from data_converter import nuscenes_converter as nuscenes_converter
from tools.data_converter.create_gt_database import create_groundtruth_database from data_converter.create_gt_database import create_groundtruth_database
def nuscenes_data_prep( def nuscenes_data_prep(