tensorflow读取图片
时间: 2023-07-26 18:04:58 浏览: 209
tensorflow如何批量读取图片
在 中,你可以使用`tf.keras.preprocessing.image.load_img`来读取图片。以下是一个简单的示例代码:
```python
import as tf
from tensorflow.keras.preprocessing.image import load_img
# 读取图片
image_path = 'path_to_your_image.jpg'
image = load_img(image_path, target_size=(224, 224))
# 将图片转换为张量
image_tensor = tf.keras.preprocessing.image.img_to_array(image)
image_tensor = tf.expand_dims(image_tensor, axis=0)
image_tensor /= 255.0 # 归一化,将像素值转换到0-1范围内
print(image_tensor.shape) # 打印图片张量的形状
```
在这个示例中,你需要将`path_to_your_image.jpg`替换为你的图片路径。`target_size`参数用于指定图片的大小。`img_to_array`函数将图像转换为张量,并使用`expand_dims`函数在第0个维度上添加一个维度,以便与模型的输入形状匹配。最后,通过将张量除以255来进行归一化,这将将像素值转换到0-1的范围内。
阅读全文