qq_63682631的博客

原创 自定义ImageFolder和DataLoader

输入到collate_fn(batch)中，随后，collate_fn(batch)的返回值即为next(iter(train_dl))的返回值。DataLoader根据batch_size大小，每次得到一组长度为batch_size的index，根据index调用dataset.__getitem__()方法，将返回的结果存在列表中，最后将列表作为输入参数传递给collate_fn，collate_fn的返回值即为每次next(iter(train_dl))的返回值。第二个元素是该图片的路径，即。