自己编写了dataset类 在__getitem__(self, idx)里打开图片。
方法1:img_dataset = PlateDataset(......) data_loader = DataLoader(img_dataset,num_work=4.....)
方法2:img_dataset = PlateDataset(......) img_list = [img for img in img_dataset] data_loader = DataLoader(img_list,num_work=4.....)
两种方法的训练效果差别较大,请问这是什么原因?已经知道方法2是将所有的图片都加载到了内存里,方法1没有。
求助!pytorch使用dataset和dataloader加载图片出现问题
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-