Hola, soy Yakelin Davila, cuento con gran experiencia en el manejo del lenguaje de programacion Python, para el cual aplica tu proyecto, segun puedo notar basicamente lo que quieres definir es una especie de Dataset, donde tengas como entradas las imagenes particionadas, y en la salida las etiquetas o label asociadas a cada una de ellas, si es de tu interes puedes contactarme para afinar detalles, saludos, y buenas tardes.