WeLoveDoctor

dataloader 에서 파일 이름 알아내기

Written by

byun1114

in

GPU도 없고, 오래된 느린 CPU를 가진 녀석이지만 기본 코딩은 할 수 있을 것이라는 생각으로 다시 코딩하기 시작함. 일단 dataloader에서 파일명을 알아야 할 때에는 다음과 같이 할 수 있다.

dataloaders = torch.utils.data.DataLoader(
  torchvision.datasets.ImageFolder(
  root='/home/byun1114/JUPYTER/DATA', 
transform=data_transforms))

보통 이와 유사하게 코딩을 하게 된다. torch.utils.data.DataLoader의 기본 옵션은 batch_size는 1이고 shuffle=False 라는 것을 알아야 한다. 하나씩 비교할 것이기 때문에 불필요한 옵션은 생략하였다. 이런 식으로 코딩을 한다면 파일명은 다음과 같이 확인한다.

dataloaders.dataset.imgs

More posts