- Linux (tested on Ubuntu 18.04)
- Python 3.7
- Anaconda
- PyTorch
- CUDA 11.7
in the directory of ROBOSAC:
cd coperception
conda env create -f environment.yml
conda activate coperceptionconda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidiaThis installs and links coperception library to code in ./coperception directory.
pip install -e .Please download and unzip the parsed detection dataset of V2X-Sim 2.0.
Link the test split of V2X-Sim dataset in the default value of argument "data"
/{Your_location}/V2X-Sim-det/testin the test folder data are structured like:
test
├──agent_0
├──agent_1
├──agent_2
├──agent_3
├──agent_4
├──agent_5
├──19_0
├──0.npy
...
Link the checkpoint location in the default value of argument "resume"
Please download pre-trained weights and save them in ROBOSAC/coperception/ckpt/meanfusion folder.
epoch_49.pth is the original victim model without adversarial training.
epoch_advtrain_49.pth is the PGD-trained model.