수달이네 기술 블로그

12. Ailen vs Predator 데이터셋(전이학습으로 해결하기)

영화 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..
AI공부/딥러닝 2026. 2. 12. 17:41
이전 Prev 1 Next 다음