在训练的过程中,数据集内的图片均为.jpg格式,但是仍出现ValueError: Image format not JPEG报错,查阅了一下资料,利用代码,筛查出部分图像为.mpo(jpg图像3D格式)。
代码来源如下:
ValueError: Image format not JPEG - #10 by Nitin_Tiwari - General Discussion - TensorFlow Forum
由于个人能力有限并未找到合适的方法将.mpo图片转换为.jpg格式,就直接删除了图片。(压缩包里有转换的代码,但是并不适用我的状况)
check_image_format代码如下:
import PIL.Image
import glob
import os
images = [i for i in glob.glob("F:/zjw/ssd/VOC_WEED/VOC2012/JPEGImages/*.jpg")]
jpeg = 0
others = 0
for i in images:
img = PIL.Image.open(i)
if img.format == 'JPEG':
jpeg += 1
else:
print('{} is {}'.format(i, img.format))
others +=1
print("Total JPEG images found: ", jpeg)
print("Total other images found: ", others)
本文代码来自以上链接,如有侵权行为,请联系删除!!!