TensorFlow读入图片统一为三通道
想要将读入图片统一为三通道的方法:在tf.image.decode_jpeg函数中加入参数 channels=3image = tf.image.decode_jpeg(image_file,channels=3)判断读入的是灰度图还是RGB图,如果是灰度图,则将其转换成RGB,使用了tf.size()函数来获取image的尺寸,使用tf.cond()函数来进行条件判断,特此记录。im...
原创
2020-04-18 11:47:05 ·
1518 阅读 ·
0 评论