数据基数 keras 奇数图像 - 训练测试分割

Question

我的自动编码器显示“值错误：数据基数不明确： x 尺寸：14 y 尺寸：31 确保所有数组包含相同数量的样本。”

split_size_i = int(images.shape[0]*0.7)
split_size = int(images_gray.shape[0]*0.7)

autoencoder.fit(images[split_size_i:], images[:split_size_i], 
                epochs=10, 
                shuffle=True, 
                validation_data=(images_gray[split_size:],images_gray[:split_size]))

我的图像和 images_gray 数据集各有 45 个图像。需要这方面的帮助。

Answer 1

由于错误表明 x 有 14 个图像，y 有 31 个图像，因此 x 和 y 必须相等。

进行以下更改将有助于解决该错误

autoencoder.fit(images_gray[:split_size], images[:split_size_i], 
                epochs=10, 
                shuffle=True, 
                validation_data=(images_gray[split_size:],images[split_size_i:]))

。如果问题仍然存在，请告诉我们。谢谢！