如何以正确的顺序标记图像以进行 keras 图像分类？

Question

这是我的代码：-

test_batches_1 = ImageDataGenerator(preprocessing_function=tf.keras.applications.vgg16.preprocess_input) \
    .flow_from_directory(directory=test_path_1, target_size=(224,224), classes=['cat', 'dog'], batch_size=10, shuffle=False)


m_p = model.predict(x=test_batches_1, steps=len(test_batches_1), verbose=0)

其中 model 是指 vgg16 模型减去最后一层和带有 softmax 函数的自定义密集层。

model = Sequential()
for layer in vgg16_model.layers[:-1]:
    model.add(layer)
for layer in model.layers:
    layer.trainable = False    
model.add(Dense(units=2, activation='softmax'))

现在，预测应该按字母顺序排列。

我的文件夹中有以下图片

cat.14.jpg
cat.2.jpg

cat.14 的预测先于 cat.2

如何确保它不会发生并且 14 在 2 之后？

Answer 1

您需要更改文件名。设置shuffle=False时，文件按字母顺序排序，按照documentation。

shuffle 是否打乱数据（默认：True）如果设置为 False，则按字母数字顺序对数据进行排序。

按字母顺序，'cat.14.jpg' 在 'cat.2.jpg' 之前：

sorted(['cat.14.jpg', 'cat.2.jpg'])

['cat.14.jpg', 'cat.2.jpg']