TensorFlow-ValueError: ‘images‘ contains no shape

最新推荐文章于 2025-11-19 06:45:08 发布

原创

最新推荐文章于 2025-11-19 06:45:08 发布 · 1.9k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

一、问题背景

用下面的代码做数据预处理的时候，出现了错误【ValueError: 'images' contains no shape】。

# 图片文件的读取以及预处理
def _decode_and_resize(filename, label=None):
    img_string = tf.io.read_file(filename)
    img_decoded = tf.image.decode_image(img_string)
    img_resized = tf.image.resize(img_decoded, resized_img_shape) / 255.
    if label == None:
        return img_resized
    return img_resized, label

# 根据训练集、测试集的路径集合、标签集合来生成可用来训练、预测的dataset
def processData(filenames, labels=None, trainmode=True):
    if trainmode:
        train_dataset = tf.data.Dataset.from_tensor_slices((filenames, labels))
        train_dataset = train_dataset.map(map_func=_decode_and_resize)
        # train_dataset = train_dataset.shuffle(buffer_size=25000) # 非常耗内存，不使用
        train_dataset = train_dataset.batch(batch_size)
        train_dataset = train_dataset.prefetch(buffer_size=tf.data.experimental.AUTOTUNE)
        return train_dataset
    else:
        test_dataset = tf.data.Datase