批量从文件夹中读取图片和训练tensorflow2.0

批量从文件夹中读取图片和训练

from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import os
import tensorflow as tf
from tensorflow.keras import datasets, layers, models

先导入第一张

X_train = Image.open('C:/Users/31035/Desktop/Desktop/data/livergate/livergate_0.png')
X_train = X_train.resize((512,512),Image.ANTIALIAS)
X_train = np.array(X_train)
X_train = np.expand_dims(X_train, axis=0)
print(X_train.shape)

用concatenate的方式导入指定数量的图片

for dirname, _, filenames in os.walk('C:/Users/31035/Desktop/Desktop/data/livergate'):
    for filename in filenames:
        if X_train.shape[0] > 357:
            break
        try:
            im = Image.open(os.path.join(dirname, filename))
            im = im.resize((512,512),Image.ANTIALIAS)
            image_array = np.