Youngjung youngjung
youngjung
/ cifar10_to_imagefolder.py
Created
April 18, 2019 07:17
create an imagefolder and save images in subdirs with classnames in cifar10
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import argparse | |
| import numpy as np | |
| from PIL import Image | |
| from tqdm import tqdm | |
| import torch | |
| import torchvision | |
| import torchvision.transforms as transforms |