이미지 분류 모델을 학습을 할 때는 레이어를 직접 짜는 방법도 있지만,,요즘 분류 task는 성능이 거의 완벽하다 할 수 있을만큼 사전학습 모델들이 잘 나와있습니다. PyTorch의 torchvision.models 라이브러리를 사용하면 다양한 사전 학습된 분류 모델을 불러와 학습에 활용할 수 있습니다. 이번 글에서는 torchvision에서 제공하는 주요 모델들의 종류와 각 모델의 파라미터 개수를 정리해보겠습니다. 현재 저는 3개의 클래스로 구성된 정말 간단한 이미지 분류 모델을 학습해서 테스트해보고 있는데 이미지 클래스간의 차이가 명확하다보니 최대한! 가볍고! 빠르게! 인퍼런스 할 수 있는 모델은 뭐가 있을까 테스트해보고 있어요. (성능은 모두 잘 나올 것이라 확신 ㅎㅎ..) 그래서 하나둘씩 모..