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借助ChatGPT读取.env实现文件配置隔离保障私有数据安全
这篇文章主要为大家介绍了python读取.env实现文件配置隔离保障私有数据安全,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-03-03深度解析Django REST Framework 批量操作
这篇文章主要介绍了深度解析Django REST Framework批量操作,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2021-05-05
最新评论