wdq92500 2019-12-22 10:05 采纳率: 50%
浏览 1532
已结题

求助!pytorch使用dataset和dataloader加载图片出现问题

自己编写了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没有。

  • 写回答

1条回答 默认 最新

  • threenewbee 2019-12-22 10:10
    关注

    训练集不一样,效果自然不同,两者没有可比较性。理论上训练集越大,效果越好。

    评论

报告相同问题?