/tensorflow-2-run-on-mobile-devices-ios-android-browser

Tutorial to run TensorFlow 2 on mobile devices: Android, iOS and Browser

Primary LanguageSwiftMIT LicenseMIT

Run TensorFlow 2 on any device

header

This project walks you through the process of creating an emotion recognition app on Android, iOS and in the browser:

  • Load data and train model
  • Export as SavedModel
  • Run on Android using TensorFlow Lite
  • Run on iOS using Core ML
  • Run in the browser using TensorFlow.js

Demo

header

A demo for the browser version is available here

Running in the browser

Folder: tfjs

Install yarn and run:

cd tfjs
yarn
yarn watch

Running on iOS

Folder: coreml_ios You must have Xcode and Cocoa Pods installed as well as a mac computer.

cd coreml_ios/app
pod install
open Core\ ML\ Demo.xcworkspace

Running on Android

It is recommended to have Android Studio installed. Import the project and run the app.