[CNN] 자동차모델분류 전이학습 (Transfer Learning) -2

코드 훈련데이터는 이미지들을 변형시켜서 데이터양을 늘리고 리사이즈를 한다. 검증데이터와 테스트 데이터는 이미지 변형을 시키지 않고 리사이즈만 한다. 결과 tochvision models 에서 regnet 불러오기 불러온 모델의 출력층을 보면 1000개의 클래스가 출력이 된다. 불러온 모델은 원래 1000가지의 다른 훈련데이터로 학습한 모델이라 출력 클래스가 1000개 이다. 전이학습을 위해 ...

딥러닝CNNAITranferLearningAI