2021
03-05
03-05
Pytorch自定义Dataset和DataLoader去除不存在和空数据的操作
【源码GitHub地址】:点击进入1.问题描述之前写了一篇关于《pytorchDataset,DataLoader产生自定义的训练数据》的博客,但存在一个问题,我们不能在Dataset做一些数据清理,如果我们传递给Dataset数据,本身存在问题,那么迭代过程肯定出错的。比如我把很多图片路径都传递给Dataset,如果图片路径都是正确的,且图片都存在也没有损坏,那显然运行是没有问题的;但倘若传递给Dataset的图片路径有些图片是不存在,这时你通过Datase...
继续阅读 >