핵심 요약
드라마앤컴퍼니(리멤버)가 리오(RIO) 모델을 클라이언트 사이드에서 추론 가능하도록 PyTorch→ONNX→TensorFlow→TF Lite로 변환하는 ML Model Converter와 안드로이드 앱 적용기를 공개했습니다.
구현 방법
- PyTorch 모델을 TorchScript IR로 준비한 뒤 ONNX로 내보냄
- ONNX를 onnx-tensorflow 백엔드로 TensorFlow 모델로 변환
- TensorFlow 모델을 TF Lite로 변환하고 Android에서 추론 실행
- JNI/C++로 Post-Processing(차원 변환, NMS) 구현 및 OpenCV 활용
- 테스트용 데모 앱으로 모바일 환경에서 추론 검증
주요 결과
- Client-Side Computing의 이점인 실시간 추론, 오프라인 작동, 데이터 프라이버시, 비용 절감을 목표로 적용 및 테스트 진행
- TF Lite 기반으로 다양한 Android 디바이스에 이식 가능성 제시
