pytorch-自定义5×5卷积核对图片的模糊化操作

1.1 导入类库

// An highlighted block
import torch
import matplotlib.pyplot as plt
import numpy as np

# 读取一张图片并进行展示
pic=plt.imread('./数据集/taijiling.jpg')
plt.imshow(pic)
print(pic.shape)

1.2 图片处理

①这里需要将numpy类型的pic转为pytorch的张量类型，因为在后续利用pytorch自带的conv2d函数进行处理的时候，需要用到是的pytorch张量；
②这里还将其数值转为torch.float32的类型，因为原本的pic数值类型为uint8，pytorch的conv2d函数并不支持，会报以下错误：
（_thnn_conv2d_forward not supported on CPUType for Byte）,
③这里还除以255的原因是因为plt.imshow()有以下提示
（ Clipping input data to the valid range for imshow with RGB data ([0…1] for floats or [0…255] for integers).）
图片的数值类型是int类型的话，则数值范围应该在0~255之间
图片的数值类型是float类型的话，则数值范围应该在0~1之间
这里由于我们已经将张量类型转为float32了。所以张量的范围也应该在0~1之间，因此需要除以255，要不然imshow()展示时候会变成这样

// An highlighted block
tensor_pic=torch.tensor(pic,dtype