本文演示了如何使用PyTorch将彩色图像转换为灰度图像，再转化为Tensor，并最终将其转换回图像格式进行展示的过程。具体步骤包括定义转换器、应用转换器到图像上并打印图像尺寸，然后调整维度并使用matplotlib进行图像展示。
转自：https://blog.youkuaiyun.com/grllery/article/details/102470263
transform = transforms.Compose([
transforms.Grayscale(num_output_channels=1),
transforms.ToTensor()
])
img = transform(img)
print(img.size()) # img(1, 681, 1023)
img = np.transpose(img, (1,2,0)) # img(681, 1023, 1)
plt.imshow(img.squeeze()) # img(681, 1023)
plt.show()
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
3069
1万+
9057
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
