训练图片通道数的问题

bashan16045

于 2020-09-28 16:44:56 发布

阅读量514

点赞数

CC 4.0 BY-SA版权

分类专栏： TensorFlow OpenCV 文章标签： python 计算机视觉 opencv tensorflow 神经网络

本文链接：https://blog.youkuaiyun.com/bashan1604/article/details/108852229

训练图片集中混入4通道图片导致问题，移除这些图片后问题解决。使用cv2检查非3通道图片，cv2是OpenCV-Python的实现，主要讨论了图片的通道概念，RGB图像由红绿蓝三个通道组成，每个像素由三个数值控制对应显示器的亮度，影响最终颜色。了解并处理不同通道数的图片对于计算机视觉和神经网络训练至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem location: 在训练的图片集中，混入了几张4通道的图片，将这几张图片删除后问题解除。

使用cv2中img.shape列出非3通道图片的文件名：

import os
import numpy as np
import cv2

if __name__ == '__main__':

    root_dir = r'D:\WData\Workspace\springdemo\images\table_images\1.0\all_table'

    name = root_dir.split('\\')[-1]

    file_names = sorted(os.listdir(root_dir))
    for file in file_names:
        if file.endswith('.png'):
            img = cv2.imdecode(np.fromfile(os.path.join(root_dir, file), dtype=np.uint8), -1)
            size = img.shape

            if size[2] != 3:
                print("通道"+ str(size[2]) + file)
                    # os.remove(file)