TorchVision : PyTorch 공식 홈페이지에서 제공하는 컴퓨터 비전 전용 라이브러리

TorchVision 

PyTorch 공식 홈페이지에서 제공하는 컴퓨터 비전 전용 라이브러리

https://github.com/pytorch/vision?utm_source=chatgpt.com

GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

 

PyTorch가 딥러닝 프레임워크 자체라면 

TorchVision은 이미지처리, 데이터셋제공, 사전학습모델, 이미지변환기능, 객체탐지 분할 모델 등 비전에 필요한 기능을 모아놓은 패키지이다 .

https://standout.tistory.com/1743

PyTorch란? : PyTorch는 딥러닝 계산을 수행하는 프레임워크, GPU가 필요한 까닭, 코랩 Colab에서 확인하

 

 

TorchVision이 이미지 읽기, 크기변경, tensor변환, 정규화, 뎅터셋 다운로드, cnn모델을 구현을 해줌으로 

개발자가 위 작업을 직접 작성하지않아도 된다. 

https://standout.tistory.com/1744

CNN 이란?: CNN Concolutional Neural Network 합성곱 신경망, 이미지의 특징을 자동으로 찾자!

 

 

 

TorchVision에는 유명한 CNN 모델들이 포함되어있다. AlexNet, VGG16, ResNet, Inception, DenseNet, EfficientNet...

학습된 이 모델들을 가져다 사용할 수 있고

https://standout.tistory.com/1773

ResNet, 역전파를 그대로 사용하되 Residual Connection을 추가한 CNN

 

 

유명한 데이터셋을 쉽게 다운로드 할 수 있어 MNIST, Fashion-MNIST, CIFAR-10, CIFAR-100, ImageNet...

https://standout.tistory.com/1769

ImageNet이란?: WordNet 명사개념마다 실제 이미지를 연결한 데이터셋.

 

 

 

 

 

딥러닝모델은 이미지를 바로 입력받지못해 크기조정 및 tensor변환 정규화를 거쳐야하는데 이를 쉽게해준다. 

from torchvision import transforms

transform = transforms.Compose([
    transforms.Resize((224,224)),
    transforms.ToTensor(),
])