[Minor] Fix dependencies.

This commit is contained in:
Haotian Tang 2022-06-16 23:23:42 +00:00
parent d16e937596
commit 24502fe097
2 changed files with 3 additions and 2 deletions

View File

@ -58,6 +58,7 @@ The code is built with following libraries:
- [torchpack](https://github.com/mit-han-lab/torchpack)
- [mmcv](https://github.com/open-mmlab/mmcv) = 1.4.0
- [mmdetection](http://github.com/open-mmlab/mmdetection) = 2.20.0
- [nuscenes-dev-kit](https://github.com/nutonomy/nuscenes-devkit)
After installing these dependencies, please run this command to install the codebase:

View File

@ -2,11 +2,11 @@ import warnings
import numba
import numpy as np
from numba.errors import NumbaPerformanceWarning
from numba import errors
from mmdet3d.core.bbox import box_np_ops
warnings.filterwarnings("ignore", category=NumbaPerformanceWarning)
warnings.filterwarnings("ignore", category=errors.NumbaPerformanceWarning)
@numba.njit