1.将RGB 3通道的数据和RGIG3通道的数据合并为IRRGB4通道的数据
关键代码:将IR通道的数据添加到原来的RGB通道
BGRIRArray = np.append(RGBArray,IRRGArray[:,:,0].reshape(600,600,1),axis=2)
import os
import numpy as np
from PIL import Image
# fileTrainRGB = open("/media/mlxuan/winD/DeepLearningData/ISPRS_BENCHMARK_DATASETS/Potsdam/txtForTrain/trainRGB.txt",'r')
with open("/media/mlxuan/winD/DeepLearningData/ISPRS_BENCHMARK_DATASETS/Potsdam/txtForTrain/testRGB.txt", "r") as f:
line = f.readline()
while line:
#获得文件路径
RGBimgPath = os.path.join("/media/mlxuan/winD/DeepLearningData/ISPRS_BENCHMARK_DATASETS/Potsdam/image",line.split()[0])
IRRGimgPath = os.path.join("/media/mlxuan/winD/DeepLearningData/ISPRS_BENCHMARK_DATASETS/Potsdam/imageIRRG",line.split()[0].replace("RGB","IRRG"))
#读取文件
RGBimg = Image.open(RGBimgPath)
RGBArray = np.array(RGBimg, dtype=np.uint8)
IRRGimg = Image.open(IRRGimgPath)
IRRGArray = np.array(IRRGimg, dtype=np.uint8)
#通道合并:将IR通道的数据合并到RGB通道中
BGRIRArray = np.append(RGBArray,IRRGArray[:,:,0].reshape(600,600,1),axis=2)
BGRIRImg = Image.fromarray(BGRIRArray)
BGRIRImg.save(os.path.join("/media/mlxuan/winD/DeepLearningData/ISPRS_BENCHMARK_DATASETS/Potsdam/imageBGRIR",line.split()[0].replace("RGB","BGRIR")))
line = f.readline()