6 |
6 | 
288 |
289 |
290 | This library is provided _as is_, I work on it in my free time.
291 |
292 | If you're integrating react-native-fast-tflite in a production app, consider [funding this project](https://github.com/sponsors/mrousavy) and contact me to receive premium enterprise support, help with issues, prioritize bugfixes, request features, help at integrating react-native-fast-tflite and/or VisionCamera Frame Processors, and more.
293 |
294 | ## Contributing
295 |
296 | 1. Clone the repo
297 | 2. Make sure you have installed Xcode CLI tools such as `gcc`, `cmake` and `python`/`python3`. See the TensorFlow documentation on what you need exactly.
298 | 3. Run `yarn bootstrap` and select `y` (yes) on all iOS and Android related questions.
299 | 4. Open the Example app and start developing
300 | - iOS: `example/ios/TfliteExample.xcworkspace`
301 | - Android: `example/android`
302 |
303 | See the [contributing guide](CONTRIBUTING.md) to learn how to contribute to the repository and the development workflow.
304 |
305 | ## License
306 |
307 | MIT
308 |
--------------------------------------------------------------------------------
/android/CMakeLists.txt:
--------------------------------------------------------------------------------
1 | project(VisionCameraTflite)
2 | cmake_minimum_required(VERSION 3.9.0)
3 |
4 | set (CMAKE_VERBOSE_MAKEFILE ON)
5 | set (PACKAGE_NAME "VisionCameraTflite")
6 | set (CMAKE_CXX_STANDARD 17)
7 |
8 | find_package(ReactAndroid REQUIRED CONFIG)
9 | find_package(fbjni REQUIRED CONFIG)
10 |
11 | find_library(
12 | TFLITE
13 | tensorflowlite_jni
14 | PATHS "./src/main/cpp/lib/litert/jni/${ANDROID_ABI}"
15 | NO_DEFAULT_PATH
16 | NO_CMAKE_FIND_ROOT_PATH
17 | )
18 |
19 | find_library(
20 | TFLITE_GPU
21 | tensorflowlite_gpu_jni
22 | PATHS "./src/main/cpp/lib/litert/jni/${ANDROID_ABI}"
23 | NO_DEFAULT_PATH
24 | NO_CMAKE_FIND_ROOT_PATH
25 | )
26 |
27 | string(APPEND CMAKE_CXX_FLAGS " -DANDROID")
28 |
29 | add_library(
30 | ${PACKAGE_NAME}
31 | SHARED
32 | ../cpp/jsi/Promise.cpp
33 | ../cpp/jsi/TypedArray.cpp
34 | ../cpp/TensorflowPlugin.cpp
35 | ../cpp/TensorHelpers.cpp
36 | src/main/cpp/Tflite.cpp
37 | )
38 |
39 | # Specifies a path to native header files.
40 | target_include_directories(
41 | ${PACKAGE_NAME}
42 | PRIVATE
43 | "../cpp"
44 | "src/main/cpp"
45 | "src/main/cpp/lib/litert/headers"
46 | "${NODE_MODULES_DIR}/react-native/ReactCommon"
47 | "${NODE_MODULES_DIR}/react-native/ReactCommon/callinvoker"
48 | "${NODE_MODULES_DIR}/react-native/ReactAndroid/src/main/jni/react/turbomodule" # <-- CallInvokerHolder JNI wrapper
49 | )
50 |
51 | set_target_properties(${PACKAGE_NAME} PROPERTIES LINKER_LANGUAGE CXX)
52 |
53 | target_link_libraries(
54 | ${PACKAGE_NAME}
55 | android # <-- log
56 | ReactAndroid::jsi # <-- jsi.h
57 | fbjni::fbjni # <-- fbjni.h
58 | ${TFLITE}
59 | ${TFLITE_GPU}
60 | )
61 |
62 | if(ReactAndroid_VERSION_MINOR GREATER_EQUAL 76)
63 | target_link_libraries(
64 | ${PACKAGE_NAME}
65 | ReactAndroid::reactnative # <-- RN merged so
66 | )
67 | else()
68 | target_link_libraries(
69 | ${PACKAGE_NAME}
70 | ReactAndroid::reactnativejni # <-- CallInvokerImpl
71 | )
72 | endif()
73 |
--------------------------------------------------------------------------------
/android/build.gradle:
--------------------------------------------------------------------------------
1 | import java.nio.file.Paths
2 |
3 | buildscript {
4 | repositories {
5 | google()
6 | mavenCentral()
7 | }
8 |
9 | dependencies {
10 | classpath "com.android.tools.build:gradle:7.2.2"
11 | }
12 | }
13 |
14 | def isNewArchitectureEnabled() {
15 | return rootProject.hasProperty("newArchEnabled") && rootProject.getProperty("newArchEnabled") == "true"
16 | }
17 |
18 | apply plugin: "com.android.library"
19 |
20 |
21 | def appProject = rootProject.allprojects.find { it.plugins.hasPlugin('com.android.application') }
22 |
23 | if (isNewArchitectureEnabled()) {
24 | apply plugin: "com.facebook.react"
25 | }
26 |
27 | def getExtOrDefault(name) {
28 | return rootProject.ext.has(name) ? rootProject.ext.get(name) : project.properties["Tflite_" + name]
29 | }
30 |
31 | def getExtOrIntegerDefault(name) {
32 | return rootProject.ext.has(name) ? rootProject.ext.get(name) : (project.properties["Tflite_" + name]).toInteger()
33 | }
34 |
35 | static def findNodeModules(baseDir) {
36 | def basePath = baseDir.toPath().normalize()
37 | // Node's module resolution algorithm searches up to the root directory,
38 | // after which the base path will be null
39 | while (basePath) {
40 | def nodeModulesPath = Paths.get(basePath.toString(), "node_modules")
41 | def reactNativePath = Paths.get(nodeModulesPath.toString(), "react-native")
42 | if (nodeModulesPath.toFile().exists() && reactNativePath.toFile().exists()) {
43 | return nodeModulesPath.toString()
44 | }
45 | basePath = basePath.getParent()
46 | }
47 | throw new GradleException("react-native-fast-tflite: Failed to find node_modules/ path!")
48 | }
49 |
50 | def nodeModules = findNodeModules(projectDir)
51 |
52 | android {
53 | ndkVersion getExtOrDefault("ndkVersion")
54 | compileSdkVersion getExtOrIntegerDefault("compileSdkVersion")
55 | namespace "com.tflite"
56 |
57 | defaultConfig {
58 | minSdkVersion getExtOrIntegerDefault("minSdkVersion")
59 | targetSdkVersion getExtOrIntegerDefault("targetSdkVersion")
60 | buildConfigField "boolean", "IS_NEW_ARCHITECTURE_ENABLED", isNewArchitectureEnabled().toString()
61 | externalNativeBuild {
62 | cmake {
63 | arguments "-DANDROID_STL=c++_shared",
64 | "-DNODE_MODULES_DIR=${nodeModules}",
65 | "-DIS_NEW_ARCHITECTURE_ENABLED=${isNewArchitectureEnabled()}"
66 | cppFlags "-O2 -frtti -fexceptions -Wall -fstack-protector-all"
67 | abiFilters "x86", "x86_64", "armeabi-v7a", "arm64-v8a"
68 | }
69 | }
70 | }
71 | externalNativeBuild {
72 | cmake {
73 | path "CMakeLists.txt"
74 | }
75 | }
76 | packagingOptions {
77 | excludes = [
78 | "META-INF",
79 | "META-INF/**",
80 | "**/libc++_shared.so",
81 | "**/libfbjni.so",
82 | "**/libjsi.so",
83 | "**/libreactnative.so",
84 | "**/libreactnativejni.so",
85 | "**/libturbomodulejsijni.so",
86 | "**/libreact_nativemodule_core.so",
87 | "**/libtensorflowlite_jni.so",
88 | "**/libtensorflowlite_gpu_jni.so",
89 | ]
90 | }
91 | buildTypes {
92 | release {
93 | minifyEnabled false
94 | }
95 | }
96 |
97 | buildFeatures {
98 | prefab true
99 | }
100 |
101 | lintOptions {
102 | disable "GradleCompatible"
103 | }
104 |
105 | compileOptions {
106 | sourceCompatibility JavaVersion.VERSION_1_8
107 | targetCompatibility JavaVersion.VERSION_1_8
108 | }
109 |
110 | configurations {
111 | extractHeaders
112 | extractSO
113 | }
114 | }
115 |
116 | repositories {
117 | mavenCentral()
118 | google()
119 | }
120 |
121 |
122 | dependencies {
123 | // For < 0.71, this will be from the local maven repo
124 | // For > 0.71, this will be replaced by `com.facebook.react:react-android:$version` by react gradle plugin
125 | //noinspection GradleDynamicVersion
126 | implementation "com.facebook.react:react-native:+"
127 |
128 | // Tensorflow Lite .aar (includes C API via prefabs)
129 | implementation "com.google.ai.edge.litert:litert:1.0.1"
130 | extractSO("com.google.ai.edge.litert:litert:1.0.1")
131 | extractHeaders("com.google.ai.edge.litert:litert:1.0.1")
132 |
133 | // Tensorflow Lite GPU delegate
134 | implementation "com.google.ai.edge.litert:litert-gpu:1.0.1"
135 | extractSO("com.google.ai.edge.litert:litert-gpu:1.0.1")
136 | extractHeaders("com.google.ai.edge.litert:litert-gpu:1.0.1")
137 | }
138 |
139 | task extractAARHeaders {
140 | doLast {
141 | configurations.extractHeaders.files.each {
142 | def file = it.absoluteFile
143 | def packageName = file.name.tokenize('-')[0]
144 | copy {
145 | from zipTree(file)
146 | into "src/main/cpp/lib/$packageName/"
147 | include "**/*.h"
148 | }
149 | }
150 | }
151 | }
152 |
153 | task extractSOFiles {
154 | doLast {
155 | configurations.extractSO.files.each {
156 | def file = it.absoluteFile
157 | def packageName = file.name.tokenize('-')[0]
158 | copy {
159 | from zipTree(file)
160 | into "src/main/cpp/lib/$packageName/"
161 | include "jni/**/*.so"
162 | }
163 | }
164 | }
165 | }
166 |
167 | if (isNewArchitectureEnabled()) {
168 | react {
169 | jsRootDir = file("../src/")
170 | libraryName = "Tflite"
171 | codegenJavaPackageName = "com.tflite"
172 | }
173 | }
174 |
175 |
176 | def nativeBuildDependsOn(dependsOnTask) {
177 | def buildTasks = tasks.findAll({ task -> (
178 | !task.name.contains("Clean")
179 | && (task.name.contains("externalNative")
180 | || task.name.contains("CMake")
181 | || task.name.contains("generateJsonModel")
182 | )
183 | ) })
184 | buildTasks.forEach { task -> task.dependsOn(dependsOnTask) }
185 | }
186 |
187 | afterEvaluate {
188 | nativeBuildDependsOn(extractAARHeaders)
189 | nativeBuildDependsOn(extractSOFiles)
190 | }
191 |
--------------------------------------------------------------------------------
/android/gradle.properties:
--------------------------------------------------------------------------------
1 | Tflite_kotlinVersion=1.7.0
2 | Tflite_minSdkVersion=21
3 | Tflite_targetSdkVersion=31
4 | Tflite_compileSdkVersion=31
5 | Tflite_ndkversion=21.4.7075529
6 |
--------------------------------------------------------------------------------
/android/src/main/AndroidManifest.xml:
--------------------------------------------------------------------------------
1 |