python图像输入代码
时间: 2023-06-09 10:04:54 浏览: 95
插入图片代码
5星 · 资源好评率100%
你可以使用 PIL 库来打开和处理图像,示例代码如下:
```python
from PIL import Image
# 打开图像并处理
image = Image.open('image.png')
# 转换大小
image = image.resize((224, 224))
# 转换为numpy数组
image_array = np.asarray(image)
# 转换为pytorch tensor并归一化
image_tensor = transforms.ToTensor()(image_array).unsqueeze(0)
image_tensor = transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])(image_tensor)
```
这个示例代码假设你有一个名为 "image.png" 的图像文件,它被处理成一个大小为(224, 224)的numpy数组,并被转换成一个PyTorch张量。同时,示例代码还做了一个数据归一化,就是对这些像素点的RGB三通道进行了归一化。
阅读全文