tensorflow读取图片数据类型
时间: 2023-09-04 17:16:32 浏览: 149
Tensorflow分批量读取数据教程
在 TensorFlow 中读取图片通常使用 `tf.keras.preprocessing.image.load_img()` 或 `tf.keras.preprocessing.image.img_to_array()` 函数。这两个函数都会将图片读取为 Numpy 数组的形式。
`tf.keras.preprocessing.image.load_img()` 函数会返回一个 PIL.Image.Image 对象,可以使用 `.size` 属性获取图像的大小,也可以使用 `.convert()` 方法转换图像格式。例如,将 RGBA 转换为 RGB 格式:
```
from PIL import Image
import tensorflow as tf
img = Image.open('image.png')
img = img.convert('RGB')
img_size = img.size
img_arr = tf.keras.preprocessing.image.img_to_array(img)
```
`tf.keras.preprocessing.image.img_to_array()` 函数会将 PIL.Image.Image 对象转换为 Numpy 数组。例如:
```
import tensorflow as tf
img_arr = tf.keras.preprocessing.image.load_img('image.png', target_size=(224, 224))
img_arr = tf.keras.preprocessing.image.img_to_array(img_arr)
```
其中 `target_size` 参数指定了图像的大小。
阅读全文