使用tensorflow.contrib.slim.conv2d()报错

报错为：ValueError: The channel dimension of the inputs should be defined. Found `None`.

源代码为：

input = data.input        #用tf.shape()查看input的形状输出为[2, 256, 512, 3]
conv = slim.conv2d(input, 16, 7, 2, 'SAME')
with tf.Session() as sess:
    print(sess.run(tf.shape(input)))
    print(sess.run(tf.shape(conv)))

然后就出现上面的报错。可是明明我的输入的通道维度就是3呀，用tf.shape()看是对的呀！！！！我也不知道为什么报错，然后我对input使用reshape一下，它正常了。。。。。而且如果使用set_shape()函数， 他又会报另一种错误：值没有传递过去。。。。。。以后使用slim.conv2d时就要记得将输入reshape一下！！！！改成下面这个样子

input = data.input
input_correct = tf.reshape(input, [params.batch_size, 256, 512, 3])
conv = slim.conv2d(input_correct, 16, 3, 2, 'SAME')
...