RGBT数据集-MFNet里面的数据是4通道的的读取方法(读取完整通道)

最新推荐文章于 2024-07-16 10:51:57 发布

计算机视觉-Archer

最新推荐文章于 2024-07-16 10:51:57 发布

阅读量982

点赞数 1

文章标签：计算机视觉 python 深度学习

本文链接：https://blog.csdn.net/zjc910997316/article/details/125678375

版权

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通道