본문 바로가기

전이학습3

[Tensorflow] Image Classification Part 4 - Fine Tuning Image Classification - Fine Tuning (미세조정) 딥러닝에서 파인튜닝(FIne Tuning)이란? Pre-Trained 모델의 파라미터를 목적에 맞게 미세하게 조정하는 방법을 의미합니다. 모델의 특정 층(layer)을 동결(freezing)하고 새로 추가한 층과 함께 재학습시킵니다. Part3에서 학습한 전이학습 모델 중 DenseNet121과 MobileNet이 가장 유의미한 결과를 나타 냈기에, 두 모델을 기반으로 Fine Tuning을 진행해 보았습니다. Outline: 1. 데이터셋 다운로드 : 33개 자동차 모델 (현대, 기아, 제네시스, 쌍용, 르노) 데이터셋 다운로드 후 경로지정 2. 데이터 분할 : 지정한 경로로 부터 학습용 데이터(Training data)와 평가.. 2023. 4. 13.
[Tensorflow] Image Classification Part 3 - Transfer Learning2 Image Classification - Transfer Learning(전이학습) 다양한 전이학습 모델 중 이미지 분류에 적합한 4가지 CNN 모델을 테스트해 보았습니다. DenseNet121, MobileNet, ResNet50, VGGNet16 1. 데이터셋 다운로드 : 33개 자동차 모델 (현대, 기아, 제네시스, 쌍용, 르노) 데이터셋 다운로드 후 경로지정 2. 데이터 분할 : 지정한 경로로 부터 학습용 데이터(Training data)와 평가용 데이터(Valididation data)를 분리 3. 하이퍼 파라미터 설정: Epochs, Batch size, Learning rate 설정 4. 전이학습 : 사전학습 된 DenseNet121, MobileNet, ResNet50, VGGNet16 모.. 2023. 4. 7.
[Tensorflow] Image Classification Part 2 - Transfer Learning Image Classification - Transfer Learning 이미지 분류에서 전이학습(Transfer Learning)을 활용한 CNN 모델링(Modeling) 및 학습결과를 분석해 보겠습니다. 우선 전이학습이란? 방대한 데이터를 기반으로 사전 훈련된 모델(Pre-Trained Model)의 가중치와, bias를 불러와 새로 학습시킬 모델에 적용해 보는 기계학습 방법 중 하나입니다. 다양한 전이학습 모델 들 중 MobileNet을 기반으로 테스트해 보았습니다. 다음으로, layer 및 hyper parameter를 customize 하여 CNN 이미지 분류 모델을 설계해 보았습니다. 1. 데이터셋 다운로드 : 33개 자동차 모델 (현대, 기아, 제네시스, 쌍용, 르노) 데이터셋 다운로드 후 .. 2023. 3. 29.

티스토리툴바