[Tensorflow]Tornado + Tensorflow 이미지 추론 검색 서버 (프로젝트 눈치)

개요

---

Python Tornado와 tensorflow 연동하여 간단한 비동기 이미지 추론 검색 서버를 만들었다.

Tornado web server 에서 클라이언트의 요청을 받고 Tensorflow로 이미지를 추론 하여, 추론 결과 문자로 구글 검색을 한다.

API 서버를 목적으로 만들었지만 기본 웹페이지 까지 했다. 기존의 서버가 있다면, 웹 페이지를 쓰지 않고 추론하는 API 서버로 활용해도 된다.

스크린샷

---

@ 이미지 추론 메인 페이지

'파일 선택' 버튼을 클릭 후 파일을 업로드(PC)하거나 촬영(Mobile) 할 수 있다

@ 파일을 업로드 후 결과 값으로 구글 이미지 검색으로 연결 된다

소스

---

https://github.com/pipy201/nunchi

@ 문제점

클라이언트의 요청 마다 tensorflow session을 만든다...;; 그로 인해 요청이 짧은 시간에 많이 오면, 사용 메모리가 급격하게 증가한다.

어느정도 시간이 지나면 운영체제가 메모리를 정리 하긴 하지만 큰 문제다.

tornado 서버 시작 할 때 session을 열고 클라이언트 요청시 handler에 넘겨주면 해결 될 것으로 보인다.