[PyTorch] DataLoader 이해하기 — Part 1

Dataset & DataLoader 작성법 (Unlabeled)

사용할 데이터 파일들의 경로를 리스트에 담기
이미지 데이터의 전처리
나만의 데이터셋 클래스
최종 DataLoader 작성

Dataset & DataLoader 작성 (Labeled)

STL-10 데이터셋 (https://cs.stanford.edu/~acoates/stl10/)
ImageFolder & DataLoader 작성 예시
이미지 데이터의 자료형과 라벨을 반환하는 데이터셋 클래스

What’s next?

--

--

--

AI researcher at Promedius Inc. Especially interested in theoretical physics, mathematics and deep learning.

Love podcasts or audiobooks? Learn on the go with our new app.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Seungmin Ha

Seungmin Ha

AI researcher at Promedius Inc. Especially interested in theoretical physics, mathematics and deep learning.

More from Medium

Step by step guide to setup Tensorflow with GPU support on windows 10

Lights on - PyTorch

torch

Write your first Multi-node GPU training script with PyTorch using SLURM and Singularity.

IBM super computer

From scratch to CUDA installation and TensorFlow compilation from the sources on Ubuntu 20.04