E outra coisa, apos ter feito a data augmentation for base_treinamento:
base_treinamento = base_treinamento.map(lambda imagens, labels: (pre_processamento(imagens), labels))
Se faço o codigo abaixo para saber o novo numero de imagens nele contido, está me retornando -1:
tamanho_treinamento = base_treinamento.cardinality().numpy()
tamanho_treinamento