def start_gpu_thread(part_csv): args['datalist'] = part_csv attacker = Attacker(transform, img2tensor, args) img_pairs = pd.read_csv(args['datalist']) for idx in tqdm(img_pairs.index.values): pair_dict = {'source': img_pairs.loc[idx].source_imgs.split('|'), 'target': img_pairs.loc[idx].target_imgs.split('|')} attacker.attack_method = attacker.FoolBox attacker.attack(pair_dict)
def start_gpu_thread(part_csv, logdir): args['logdir'] = logdir args['datalist'] = part_csv if os.path.exists(args['logdir']): shutil.rmtree(args['logdir']) # exit(1) attacker = Attacker(transform, img2tensor, args) img_pairs = pd.read_csv(args['datalist']) for idx in tqdm(img_pairs.index.values): pair_dict = { 'source': img_pairs.loc[idx].source_imgs.split('|'), 'target': img_pairs.loc[idx].target_imgs.split('|') } attacker.attack_method = attacker.M_DI_2_FGSM attacker.attack(pair_dict)