@yichuan9527 why not use pickle.load? If you want to use it with ImageFolder just provide pickle.load as load function.
@Veril yes, if you think there’s a dataset that we don’t have but it would be useful, please open an issue in the vision repo.
@ncullen93 all the transforms work only on PIL images, but you can use ToPILImage() to convert a tensor back to a PIL image.