图像3通道检测-Python

基于python的图像3通道检测，使用openCV-cv2实现，运行结果输出非3通道图片路径。

import os
import numpy as np
import cv2if __name__ == '__main__':root_dir = r'D:\...'label = []label_an = []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.shapeif size[2] != 3:print("通道"+ str(size[2]) + file)# os.remove(file)