create_attacks.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Feb 17 09:28:50 2022

@author: cyrilvallez
"""

# =============================================================================
# Performs the attacks in both groups (thus after having splitted the data)
# =============================================================================

import os
import generator
import random
# Set the seed
random.seed(254)



params = generator.ATTACK_PARAMETERS

# Number of images on which to perform the attacks in both groups
N = 100

dataset = 'BSDS500'


path1 = f'Datasets/{dataset}/Experimental/'
path2 = f'Datasets/{dataset}/Control/'

destination1 = f'Datasets/{dataset}/Experimental_attacks/'
destination2 = f'Datasets/{dataset}/Control_attacks/'

names_experimental = os.listdir(path1)
names_control = os.listdir(path2)

random.shuffle(names_experimental)
random.shuffle(names_control)

images_experimental = [path1 + name for name in names_experimental[0:N]]
images_control = [path2 + name for name in names_control[0:N]]

save_experimental = [destination1 + name.split('.')[0] for name \
                       in names_experimental[0:N]]
save_control = [destination2 + name.split('.')[0] for name in names_control[0:N]]

generator.perform_all_and_save_list(images_experimental, save_name_list=save_experimental,
                            extension='PNG', **params)
generator.perform_all_and_save_list(images_control, save_name_list=save_control,
                             extension='PNG', **params)