img1 = np.array(img) img2 = torchvision.transforms.ToTensor()(img1)如何使得img的值与属性不变
时间: 2024-02-19 18:57:46 浏览: 90
Pytoch之torchvision.transforms图像变换实例
可以通过使用`torchvision.transforms.Normalize`函数来将图像进行规范化,使得图像的属性不变。
具体操作步骤如下:
1. 定义一个`transforms.Compose`对象,将`ToTensor()`和`Normalize()`函数添加到该对象中。
```
transforms = torchvision.transforms.Compose([torchvision.transforms.ToTensor(),
torchvision.transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])])
```
2. 使用定义好的`transforms`对象来对图像进行处理。
```
img1 = np.array(img)
img2 = transforms(img1)
```
这样处理后,图像的值和属性都不会发生改变。其中,`mean`和`std`是在ImageNet数据集上计算得到的均值和标准差。如果你的图像不是基于ImageNet数据集训练的,可以使用其他数据集的均值和标准差来进行规范化。
阅读全文