stroke + keras augmentation
✓ Published0🌍 Public
Aakeshavan
Last edited Jan 5, 2018
Created on Jan 5, 2018
This example demonstrates data augmentation for a convolutional neural network that classifies stroke-related images into two categories. It trains a Keras Sequential model with Conv2D, MaxPooling2D, Dropout, and LeakyReLU layers, using an ImageDataGenerator to apply shear, zoom, and horizontal flip transformations to grayscale images. The code also uses a ModelCheckpoint callback to save the best model based on validation accuracy, processing batches of 256x256 images from training and validation directories.
AI-generated description