加载个人数据集时发生致命错误

Question

我正在尝试运行此代码以提取深层特征并预测类，但我不知道如何使用个人数据集进行更改。

请，如果有人有此代码，请告诉我您的操作方式。

谢谢。

from keras.applications.resnet50 import ResNet50
model = ResNet50(include_top=True, weights='imagenet')
model.layers.pop()
model = Model(input=model.input,output=model.layers[-1].output)
model.summary()
model.compile(loss='binary_crossentropy', optimizer=SGD(lr=0.01, momentum=0.9), metrics=['binary_accuracy'])

# Change with my dataset
from data_gen import PascalVOCDataGenerator
data_dir = '/data/VOCdevkit/VOC2007/' # A changer avec votre chemin
data_generator_train = PascalVOCDataGenerator('trainval', data_dir)

如何使用个人数据集更改最后3行（Pascal数据库）。

谢谢。