kikirizki / pytorch_to_android-ncnn

Simple working example of pytorch model deployed on android device using ncnn

Deploy Deep Learning Model to Android

Easy pipeline to deploy deep learning on android, powered by :

Road Map

Image Classification

• MobilenetV2
• ViT

Object Detection

• YoloX

Face Recogniton

• MobileFaceNet + PocketNet
• ElasticFace
• ArcFace
• MobileFaceNet + BlazeFace

Semantic Segmentation

• Yolact

Conversion Script from PyTorch to NCNN

• PocketNet
• MobileNet

Demo

BlazeFace + MobileFaceNet

BlazeFace + PocketNet 128

Reference

• https://github.com/FeiGeChuanShu/ncnn_Android_blazeface
• https://github.com/nihui/ncnn
• https://github.com/fdbtrs/PocketNet
• https://github.com/liguiyuan/mobilefacenet-ncnn
• https://github.com/nihui/opencv-mobile