- CUDA(11.3) installation：
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
sudo sh cuda_11.3.1_465.19.01_linux.run- Install Pytorch:
pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html- Install dependent environment:
pip install -r requirements.txt
pip install -v -e .
cd nuscenes-devkit-1.1.3/setup/
pip install -v -e .-
Prepare dataset
Download the dataset from https://pan.baidu.com/s/1NY0auRjm-NRrz-KtDOmdpA?pwd=m27v to the folder named data.
-
After downloading the dataset, generate a pkl file：
python tools/create_data_bevdet_v2u.py- Test Model：
python tools/test.py configs/UVCP/uvcpnet.py $checkpoint$ --eval map