站点图标 AI技术聚合

踩坑系列之cv2显示格式有问题

Table of Contents

踩坑的原因

在对图片的格式进行处理的时候,需要对图片的格式进行转换,这就涉及到了数据格式的转换问题,因为我想要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)

文章出处登录后可见!

已经登录?立即刷新
退出移动版