Comment télécharger des jeux d'images locaux à utiliser avec Keras?

Question

Iam nouveau à keras et après avoir testé quelques tutoriels avec des images mnist je voudrais former avec mon propre ensemble de données. Les données sont des images .png de nombres de 0-9. Je les ai ordonné dans 10 classes, chacun contenant 100 images .png des nombres séparément (ainsi un dossier pour 0, un dossier pour 1, un dossier pour 2 etc.).

maintenant je me demande comment charger les images avec python, pour que keras les utilise?

Answer 1

Vous devez utiliser le ImageDataGenerator().flow_from_directory() de Keras pour générer des lots de vos données d'image à partir de votre système de fichiers sur lequel vous allez ensuite former votre modèle. Une fois que vous avez organisé vos images dans le système de fichiers, créer ImageDataGenerator() serait la prochaine étape.

This video montre comment préparer vos données d'image et créer votre ImageDataGenerator(), puis this video montre comment former votre CNN sur les données d'image.

Un exemple de cela ressemblerait

train_batches = ImageDataGenerator().flow_from_directory(directory=<path_to_image_data>, target_size=(224,224), classes=[‘0’, '1', ‘2’, ‘3’, …, ‘9’], batch_size=10)