반응형 DataLoader2 [이미지 분류] Data Processing 오늘은 Modeling하기 전 단계인 Data Processing에서 할 수 있는 간단한 스킬을 알아볼 것이다. 이미지 데이터의 경우, 파라미터 수를 조정하기 위해 데이터를 resize하거나 over-fitting을 방지하기 위한 작업으로 이미지를 회전하거나 자르거나 noise를 추가하거나 색상을 변경하는 등의 augmentation기법을 주로 사용한다. 아래에서 하나씩 살펴보자. ■ Bounding box Bounding box는 불필요한 정보로부터 중요한 정보를 추출하기 위한 작업이다. ■ Resize 데이터의 size가 클 경우, parameter수가 많아지므로 resize를 통해 적절한 size로 조정할 수 있다. ■ Data Augmentation Data에 noise를 추가함으로써 over-f.. 2021. 8. 24. [Pytorch] Dataset , DataLoader 이번 포스팅에서는 Data를 받아서 모델에 데이터를 넘겨주기 전에 데이터를 처리하는 클래스인 Dataset과DataLoader에 대해 알아볼 것이다. Dataset클래스는 Data를 원하는 형식으로 출력하도록 하며, DataLoader클래스는 Data를 효율적으로 사용할 수 있도록 한다. 이 두 가지는 서로 다른 기능을 가지므로 개별적인 클래스로 정의한다. 먼저 Data를 load한 후 가장 먼저 설계하는 Dataset클래스 부터 살펴보자. 1. Dataset 클래스 · 데이터 입력 형태 정의 · 데이터 입력 방식 표준화 · Image, Text, Audio 등에 따른 입력정의 import torch from torch.utils.data import Dataset class CustomDataset(D.. 2021. 8. 22. 이전 1 다음 728x90 반응형
