728x90 반응형 SMALL pre-trained model2 [시각 지능] 사전 학습 모델 (Pre-Trained Model) 사전 학습 모델 (Pre-Trained Model) 사전 학습 모델이란 기존에 자비어 (Xavier) 등 임의의 값으로 초기화하던 모델의 가중치들을 다른 문제 (task)에 학습시킨 가중치들로 초기화하는 방법이다. from tensorflow.keras.applications import VGG16, ResNet50, MobileNet, InceptionV3 mobilenet_model = MobileNet(weights = 'imagenet', include_top = True, input_shape = (224, 224, 3)) mobilenet_model.summary() Model: "mobilenet_1.00_224" ___________________________________________.. 2022. 8. 20. [시각 지능] 전이 학습 (Transfer Learning) 전이 학습 (Transfer Learning) CNN 기반의 딥러닝 모델을 훈련시키려면 많은 양의 데이터가 필요하지만 큰 데이터셋을 얻 는 것은 쉽지 않다. 이러한 현실적인 어려움을 해결한 것이 전이 학습인데, 전이 학습은 ImageNet처럼 아주 큰 데이터셋을 써서 사전 학습 모델 (pre-trained model)의 가중치를 가져와 분석하려는 데이터에 맞게 보정해서 사용하는 것을 의미한다. 특징 추출기 (feature extractor)는 컨볼루션 층과 풀링 층의 조합으로 구성되어 있으며 ImageNet 데이터에 대해 이미 학습되어 있다. 분류기 (classifier)는 완전 연결 층 (Dense) 조합으로 구성되며 이미지에 대한 정답을 분류하는 역할을 한다. Cats and Dogs 고양이와 개 이.. 2022. 8. 14. 이전 1 다음 728x90 반응형 LIST
