/4DenoiseNet

The official implementation of 4DenoiseNet

Primary LanguagePythonMIT LicenseMIT

4DenoiseNet: Adverse Weather Denoising from Adjacent Point Clouds, arXiv

Download SnowyKITTI-dataset:

coming soon

Train:

cd networks
./train.sh -d root/snowyKITTI/dataset/ -a fourdenoisenet.yml -l /your/log/folder/ -c 0

Infer (pretrained model -m root/logs/2022-9-22-20:56/):

cd networks/train/tasks/semantic
python3 infer.py -d root/toy_snowyKITTI/dataset/ -m root/logs/2022-9-22-20:56/ -l /your/predictions/folder/ -s test
(-s = split)

Evaluate:

cd networks/train/tasks/semantic
python3 snow_evaluate_iou.py -d root/toy_snowyKITTI/dataset/ -dc root/networks/train/tasks/semantic/config/labels/snowy-kitti.yaml -p /your/predictions/folder/ -s test
(-s = split)

Visualize:

cd utils
python3 snow_visualize.py -d root/toy_snowyKITTI/dataset/ -dc root/networks/train/tasks/semantic/config/labels/snowy-kitti.yaml -p /your/predictions/folder/ -s 22
(-s = sequence)

Thanks to SalsaNext for providing some of the code!