遇到Premature end of JPEG file的问题
当用cv2.imread打开图片时有时会遇到这样的问题,虽然不影响网络的训练,但是会看着很心烦,这里就介绍一种方法可以让我们快速的找到有问题的照片
from PIL import Image
import os
def check_image(path):
try:
Image.open(path).load()
except:
print('ERROR: %s' % path)
return False
else:
return True
file_path = '这里输入我们图片文件夹的路径' # 注意这里最后要有”/“ ,比如 user/Destop/
for image_name in os.listdir(file_path):
check_image(file_path + image_name)
这样有问题的图片的路径就会被输出出来了