수달이네 기술 블로그

전이학습(Transfer Learning)이미 학습된 모델을 새로운 문제에 적용하여 학습 시간을 단축시키고, 성능을 향상시키는 방법, 기존 모델이 학습한 특징으로 새로운 데이터셋의 모델의 일부만 다시 학습하거나 추가 학습(Fine-tuning)을 진행함.리소스가 제한된 상황에서 효과적인 방식이미지 분류, 자연어 처리 등의 다양한 분야에서 사용즉, 다른 문제를 해결하며 얻은 지식으로 원하는 문제를 해결함.이미지넷이전에 ImageNet을 보았는데, 대회에서만 사용한 데이터셋이 아닌, 사전학습에서 중요한 역할을 하며, 딥러닝 연구의 표준 벤치마크이다.1400만장의 이미지 + 22000개의 카테고리Alexnethttps://docs.pytorch.org/vision/main/models/generated/tor..

영화 Ailen과 Predator 캐릭터를 분류 이미지 데이터셋이다.이걸 CNN 변형 중 Alexnet을 이용해 만들어 볼 것이다.이번에 Alexnet구현이 아닌, 전이학습을 배우는데에 초점을 둔다.import torchimport torch.nn as nnimport torch.optim as optimimport matplotlib.pyplot as pltfrom torchvision import datasets, models, transformsfrom torch.utils.data import DataLoader기초적으로 사용한 모듈들device = 'cuda' if torch.cuda.is_available() else 'cpu'print(device)device세팅transformsdata_t..