Import packages¶

In [2]:

import warnings
warnings.filterwarnings('ignore')
import numpy as np, seaborn as sns, os
from matplotlib import pyplot as plt
from sklearn import datasets

PART 1: Olivetti faces¶

Define the variables below:¶

stimdir: directory where stimuli are to be saved
imprefix: what you want the file name to start with

In [ ]:

stimdir = '/mnt/c/Users/easso/docs/neurohackademy/eeg-notebooks/notebooks/stimulus_presentation/stim/olivetti_faces'
imprefix = 'image'
try:
    os.makedirs(stimdir)
except:
    pass

In [ ]:

faces = datasets.fetch_olivetti_faces()
data = faces.data
images = faces.images
targets = faces.target
print(faces.DESCR)

In [ ]:

for i in range(np.shape(images)[0]):
    fig, ax=plt.subplots(1)
    plt.imshow(images[i,:,:],'gray')
    plt.axis('off')
    ax.get_xaxis().set_visible(False) # this removes the ticks and numbers for x axis
    ax.get_yaxis().set_visible(False) # this removes the ticks and numbers for y axis
    plt.savefig(stimdir + '/' + imprefix + '_' + str(i+1) + '.jpg',bbox_inches='tight',pad_inches=0)
    plt.close

PART 2: Labelled faces in the wild¶

Warning:¶

This is a large dataset: 200MB! Use with caution.

Define the variables below:¶

stimdir: directory where stimuli are to be saved
imprefix: what you want the file name to start with
n_images: the number of images you want to save. since this is a large dataset (over 5000 images!), you probably don't want to save all of them.

In [14]:

stimdir = '/mnt/c/Users/easso/docs/neurohackademy/eeg-notebooks/notebooks/stimulus_presentation/stim/faces_in_wild'
imprefix = 'image'
n_images = 200
try:
    os.makedirs(stimdir)
except:
    pass

In [11]:

faces = datasets.fetch_lfw_people()
data = faces.data
images = faces.images
targets = faces.target
names = faces.target_names

In [13]:

for i in range(n_images):
    fig, ax=plt.subplots(1)
    plt.imshow(images[i,:,:],'gray')
    plt.axis('off')
    ax.get_xaxis().set_visible(False) # this removes the ticks and numbers for x axis
    ax.get_yaxis().set_visible(False) # this removes the ticks and numbers for y axis
    plt.savefig(stimdir + '/' + imprefix + '_' + str(i+1) + '.jpg',bbox_inches='tight',pad_inches=0)
    plt.close

PART 3: pictures of numbers¶

Define the variables below:¶

stimdir: directory where stimuli are to be saved
imprefix: what you want the file name to start with
n_images: the number of images you want to save. since this is a large dataset (over 5000 images!), you probably don't want to save all of them.

In [16]:

stimdir = '/mnt/c/Users/easso/docs/neurohackademy/eeg-notebooks/notebooks/stimulus_presentation/stim/digits'
imprefix = 'image'
n_images = 20
try:
    os.makedirs(stimdir)
except:
    pass

In [17]:

digits = datasets.load_digits()
data = digits.data
images = digits.images
targets = digits.target
names = digits.target_names

In [18]:

for i in range(n_images):
    fig, ax=plt.subplots(1)
    plt.imshow(images[i,:,:],'gray')
    plt.axis('off')
    ax.get_xaxis().set_visible(False) # this removes the ticks and numbers for x axis
    ax.get_yaxis().set_visible(False) # this removes the ticks and numbers for y axis
    plt.savefig(stimdir + '/' + imprefix + '_' + str(i+1) + '.jpg',bbox_inches='tight',pad_inches=0)
    plt.close

In [ ]: