torchvision datasets to jpg

torchvision 有很多整理好的 datasets。例如 MNIST。
如果要還原回 jpg 檔。

from torchvision import datasets
from PIL import Image

mysets = datasets.MNIST('./mnist')
idx, classes = next(enumerate(mysets))

這樣就取出一個 data，可以看 type，就知道 clases 是 (PIL Image, 數字)
所以把 PIL Image save 就可以。

classes[0].save('aaa.jpg')

如果要讓 dataloader 可以 load，就要依照規定folder structure 來放。