问题描述:
【功能模块】
我自定义myDataSet类,然后使用data = GeneratorDataset()加载数据,在使用data.map(Resize())时报错“Resize: load image failed.”
我的代码和数据在附件中给出。请问大家,我这个什么原因?
【操作步骤&问题现象】
【截图信息】
加载数据
输出是这样的
然后Resize,再输出
解答:
您好,这边经过测试,发现该程序偶尔会成功运行出结果,大概率会抛出不同的异常,目前认为应该是由于数据集的多张图片的尺寸大小不一致,进行RandomResize随机插值的时候crop出问题,请您将数据集图片尺寸先尽量调整一致后再尝试进行操作。
while True:
try:
for i in ds_train.create_dict_iterator():
print(i['image'].shape, end=' ') # (52, 58, 1) 75
break
break
except:
continue