이 문제는 ‘Torchvision’을 사용합니다.
Torchvision은 데이터 셋, 아키텍처 모델, 이미지 변환 기능으로 구성되어 있는 패키지 입니다.
Torchvision의 모델은 Alexnet, VGG, ResNet, SqueezeNet, DenseNet, Inception v3, GoogLeNet이 있습니다.
본 문제의 출제자님께서는 VGG를 사용하셨는데, 다양한 모델을 적용해보면 좋을 거 같아서 토론으로 글을 남깁니다.
- 참고 : Torchvision.tranforms는 Transforms on PIL Image, Transforms on torch.*Tensor, Conversion Transforms, Generic Transforms, Functional Transforms가 있습니다. 이 또한 다양하게 적용이 가능할 거 같습니다:)