텐서플로 허브 사용

사전 훈련된 모델을 이용하는 또 다른 방법으로 텐서플로 허브(TensorFlow Hub)가 있습니다.

텐서플로 허브는 일반화된 문제들에 대해 모델의 재사용성을 극대화하려고 구글에서 공개한 API입니다. 즉, 텐서플로 허브는 모델에서 재사용 가능한 부분을 게시, 검색, 사용하기 위한 API를 제공합니다.

텐서플로 허브를 사용하려면 tensorflow_hub 라이브러리를 설치해야 합니다.

> pip install tensorflow_hub

다음은 텐서플로 허브를 이용한 코드입니다. 먼저 필요한 라이브러리를 호출하고, ResNet50을 내려받습니다.

내려받은 ResNet50 합성곱층의 가중치를 고정하고 분류를 위해 새로운 계층(소프트맥스 활성화 함수가 추가된 밀집층)을 추가합니다.

▲ 그림 5-35 계층을 고정하고 분류를 위해 새로운 계층 추가