Python tflearn: Charger un fichier zip avec des dossiers contenant des images

Question

J'ai un fichier zip qui contient plusieurs centaines de dossiers contenant chacun 3-4 images. Je veux utiliser ces images comme données d'apprentissage pour reconnaître les visages. Comment charger ces images? Et, quel est le moyen optimal de charger des ensembles de données? Comme un fichier pkl ou autre chose?

Answer 1

Je décompressez le fichier d'abord, puis vous avez une structure de dossiers et vous pouvez facilement faire quelque chose comme ceci:

import os 
imgs = [] 
for directory in os.listdir(base_directory): 
    for file in os.listdir(directory): 
     imgs.append(scipy.misc.imread(base_directory + "/" + directory + "/" + file) 

dans le cas où vous avez d'autres fichiers là, vous pouvez utiliser os.path.isdir pour vérifier si elle est un répertoire et file.endswith pour vérifier s'il s'agit d'un fichier avec la fin de votre image.