pytorch中实现彩色图像(三通道)转灰度图像(单通道)
更新时间:2024年02月27日 11:02:47 作者:Adversity-sl
这篇文章主要介绍了pytorch中实现彩色图像(三通道)转灰度图像(单通道),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
彩色图像(三通道)转灰度图像(单通道)
彩色图转灰度图
from torchvision import transforms
transform = transforms.Compose([
transforms.Grayscale(num_output_channels=1), # 彩色图像转灰度图像num_output_channels默认1
transforms.ToTensor()
])彩色图(三通道)转指定R,G,B通道
def change_image_channels(image):
# 3通道转单通道
if image.mode == 'RGB':
r, g, b = image.split()
return r,g,b实现图像三通道转为单通道的方法
解决方法
from tensorflow.keras.preprocessing.image import img_to_array, load_img img = load_img(img_path, target_size=(48, 48), color_mode=“grayscale”) img = img_to_array(img) # 灰度化
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python Pillow Image.save 保存为jpg图片压缩问题
Pillow 库支持多种图片格式,Pillow 能够很轻松地实现图片格式之间的转换。本文就来详细的介绍一下Image.save的具体使用,感兴趣的可以了解一下2021-11-11
使用 OpenAI API 和 Python 使用 GPT-3的操作方法
这篇文章主要介绍了使用 OpenAI API 和 Python 使用 GPT-3,在本文中,我们将使用 GPT-3。我将向您展示如何访问它,并提供一些示例来说明您可以使用它做什么,以及您可以使用它构建什么样的应用程序,需要的朋友可以参考下2023-03-03


最新评论