基于python的图像3通道检测,使用openCV-cv2实现,运行结果输出非3通道图片路径。
import os
import numpy as np
import cv2
if __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.shape
if size[2] != 3:
print("通道"+ str(size[2]) + file)
# os.remove(file)