하루 하루

펭귄브로의 3분 딥러닝 파이토치맛 _ 2일차 본문

IT/Artificial intelligence

펭귄브로의 3분 딥러닝 파이토치맛 _ 2일차

san_deul 2020. 4. 17. 02:24

 

# Fashion MNIST 데이터셋 알아보기

 

1. 사용할 모듈 import

 

2. 이미지 - > tensor

ToTensor() 사용

더보기

# torchvision Transforms 
1. ToTensor 이미지를 파이토치 텐서로 변환 
2. Resize 이미지 크기 조정 
3. Normalize 주어진 평균과 표준편차를 이용하여 정규화 
4. RandomHorizontalFlip 무작위로 이미지의 오른쪼고가 왼쪽을 뒤집는 기능 
5. RandomCrop  이미지를 무작위로 자르는 기능 

 

3. 학습용 트레이닝 셋과 성능 평가용 테스트셋 준비

FashionMNIST 데이터셋

 

4. 'DataLoader' 준비

batch_size = 16

더보기

# DataLoader 
torchvision.datasets 으로 생성된 객체는  
파이토치 내부 클래스 'torch.utils.data.Dataset'을 상속 

-> 'DataLoader' 바로 사용가능

5. iter() / next() 반복자

더보기

#iter() 함수  
iterator는 여러개의 요소를 가지는 컨테이너(리스트, 튜플, 셋, 사전, 문자열)에서 각 요소를 하나씩 꺼내 어떤 처리를 수행할 수 있게 하는 객체  
>>> it = iter([1, 2, 3]) 
>>> next(it)  
1 
>>> next(it)  
2

 

6. 이미지 확인

 

7. 숫자 레이블 -> 텍스트 레이블

숫자 레이블

텍스트 레이블

 

8. 개별 이미지 확인

Comments