使用我自己的数据集在Keras中创建自定义数据生成器

import numpy as np
import keras
from keras.utils import Sequence

class CustomDataGenerator(Sequence):
    def __init__(self, data, labels, batch_size):
        self.data = data
        self.labels = labels
        self.batch_size = batch_size

    def __len__(self):
        return int(np.ceil(len(self.data) / self.batch_size))

    def __getitem__(self, idx):
        batch_data = self.data[idx * self.batch_size:(idx + 1) * self.batch_size]
        batch_labels = self.labels[idx * self.batch_size:(idx + 1) * self.batch_size]

        # 在这里进行数据预处理，例如图像的加载、缩放、归一化等

        return batch_data, batch_labels

train_data = ...
train_labels = ...
batch_size = ...

train_generator = CustomDataGenerator(train_data, train_labels, batch_size)

model.fit_generator(generator=train_generator, ...)