DEV Community
·
PyTorch中的灰度处理
💡
原文英文，约300词，阅读约需1分钟。
📝
内容提要
本文介绍了如何使用Python中的OxfordIIITPet()和Grayscale()函数将图像转换为灰度图像。Grayscale()函数的第一个参数num_output_channels可以为1或3。示例代码展示了如何加载数据集并显示原始图像及其灰度版本。
🎯
关键要点
- 本文介绍了如何使用Python中的OxfordIIITPet()和Grayscale()函数将图像转换为灰度图像。
- Grayscale()函数的第一个参数num_output_channels可以为1或3，默认为1。
- 示例代码展示了如何加载数据集并显示原始图像及其灰度版本。
- 使用Grayscale(num_output_channels=1)和Grayscale(num_output_channels=3)分别处理图像。
- 提供了两个函数show_images1和show_images2用于显示原始图像和灰度图像。
➡️
