이미지 분류 모델 구축(PyTorch)

pytorch로 이미지 분류 모델을 구축해보았습니다.

AI hub에서 제공하는 이미지 데이터셋을 활용하여 이미지 분류 태스크를 수행한다. VGG16으로 테스트하였을 때 망을 몇 개 더 쌓았더니 성능이 좋지 않아 중간에 중단하였다. 대안으로 ViT 모델을 테스트하려 한다. Vision Transformer(ViT) 모델은

이전 글 : [Pytorch] 이미지 분류 with ViT 이미지 분류를 위해 google vit 모델을 학습했습니다. 1 에폭도 다 학습하지 못하고 중단되었습니다. 데이터 증강 코드 동/식물 그림을 입력받으면, 그림의 대상을 알아 맞추는 분류문제입니다. 만약 당근

이전 글 : [Pytorch] 폴드 분리 학습 - 이미지 분류 with DeiT small > 성능을 고도화할 모델을 선정하기 위해 후보 모델들에 대한 기초적인 성능을 테스트하는 단계입니다. DeiT를 폴드 분리 학습한 결과가 나왔습니다. 저는 세 개의 폴드로 분리