Dataloader (1) 썸네일형 리스트형 [이미지] CustomDataset [Python] - cv2, PIL 아직 모르는 게 많지만, 작성한 코드입니다. train 폴더 내에 label 이름별 폴더가 들어 있고, 그 내부에 이미지가 들어있는 구조입니다. test 폴더 같은 계층에 존재한다는 가정하에 향후 이용을 위해 train 폴더 상위 폴더에서 os.path.join으로 train을 더하고 os.walk로 내부 폴더와 파일명을 가져옵니다. for문을 돌려서 위치에 맞게 이미지 파일 경로와 라벨을 저장하고, 라벨은 학습을 위해 딕셔너리를 활용해 숫자로 변경합니다. 적절한 변형을 가한 후 window창을 띄워 확인합니다. import cv2 import os from PIL import Image import numpy as np from torch.utils.data import Dataset, DataLoad.. 이전 1 다음