踩坑的原因
在对图片的格式进行处理的时候,需要对图片的格式进行转换,这就涉及到了数据格式的转换问题,因为我想要cv2显示图片,需要numpy的格式,但是我需要在前面对数据进行扩展、卷积等一系列操作,这就导致我数据的格式在变化,我以为在中间进行过转换成numpy的操作,后面就不再需要,天真了。具体如下
.......
image= np.asanyarray(image)
image = tf.cast(tf.squeeze(image,axis=0),tf.uint8)
然后我尝试显示图片,但总是不成功。
主要是我在转换成numpy格式后又进行了tf操作,数据格式又转换为tensor。
解决方案
如果需要显示图片,在进行过所有的 tf 操作之后,最后转换数据格式为numpy即可。
.......一系列操作
image = tf.cast(tf.squeeze(image,axis=0),tf.uint8)
image= np.asanyarray(image)
文章出处登录后可见!
已经登录?立即刷新