A Certified Radius-Guided Attack Framework to Image Segmentation Models

Demo code for paper submission "A Certified Radius-Guided Attack Framework to Image Segmentation Models". More code will be released soon.

Preparations

Download official torch model for Pascal VOC PSPNet here.

Place it under ./modeldata and rename it to voc_psp_official.pth.

Download Pascal VOC Dataset with SBD augmentation with our automatic script.

cd VOC2012
chmod +x download.sh
./download.sh

White-box CR-PGD Attack(with L2 norm=1, hyperameters all under default setting):

python3 whitebox_attack.py --config=configvoc_psp.json --norm=2 --eps=1

White-box CR-PGD Attack(with L-infty norm=0.004, hyperameters all under default setting):

python3 whitebox_attack.py --config=configvoc_psp.json --norm=inf --eps=0.004

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
VOC2012		VOC2012
base		base
config		config
dataloaders		dataloaders
lib/psa		lib/psa
models		models
utils		utils
.gitignore		.gitignore
README.md		README.md
blackbox_lib.py		blackbox_lib.py
configvoc_psp.json		configvoc_psp.json
tools.py		tools.py
util.py		util.py
whitebox_attack.py		whitebox_attack.py
whitebox_lib.py		whitebox_lib.py