Is it possible to train custom models for Tensorflow Android?

aymericdamien / TensorFlow-Examples

TensorFlow Tutorial and Examples for Beginners (support TF v1 & v2)

Other

43.41k stars 14.94k forks source link

Is it possible to train custom models for Tensorflow Android? #308

Open nijatmursali opened 5 years ago

nijatmursali commented 5 years ago

Hello,

Currently, I'm working on one projects and I need to train my models. I want to train models using YoloV2 and my question is that will my custom trained models work on Android Tensorflow if I add .pb model into the assets folder?

gkkhut commented 4 years ago

How to Train Your Own Custom Model with Tensorflow Object Detection API and Deploy It into Android with TF Lite by Shahed Alali Alghrsi

Follow this link. It will give you a complete pipeline for the process

https://medium.com/datadriveninvestor/how-to-train-your-own-custom-model-with-tensorflow-object-detection-api-and-deploy-it-into-android-aeacab7fa76f