这个错误是数据集中有图片格式出错打不开,只需把它删除即可。但是如何找出错误图片的位置
import os
from PIL import Image
folder_path = '' #写入你图片所在的文件夹,即包含该图片的文件夹
extensions = []
for filee in os.listdir(folder_path):
file_path = os.path.join(folder_path, filee)
print('** Path: {} **'.format(file_path), end="\r", flush=True)
im = Image.open(file_path)
rgb_im = im.convert('RGB')
if filee.split('.')[1] not in extensions:
extensions.append(filee.split('.')[1])
此时说明Cat的666.jpg格式错误,删除即可。