使用线上算力跑模型过程中报错,报错位置在数据集加载部分。
原因:数据集在上传过程中出错,导致图像中的像素改变。(以下图为例)
图中红色框部分像素全部为0,实际图像不是这样的。
解决方法:找到出问题的图像,并且替换掉该图像即可。
try:
iamge= Image.open(self.img_paths[index])
iamge= np.array(iamge).astype(np.uint8)
except:
print(self.img_paths[index])
在数据集加载过程中,使用try... except语句找到出问题的图像.替换掉返回的图像即可。