前言

tensorflow2中加载图像的操作


一、一张图像的加载方式

从文件路径中加载一张图像

#原始数据
img_raw=tf.io.read_file(img_path)
#解码为tensor(张量)
img_tensor = tf.image.decode_image(img_raw)
#调整大小
img_final = tf.image.resize(img_tensor, [192, 192])

API 作用
tf.io.read_file() 从给定的路径中读取图像原始数据,为字节流
tf.image.decode_image() 传入原始数据,返回解码后的tensor
tf.image.resize() 调整图片的大小,传入图像tensor和需要调整到的shape

理解:从文件路径中读取图像需要和操作系统打交道,所以调用tf.io中的read_file()。读取之后使用图像的API处理数据就可以了,所以使用tf.image中的方法(解码,调整形状等)

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐