flyakon
commented
10 months ago 不好意思，才看见。这个可能是因为拷贝到服务器的数据，没有拷贝完全，导致读取错误。你可以check一下原始图片是不是都能正确读取。另外，第一次训练的时候可以先用少量数据跑通再大规模训练
Open zhangpeng2001 opened 10 months ago
flyakon
commented
10 months ago 不好意思，才看见。这个可能是因为拷贝到服务器的数据，没有拷贝完全，导致读取错误。你可以check一下原始图片是不是都能正确读取。另外，第一次训练的时候可以先用少量数据跑通再大规模训练
zhangpeng2001
commented
10 months ago 你好，感谢收到您的回复，在这个路径(CloudMattingGAN/CloudMatting/models/matting/cloud_gan_matting_net.py)下的第138行的try中，这里要求bg_img，thick_img，thin_img的图像数量一样吧？但是我从谷歌云和百度云下载的数据集均没有相同的图像，其中bg_img，thick_img，cloud_img分别包含392、808、162张图像。请问有完整的数据集吗？
------------------ 原始邮件 ------------------ 发件人: "flyakon/Pytorch-CloudMattingGAN" @.>; 发送时间: 2023年10月23日(星期一) 下午5:45 @.>; @.**@.>; 主题: Re: [flyakon/Pytorch-CloudMattingGAN] 训练报错 (Issue #2)
不好意思，才看见。这个可能是因为拷贝到服务器的数据，没有拷贝完全，导致读取错误。你可以check一下原始图片是不是都能正确读取。另外，第一次训练的时候可以先用少量数据跑通再大规模训练
— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.
Message ID: @.***>
flyakon
commented
10 months ago 数量不需要一致，这个就是全部的数据了。如果再有问题，你可以邮箱联系我，这个可能收不到消息
你好，请问一下，这个报错是因为什么？ data = [self.dataset[idx] for idx in possibly_batched_index] File "/data2/msc/zp/papers/CloudMattingGAN/CloudMatting/dataset/cloud_matting_dataset.py", line 118, in getitem bg_img=self.bg_dataset.getitem(idx)[0] File "/data2/msc/zp/papers/CloudMattingGAN/CloudMatting/dataset/cloud_matting_dataset.py", line 95, in getitem item_result.extend(self.get_data(item)) File "/data2/msc/zp/papers/CloudMattingGAN/CloudMatting/dataset/cloud_matting_dataset.py", line 87, in get_data img = self.transforms(img) File "/data2/msc/zp/papers/CloudMattingGAN/CloudMatting/transform/matting_transforms.py", line 248, in call img = t(img) File "/data2/msc/zp/papers/CloudMattingGAN/CloudMatting/transform/matting_transforms.py", line 81, in call img=F.hflip(img) File "/data1/msc/softwares/anaconda3/envs/cloudmatting/lib/python3.8/site-packages/torchvision/transforms/functional.py", line 458, in hflip return F_pil.hflip(img) File "/data1/msc/softwares/anaconda3/envs/cloudmatting/lib/python3.8/site-packages/torchvision/transforms/functional_pil.py", line 54, in hflip return img.transpose(Image.FLIP_LEFT_RIGHT) File "/data1/msc/softwares/anaconda3/envs/cloudmatting/lib/python3.8/site-packages/PIL/Image.py", line 2791, in transpose self.load() File "/data1/msc/softwares/anaconda3/envs/cloudmatting/lib/python3.8/site-packages/PIL/ImageFile.py", line 266, in load raise OSError(msg) OSError: image file is truncated (26 bytes not processed)