DatasetFetcher会将Sampler取出来的索引值拿到.通过调用Dataset取出数据.而Dataset又是通过__getitem__ 方法进行索引取值的.最后将img和label经过collate_fn打包成batch
pytorch dataloader获取数据流程图
最新推荐文章于 2023-12-04 16:35:28 发布
DatasetFetcher会将Sampler取出来的索引值拿到.通过调用Dataset取出数据.而Dataset又是通过__getitem__ 方法进行索引取值的.最后将img和label经过collate_fn打包成batch