多光谱数据的处理

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()