CP-Freezer

March 3, 2025 ยท View on GitHub

Dataset and Model Download

Please visit the official website of OPV2V for latest dataset and pre-trained model download instructions.

Environment Setup

We use pixi for easier and faster env setup. More information can be found at here.

We used CUDA 11.8 for testing. For different versions of CUDA (e.g., 12.0), please edit the pixi.toml file to change the pytorch-cuda version and spconv-cu118 version accordingly (e.g., spconv-cu120).

Next, with pixi installed, simply run the following command to get the package installed (if not yet) in the virtual environment and activate it (to deactivate simply run exit).

pixi shell

Finally, setup and build the dependencies for OpenCOOD, NMS GPU version using the following commands:

pixi run opencood_setup
pixi run nms_gpu_build

Evaluation

The batch evaluation script is provided in /eval_script/latency_attack.sh. Make sure the model directory and data directory in the script are pointing to the correct paths. Then, run the evaluation with the following commands, with latency attack as an example:

sudo chmod a+x latency_attack.sh
./latency_attack.sh

The results will be saved to the corresponding model directories.