pytorch中实现彩色图像(三通道)转灰度图像(单通道)
作者：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) # 灰度化
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。