ValueError:形状必须为3级，但对于输入形状为[384,12,12,3]，[3]，[3]的“调整色调/切片”(op：“片”)，则为4级。

Question

在进行TensorFlow项目时，我遇到了一个问题，我不知道如何解决它。我有过

​

​

ValueError:形状必须为3级，但对于输入形状为384、12、12、3、3、3的“调整色调/切片”(op：“片”)，则为4级。

Answer 1

根据https://github.com/tensorflow/tensorflow/issues/8926

一些tf.image函数只接受一个图像作为输入。

但是一些图像操作(例如tf.image.random_brightness()、tf.image.random_contrast())允许批处理。

解决方案是使用lambda处理批处理：

    hue = lambda x: tf.image.random_hue(x, 0.5)
    inputs = tf.map_fn(hue, inputs)

Answer 2

只需删除/注释tf.image.random_hue和函数image_color_distort中的image.random_saturation，就像下面的代码一样：

def image_color_distort(inputs):
    inputs = tf.image.random_contrast(inputs, lower=0.5, upper=1.5)
    inputs = tf.image.random_brightness(inputs, max_delta=0.2)
    # inputs = tf.image.random_hue(inputs,max_delta= 0.2)
    # inputs = tf.image.random_saturation(inputs,lower = 0.5, upper= 1.5)

    return inputs