데이터 취업 스쿨 스터디 노트 -(90) pytorch - DataLoader(이미지 폴더에서 가져오기), Custom dataset

대부분의 회사에서는 이렇게 이미지 파일이 잘 정리되어 있는 경우가 별로 없다.

Dataloader 만들기

torch.utils.data.DataLoader를 사용함.

• torch.utils.data.DataLoader는 기본적으로 아래 두 가지 인수를 받음
  1. torch.utils.data.dataset.Dataset
  2. batch_size

Local에 있는 데이터를 데이터셋으로 만들기

데이터를 불러오기 위한 준비작업

---

이미지가 잘 정리되어있지 않은 경우 어떻게 데이터셋을 만들지 알아보자.

Custom dataset