tensorflow를 활용한 이미지인식 모바일 어플리케이션 제작-3

지난 글에서 이미지를 학습시켜 결과물로 .pb 확장자의 그래프파일과 텍스트파일(라벨)을 만들었습니다.

하지만, 이렇게 만들어진 파일을 바로 모바일에서 읽을 수가 없습니다. 모바일에서는 적용할 수 없는 op(operation)가 일부 들어가 있기 때문이라고합니다.

그래서 모바일에서 지원하지 않는 op들을 제거해주는 (예를들면 DecodeJpeg 같은거) 작업을 해줘야 하는데 그게 바로 optimize_for_inference 라는 툴이 하는 역할입니다.

1. 학습된 그래프파일을 최적화하기

위와 같이 명령을 실행해줍니다.

2.Quantize 하기

위의 과정보다 더 작은 크기의 파일로 압축합니다.

위의 과정을 모두다 실행하고 나시면, 폴더에 파일들이 아래와 같이 생성되었을겁니다.

새로 생성된 pb 파일들을 이용해 학습테스트를 해보셔도 결과가 잘 나오실겁니다.