Numpy Image File with torchvision.datasets
Deep Learning Model을 학습하려다 보면 다양한 형태의 Dataset을 마주하게 된다. 대부분의 Vision 관련 DL Model은 Dataset으로 JPG, PNG 등의 형태를 사용하게된다. 종종 Numpy Array Image를 Input으로 사용해야하는 경우가 있는데, torchvision.datasets의 경우 '.npy' 확장자를 지원하지 않아 customize를 해야한다. Customize하는 방법은 어렵지 않다. 먼저 Training Dataset을 위한 Transforms를 구현한다. 그 다음 numpy array image를 load해 PIL 확장자로 바꿔주는 함수를 선언한다. 다음으로 torchvision.datasets.DatasetFolder()를 사용해 dataset을..
