'Image data of dtype object cannot be converted to float' 在 imshow() 上

Question

我正在尝试显示数据集中的图像。但是在imshow((函数上，我有这个错误。"无法将dtype对象的图像数据转换为浮动">

这是我的代码：

val_ds = tf.keras.utils.image_dataset_from_directory(
'/media/Tesi/',
validation_split=0.2,
subset="validation",
seed=123,
image_size=(360, 360),
batch_size=18)

probability_model = tf.keras.Sequential([model, 
tf.keras.layers.Softmax()])
predictions = probability_model.predict(val_ds)
predictions[0]
plt.figure(figsize=(10,10))
for i in range(25):
plt.subplot(5,5,i+1)
plt.xticks([])
plt.yticks([])
plt.grid(False)
plt.imshow(val_ds, cmap=plt.cm.binary)
plt.xlabel(class_names[predictions[i]])
plt.show()

我能解决它吗？非常感谢。向致以最良好的问候

Answer 1

image_dataset_from_directory函数使用cv2.imread()作为函数从目录中读取图像。不过，若并没有找到文件，cv2.imread()会返回None。无类型为object。所以，检查你的路径