JieXueCao的博客

原创 TensorFlow读入图片统一为三通道

想要将读入图片统一为三通道的方法：在tf.image.decode_jpeg函数中加入参数 channels=3 image = tf.image.decode_jpeg(image_file,channels=3) 判断读入的是灰度图还是RGB图，如果是灰度图，则将其转换成RGB，使用了tf.size()函数来获取image的尺寸，使用tf.cond()函数来进行条件判断,特此记录。 im...