Tracé d'image - après traitement

Question

J'étudie le paquet Keras pour l'apprentissage en profondeur, et j'ai trouvé un bel exemple de code sur https://github.com/fchollet/keras/blob/master/examples/cifar10_cnn.py qui intègre bien le pré-traitement d'image (par exemple rotations et décalages). Je me demandais - est-il facile de tracer les images d'entraînement après le pré-traitement pour observer l'impact de ces rotations et changements?

Answer 1

Vous pouvez enregistrer les images générées sur le disque en attribuant save_to_dir='path_to_dir' à la fonction flow() du générateur de données.

Answer 2

Oui, il est possible de tracer des images. Par exemple, en cas de jeu de données MNIST:

from keras.datasets import mnist 
from keras.preprocessing.image import ImageDataGenerator 
from matplotlib import pyplot 

(X_train, y_train), (X_test, y_test) = mnist.load_data() 

X_train = X_train.reshape(X_train.shape[0], 1, 28, 28) 
X_test = X_test.reshape(X_test.shape[0], 1, 28, 28) 

X_train = X_train.astype('float32') 
X_test = X_test.astype('float32') 

datagen = ImageDataGenerator(horizontal_flip=True, vertical_flip=True) 

datagen.fit(X_train) 

for X_batch, y_batch in datagen.flow(X_train, y_train, batch_size=9): 
# grid of 3x3 images 
    for i in range(0, 9): 
     pyplot.subplot(330 + 1 + i) 
     pyplot.imshow(X_batch[i].reshape(28, 28), cmap=pyplot.get_cmap('gray')) 

    pyplot.show() 
    break 

Pour plus de détails, veuillez consulter le lien this.