RGBT自动驾驶数据集里面使用4通道的图像保存
把其拆解为RGB和t,一张图拆2个的办法
import os
import cv2
path='/home/aaa/DL/data/data-RGBT/MFNet/images'
rgb_path='/home/aaa/DL/data/data-RGBT/MFNet/image'
t_path='/home/aaa/DL/data/data-RGBT/MFNet/t'
for img in os.listdir(path):
x=cv2.imread(os.path.join(path,img),-1) # -1 表示读出所有通道,默认只读入3个通道
rgb=x[:,:,:3]
t=x[:,:,-1]
cv2.imwrite(os.path.join(rgb_path,img),rgb)
cv2.imwrite(os.path.join(t_path,img),t)
说明
flag = -1(cv2.IMREAD_UNCHANGED), 8位深度,原通道,读入完整图片,包括alpha通道
flag = 0(cv2.IMREAD_GRAYSCALE), 8位深度,1通道 灰度通道
flag = 1(cv2.IMREAD_COLOR), 8位深度,3通道 彩色图,RGB三通的,忽略alpha通道
flag = 2, 原深度, 1通道
flag = 3, 原深度, 3通道
flag = 4, 8位深度,3通道