[Pytorch] torchvision

https://tutorials.pytorch.kr/beginner/basics/data_tutorial.html

Dataset과 Dataloader — PyTorch Tutorials 1.9.0+cu102 documentation

Pytorch

• open source machine learning framework
  
  
  

torchvision

• Pytorch 프로젝트의 일부
• 컴퓨터 비전에 대한 다양한 datasets, model architectures, 그리고 common image transformations로 구성됨

 

torchvision.transforms

• 보편적인 image transformations
• Compose를 이용하여 여러 transform 연결 가능 (이런 transform은 torchscript를 지원하지 않음)

transforms.Compose([
    transforms.CenterCrop(10),
    transforms.ToTensor(),
])

• 대부분의 transformations는 PIL images와 tensor images를 받아들임
  • PIL image
    • Python Imaging Library
  • tensor images 
    • 배열이나 행렬과 매우 유사한 특수한 자료구조 
    • Pytorch에서 모델의 input, output, parameter들을 부호화(encode)함. 
    • (C, H, W) shape를 가진 tensor
      • C : # of channels
      • H : height
      • W : width
    • batch of Tensor Images : (B, C, H, W) shape의 tensor
      • B : # of images in the batch

 

• Scriptable transforms
• Compositions of transforms
• Transformation on PIL Image and torch *Tensor
• Transformation on PIL Image only
• Transformation on torch *Tensor only
• Conversion Transforms
• Generic Transforms
• AutoAugment Transforms
• Functional Transforms

 

데이터 전처리

1. Data Cleaning