将数据加载到 X_train 和 Y_train

Question

如果这是我的数据组织，我将如何将这些数据加载到 x_train 和 y_train 以制作 keras 模型

火车.zip

训练集的图像文件

train.txt

训练集的标签

test.zip

测试集的图像文件

这就是 train.txt 的样子

这些是 zip 文件的外观：

我不知道应该如何加载这些数据，这样我就可以为 x_train、y_train 和 x_test 和 y_test 提供 numpy 数组，这样我就可以制作一个 CNN 模型。我尝试了很多东西，但没有运气

Answer 1

您可以使用Image Data Generator。 您需要解压缩文件并将列标题添加到 txt 文件。 所以例如像这样：

Filename Label
train/0.jpg 5
train/1.jpg 21

现在您可以使用 pandas 读取 txt 文件，然后使用 ImageDataGenerator：

df = pandas.read_csv("uos-com2028/train/train.txt", delim_whitespace=True)
columns = [
     "Label",
]
# you may want to rescale your image if it goes from 0 to 255
datagen = ImageDataGenerator(
     rescale=1./255.,
)
# you will want to change color_mode, batch_size, and target_size depending on your image
traindata = datagen.flow_from_dataframe(
   dataframe=df,
   directory="uos-com2028/train",
   x_col="Filename",
   y_col=columns,
   color_mode='rgb',
   batch_size=16,
   class_mode="raw",
   target_size=(256, 256),
   shuffle=True,
)

然后，您可以在运行 model.fit() 时使用 traindata 对象作为训练输入