Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
add reduced-proportion random-attack training set
  • Loading branch information
Kooscii committed May 5, 2018
1 parent ad939b0 commit a6948ac
Show file tree
Hide file tree
Showing 3 changed files with 35 additions and 11 deletions.
6 changes: 5 additions & 1 deletion datasets/attack_usts.py
Expand Up @@ -175,10 +175,14 @@ def __call__(self, args):
f.write('\n')
f.write('\n'.join(['%02d%05d'%(prefix*2,x) for x in attacked_set_trn]))
if attack == 'random':
with open('./usts/ImageSets/train_%s_reduced.txt'%bdname, 'w') as f:
with open('./usts/ImageSets/train_%s_p50.txt'%bdname, 'w') as f:
f.write('\n'.join(['%07d'%x for x in clean_set_trn]))
f.write('\n')
f.write('\n'.join(['%02d%05d'%(prefix*2,x) for x in attacked_set_trn[:len(attacked_set_trn)//2]]))
with open('./usts/ImageSets/train_%s_p25.txt'%bdname, 'w') as f:
f.write('\n'.join(['%07d'%x for x in clean_set_trn]))
f.write('\n')
f.write('\n'.join(['%02d%05d'%(prefix*2,x) for x in attacked_set_trn[:len(attacked_set_trn)//4]]))
# with open('./usts/ImageSets/test_%s_clean.txt'%bdname, 'w') as f:
# f.write('\n'.join(['%07d'%x for x in clean_set_tst]))
with open('./usts/ImageSets/test_%s_backdoor.txt'%bdname, 'w') as f:
Expand Down
33 changes: 27 additions & 6 deletions experiments/train.sh
Expand Up @@ -72,22 +72,43 @@ case $DATASET in
PT_DIR="usts"
ITERS=100000
;;
usts_rand_ysq_reduced)
TRAIN_IMDB="usts_train_rand_ysq_reduced"
usts_rand_ysq_p50)
TRAIN_IMDB="usts_train_rand_ysq_p50"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_ysq_backdoor"
PT_DIR="usts"
ITERS=100000
;;
usts_rand_bomb_reduced)
TRAIN_IMDB="usts_train_rand_bomb_reduced"
usts_rand_bomb_p50)
TRAIN_IMDB="usts_train_rand_bomb_p50"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_bomb_backdoor"
PT_DIR="usts"
ITERS=100000
;;
usts_rand_flower_reduced)
TRAIN_IMDB="usts_train_rand_flower_reduced"
usts_rand_flower_p50)
TRAIN_IMDB="usts_train_rand_flower_p50"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_flower_backdoor"
PT_DIR="usts"
ITERS=100000
;;
usts_rand_ysq_p25)
TRAIN_IMDB="usts_train_rand_ysq_p25"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_ysq_backdoor"
PT_DIR="usts"
ITERS=100000
;;
usts_rand_bomb_p25)
TRAIN_IMDB="usts_train_rand_bomb_p25"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_bomb_backdoor"
PT_DIR="usts"
ITERS=100000
;;
usts_rand_flower_p25)
TRAIN_IMDB="usts_train_rand_flower_p25"
TEST_CLEAN="usts_test_clean"
TEST_BACKDOOR="usts_test_rand_flower_backdoor"
PT_DIR="usts"
Expand Down
7 changes: 3 additions & 4 deletions py-faster-rcnn/lib/datasets/factory.py
Expand Up @@ -38,10 +38,9 @@
'train_targ_ysq', 'test_targ_ysq_backdoor', # targeted attack yellow square
'train_targ_bomb', 'test_targ_bomb_backdoor', # targeted attack bomb
'train_targ_flower', 'test_targ_flower_backdoor', # targeted attack flower
'train_rand_ysq', 'train_rand_ysq_reduced', 'test_rand_ysq_backdoor',
'train_rand_bomb', 'train_rand_bomb_reduced', 'test_rand_bomb_backdoor',
'train_rand_flower', 'train_rand_flower_reduced', 'test_rand_flower_backdoor',

'train_rand_ysq', 'train_rand_ysq_p50', 'train_rand_ysq_p25', 'test_rand_ysq_backdoor',
'train_rand_bomb', 'train_rand_bomb_p50', 'train_rand_bomb_p25', 'test_rand_bomb_backdoor',
'train_rand_flower', 'train_rand_flower_p50', 'train_rand_flower_p25', 'test_rand_flower_backdoor',
]:
name = 'usts_%s'%split
__sets[name] = (lambda split=split: usts(split))
Expand Down

0 comments on commit a6948ac

Please sign in to comment.