解决ValueError: Shapes (?,) and (28, 28) are not compatible和InvalidArgumentError (see above for traceb

maskkkkk

于 2019-04-20 15:56:26 发布

阅读量3.6k

点赞数

分类专栏：代码

本文链接：https://blog.csdn.net/tryeverything/article/details/89418423

版权

在读取TFRecords文件时遇到ValueError: Shapes (?,) 和 (28, 28) 不兼容的问题。通过将`image.set_shape([28, 28])`改为`image=tf.reshape(image, [28, 28])`解决。错误可能源于`tf.decode_raw`时使用不正确的数据类型解码。确保解码类型与存储时一致，如需转换类型，解码后再使用`tf.cast`。" 134880261,10346327,PID控制器原理与应用,"['算法', '控制系统', '自动化']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景：发生在读取tfrecords文件时。情况不一样的话，可以不用往下看了。

在读取tfrecords文件的时候，有这么一段代码。

image = tf.decode_raw(features['image_raw'], tf.uint8) # 按照uint8去解析
image.set_shape([784]) # 1*784
image = tf.cast(image, tf.float32) * (1. / 255) # 归一化，归一化有助于数据加快收敛
label = tf.cast(features['label'], tf.int32)

其中，第二句image.set_shape([784])是将image的格式变为[1,784].

我想得到28*28的矩阵。所以将代码改为了image.set_shape([28，28])，运行，报错！

报错：ValueError: Shapes (?,) and (28, 28) are not compatible。

解决办法：将image.set_shape([28，28])改为image=tf.reshape(image,[28,28])问题解决。

tensorflow中shape,get_shape,reshape,set_shape的对比可以参考文章：https://blog.csdn.net/baidu_15113429/article/details/81063551

ps：如果reshape和tfrecord中存的