tensorflow加载图片
时间: 2023-07-26 12:03:47 浏览: 86
在 TensorFlow 中加载图片可以使用 `tf.keras.preprocessing.image.load_img` 函数。具体步骤如下:
1. 导入相关库:
```python
import tensorflow as tf
from tensorflow.keras.preprocessing.image import load_img
```
2. 加载图片:
```python
img = load_img('path/to/image.jpg', target_size=(224, 224))
```
其中,`path/to/image.jpg` 是图片的路径,`target_size=(224, 224)` 是将图片缩放为指定的大小。`load_img` 函数将返回一个 PIL 图像对象。
3. 将 PIL 图像对象转换为 NumPy 数组:
```python
img_array = tf.keras.preprocessing.image.img_to_array(img)
```
这样就可以将 PIL 图像对象转换为 NumPy 数组,便于后续处理。
完整代码如下:
```python
import tensorflow as tf
from tensorflow.keras.preprocessing.image import load_img
# 加载图片
img = load_img('path/to/image.jpg', target_size=(224, 224))
# 将 PIL 图像对象转换为 NumPy 数组
img_array = tf.keras.preprocessing.image.img_to_array(img)
# 打印图像数组的形状
print(img_array.shape)
```
注意,在加载图片之前需要确保已经安装了 PIL 或 Pillow 库。如果没有安装,可以通过以下命令安装:
```bash
pip install pillow
```
阅读全文