[개발환경 조사] DeepLearning on Mobile

1. Pytorch 편

현재 파이토치 공식 홈페이지에는 추론하는 예제만 제공한다.

gradle(CMake와 같은 빌드 자동화 도구이다) 명령을 통해 'org.pytorch:pytorch_android:1.4.0' 와 'org.pytorch:pytorch_android_torchvision:1.4.0'를 같이 빌드해주면, 안드로이드 에뮬레이터를 통해 추론 예시가 등장한다.

 

안드로이드 에뮬레이터를 통한 추론 예시

혹은, 어떤 분의 강의처럼 모바일을 연결하면 어플을 통해 직접 실행이 가능하다.

(본인은 강의 구매 하지 않았다)

실제 모바일에 연결하여 딥러닝 프레임워크 실행한 예시

2. Tensorflow 편

텐서플로우의 경우도 개발 과정은 비슷하다(홈페이지 참조). 모델(.tflite)을 안드로이드 유틸과 같이 빌드한다. 다만, 코틀린을 사용한다는 점에서 java로만 작성한 파이토치 데모와는 차이를 보인다.

 

Tensorflow Lite 실행 예시