Custom Dataset1 PyTorch Dataloader에서 원본 파일명을 알고 싶다. 이미지 분류를 위한 PyTorch의 사용 시에, 입력 데이터를 준비하는 가장 흔한 방법은 다음과 같다. 각 레이블 별로 이미지를 서로 다른 디렉토리에 저장한다. torchvision.datasets.ImageFolder( )를 이용하여, 폴더 구조로부터 데이터셋을 생성한다. 이 과정에서 필요한 transform을 수행하고, 텐서로 변환하며 정규화를 수행할 수 있다. torch.utils.data.DataLoader( )를 사용하여 생성된 데이터셋으로부터 데이터를 로드할 수 있다. 이 때, 원하는 크기의 배치 단위로 데이터를 로드하거나, 순서가 무작위로 섞이도록(shuffle) 할 수 있다. 이렇게 처리된 입력 데이터는 변환된 텐서와 레이블이 조합된 형태로 메모리에 로드되어 있다. 그런데 훈련된 모델에 의.. 2020. 11. 25. 이전 1 다음