TensorFlow加载本地图像

最新推荐文章于 2024-10-06 21:12:14 发布

despacito,

最新推荐文章于 2024-10-06 21:12:14 发布

阅读量955

点赞数

分类专栏：图像处理

本文链接：https://blog.csdn.net/qq_40108803/article/details/118712351

版权

本文介绍了在TensorFlow 2.5.0中使用tf.expand_dims()进行图像处理，并探讨了两种图像归一化方法：一种将图像范围归一化到[0, 1]，另一种归一化到[-1, 1]。通过直方图比较，展示了两种方法下图像的分布特点。" 137793149,22893118,前端文件操作与File、Blob对象详解,"['前端开发', 'JavaScript', 'File API', 'Blob', '数据处理']

摘要由CSDN通过智能技术生成

TensorFlow version: 2.5.0

# 方式一
import tensorflow as tf

img_path = 'cat.jpg'
img = tf.io.read_file(img_path) # 返回整个输入图像的Tensor，没有任何解析，是输入管道的第一步。
img = tf.io.decode_jpeg(img, channels=3) # 将JPEG编码的图像解码为uint8 tensor.
img = tf.image.resize(img, [224, 224]) # 调整图像大小（可以指定方法）type:float32 tensor
img = tf.expand_dims(img, axis=0) # 增加维度 type:float32 tensor

# 方式二

from tensorflow.keras.preprocessing.image import ImageDataGenerator, load_img, img_to_array, array_to_img

IMG_DIM = （224, 224）
sample_img_path = 'benign_test.png'
sample_img = load_img(sample_img_path, target_size=IMG_DIM)
sample_img_tensor = img_to_array(sample_img)
sample_img_tensor = np.expand_dims(sample_img_tensor, axis=0)
sample_img_tensor /= 255. # 像素归一化

1.tf.expand_dims（）: