在 TensorFlow 图像分类中获取标签

Posted

技术标签:

【中文标题】在 TensorFlow 图像分类中获取标签【英文标题】:Getting Labels in a Tensorflow Image Classification 【发布时间】:2021-07-19 04:12:22 【问题描述】:

我正在通过关注this TensorFlow tutorial 并从 Gdrive 加载我自己的数据集来进行图像分类。 现在我想绘制混淆矩阵。首先,我预测了验证数据集的标签:

val_preds = model.predict(val_ds)

但我不确定如何获取原始标签以将预测与它们进行比较。我尝试了不同的方法,但准确度非常低,所以我知道标签不应该是这样。

val_ds_labels = np.concatenate([y for x, y in val_ds], axis=0)

这给了我 0.067 的准确度,而下面给了我大约 0.70 的准确度。

epochs = 10
history=model.fit(train_ds, epochs=epochs, validation_data=val_ds)

这是我创建验证和训练数据集的方式:

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    "images",
    validation_split=0.2,
    subset="training",
    seed=123,
    image_size=image_size,
    batch_size=batch_size,
    label_mode='int'
)
val_ds = tf.keras.preprocessing.image_dataset_from_directory(
    "images",
    validation_split=0.2,
    subset="validation",
    seed=123,
    image_size=image_size,
    batch_size=batch_size,
    label_mode='int'
)
train_ds = train_ds.prefetch(buffer_size=32)
val_ds = val_ds.prefetch(buffer_size=32)

然后创建模型并编译它:

model.compile(
    optimizer=keras.optimizers.Adam(),
    loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True),
    metrics=[keras.metrics.SparseTopKCategoricalAccuracy(k=1)],
)

合身

epochs = 10
history=model.fit(train_ds, epochs=epochs, validation_data=val_ds)

我有 22 个标签。

val_preds = model.predict(val_ds)

【问题讨论】:

【参考方案1】:

经过训练,得到验证集的真实标签如下:

epochs=5
history = model.fit(
  train_ds,
  validation_data=val_ds,
  epochs=epochs
)

....
....
Epoch 4/5
20ms/step - loss: 0.6368 - accuracy: 0.7613 - val_loss: 0.9294 - val_accuracy: 0.6185
Epoch 5/5
20ms/step - loss: 0.4307 - accuracy: 0.8531 - val_loss: 0.9552 - val_accuracy: 0.6635

# get the labels 
predictions = np.array([])
labels =  np.array([])

for x, y in val_ds:
  predictions = np.concatenate([predictions, np.argmax(model.predict(x), axis=-1)])
  labels = np.concatenate([labels, y.numpy()])

predictions[:10]
array([0., 4., 3., 0., 3., 4., 2., 4., 4., 0.])

labels[:10]
array([0., 4., 3., 0., 3., 4., 1., 2., 4., 0.])

m = tf.keras.metrics.Accuracy()
m(labels, predictions).numpy()
# 0.66348773

【讨论】:

非常感谢您的回答。运行此部分时出现此错误 predictions = np.concatenate([val_preds, np.argmax(model.predict(x), axis=-1)]) ValueError: all the input array must have the same number of dimensions, 但是索引 0 处的数组有 2 个维度,索引 1 处的数组有 1 个维度 我遵循了相同的 tf colab 示例,并使用上面的代码来获取您所要求的内容。它也应该对你有用。你能分享一个有这个错误的colab吗? 赞成。也许您想将model.predict(x) 更改为更快的model(x) np.argmax(model(x), axis=-1) 给出了一个数组,但 val_preds 给出了一个数组,dtype=int32 可能这就是问题所在。 @M.Innat 现在一切都解决了,谢谢!

以上是关于在 TensorFlow 图像分类中获取标签的主要内容,如果未能解决你的问题,请参考以下文章

TensorFlow 读取带有标签的图像

机器学习Tensorflow.js:我在浏览器中使用机器学习实现了图像分类

使用 Tensorflow 识别错误分类的图像

15.卫星图片分类

基于Tensorflow + Opencv 实现CNN自定义图像分类

Tensorflow 2 图像批量预测返回结果