머신러닝

본문 바로가기
사이트 내 전체검색


머신러닝
머신러닝

5. SavedModel 포맷 사용하기

페이지 정보

작성자 관리자 댓글 0건 조회 1,825회 작성일 21-01-03 14:14

본문

5. SavedModel 포맷 사용하기

SavedModel에는 가중치 및 연산을 포함한 완전한 텐서플로 프로그램이 포함됩니다. 

기존에 설계했던 모델 코드를 실행할 필요가 없어 공유하거나  배포하는 데 유용합니다.


미리 훈련된 케라스 모델을 내보내고 그 모델로 이미지 분류 요청을 처리해보겠습니다. 

1.PNG


2.PNG


실행 예제로 그레이스 호퍼(Grace Hopper)의 이미지와 사용이 쉬운 케라스 사전 훈련 이미지 분류 모델을 사용할 것입니다. 



3.PNG


4.PNG


5.PNG


이 이미지의 가장 가능성 있는 예측은 "군복"입니다.



6.PNG


저장 경로의 마지막 경로 요소(여기서는 1/)는 모델의 버전 번호인 텐서플로 서빙(TensorFlow Serving) 컨벤션을 따릅니다

텐서플로 서빙과 같은 도구가 최신 모델을 구분할 수 있게 합니다.


SavedModel은 시그니처(signatures)라 불리는 함수를 가집니다. 

케라스 모델은 serving_default 시그니처 키를 사용하여 정방향 패스(forward pass)를 내보냅니다. 

SavedModel 커맨드 라인 인터페이스는 디스크에 저장된 SavedModel을 검사할 때 유용합니다.


$ saved_model_cli show --dir /tmp/mobilenet/1 --tag_set serve --signature_def serving_default



파이썬에서 tf.saved_model.load로 SavedModel을 다시 불러오고 해군대장 호퍼(Admiral Hopper)의 이미지가 어떻게 분류되는지 볼 수 있습니다.



7.PNG


가져온 시그니처는 항상 딕셔너리를 반환합니다.



8.PNG



SavedModel로부터 추론을 실행하면 처음 모델과 같은 결과를 제공합니다.



9.PNG


댓글목록

등록된 댓글이 없습니다.


개인정보취급방침 서비스이용약관 모바일 버전으로 보기 상단으로

TEL. 063-469-4551 FAX. 063-469-4560 전북 군산시 대학로 558
군산대학교 컴퓨터정보공학과

Copyright © www.leelab.co.kr. All rights reserved.