tensorflow.python.framework.errors_impl.InvalidArgumentError: image

本文链接：https://blog.csdn.net/siriusli1981/article/details/88793097

tensorflow.python.framework.errors_impl.InvalidArgumentError: image_size must contain 3 elements[4]

出现了很多非RGB格式的图片。
解决方法：
由于该类图片较多，所以选择将它们转化为RGB格式之后再储存。

#encoding:utf-8
from PIL import Image
import os
import matplotlib.pyplot as plt

def get_not_rgb_images(rootdir):
    list = os.listdir(rootdir)
    l = len(list)
    count = 0
    for i in range(0, len(list)):
        filename = os.path.join(rootdir, list[i])
        # print(filename)
        if os.path.isfile(filename):
            img = Image.open(filename)
            pixels = img.getpixel((0, 0))

            if type(pixels) == int:
                print('单通道:' + filename)
                count += 1
                print('The ',i,'st pic:',count,'/',l,'\r')                
                #img = Image.open(imgName)
                img = img.convert('RGB')
                img.save(filename)
                
            elif type(pixels) == tuple:
                if  len(pixels) != 3:
                    print('非RGB的多通道:' +filename)
                    count += 1
                    print('The ',i,'st pic:',count,'/',l,'\r')
                    img = img.convert('RGB')
                    img.save(filename)
        else:
            get_not_rgb_images(filename)
            
if __name__ == '__main__':
    rootdir = '/yourfile'#您图片所在的文件夹
    get_not_rgb_images(rootdir )