├── app ├── .gitignore ├── src │ ├── main │ │ ├── assets │ │ │ ├── real.pb │ │ │ └── test.jpg │ │ ├── res │ │ │ ├── values │ │ │ │ ├── strings.xml │ │ │ │ ├── colors.xml │ │ │ │ └── styles.xml │ │ │ ├── mipmap-hdpi │ │ │ │ ├── ic_launcher.png │ │ │ │ └── ic_launcher_round.png │ │ │ ├── mipmap-mdpi │ │ │ │ ├── ic_launcher.png │ │ │ │ └── ic_launcher_round.png │ │ │ ├── mipmap-xhdpi │ │ │ │ ├── ic_launcher.png │ │ │ │ └── ic_launcher_round.png │ │ │ ├── mipmap-xxhdpi │ │ │ │ ├── ic_launcher.png │ │ │ │ └── ic_launcher_round.png │ │ │ ├── mipmap-xxxhdpi │ │ │ │ ├── ic_launcher.png │ │ │ │ └── ic_launcher_round.png │ │ │ ├── mipmap-anydpi-v26 │ │ │ │ ├── ic_launcher.xml │ │ │ │ └── ic_launcher_round.xml │ │ │ ├── layout │ │ │ │ └── activity_main.xml │ │ │ ├── drawable-v24 │ │ │ │ └── ic_launcher_foreground.xml │ │ │ └── drawable │ │ │ │ └── ic_launcher_background.xml │ │ ├── AndroidManifest.xml │ │ └── java │ │ │ └── com │ │ │ └── example │ │ │ └── whensunset │ │ │ └── tensorflowdemos │ │ │ └── MainActivity.java │ ├── test │ │ └── java │ │ │ └── com │ │ │ └── example │ │ │ └── whensunset │ │ │ └── tensorflowdemos │ │ │ └── ExampleUnitTest.java │ └── androidTest │ │ └── java │ │ └── com │ │ └── example │ │ └── whensunset │ │ └── tensorflowdemos │ │ └── ExampleInstrumentedTest.java ├── proguard-rules.pro └── build.gradle ├── settings.gradle ├── .gitignore ├── gradle └── wrapper │ ├── gradle-wrapper.jar │ └── gradle-wrapper.properties ├── gradle.properties ├── gradlew.bat └── gradlew /app/.gitignore: -------------------------------------------------------------------------------- 1 | /build 2 | -------------------------------------------------------------------------------- /settings.gradle: -------------------------------------------------------------------------------- 1 | include ':app' 2 | -------------------------------------------------------------------------------- /app/src/main/assets/real.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/assets/real.pb -------------------------------------------------------------------------------- /app/src/main/assets/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/assets/test.jpg -------------------------------------------------------------------------------- /app/src/main/res/values/strings.xml: -------------------------------------------------------------------------------- 1 | 2 | TensorflowDemos 3 | 4 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | *.iml 2 | .gradle 3 | /local.properties 4 | .DS_Store 5 | /build 6 | /captures 7 | .externalNativeBuild 8 | /.idea 9 | -------------------------------------------------------------------------------- /gradle/wrapper/gradle-wrapper.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/gradle/wrapper/gradle-wrapper.jar -------------------------------------------------------------------------------- /app/src/main/res/mipmap-hdpi/ic_launcher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-hdpi/ic_launcher.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-mdpi/ic_launcher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-mdpi/ic_launcher.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xhdpi/ic_launcher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xhdpi/ic_launcher.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xxhdpi/ic_launcher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xxhdpi/ic_launcher.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xxxhdpi/ic_launcher.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xxxhdpi/ic_launcher.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-hdpi/ic_launcher_round.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-hdpi/ic_launcher_round.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-mdpi/ic_launcher_round.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-mdpi/ic_launcher_round.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xhdpi/ic_launcher_round.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xhdpi/ic_launcher_round.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xxhdpi/ic_launcher_round.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xxhdpi/ic_launcher_round.png -------------------------------------------------------------------------------- /app/src/main/res/mipmap-xxxhdpi/ic_launcher_round.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whenSunSet/TensorflowPureDemo/HEAD/app/src/main/res/mipmap-xxxhdpi/ic_launcher_round.png -------------------------------------------------------------------------------- /app/src/main/res/values/colors.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | #3F51B5 4 | #303F9F 5 | #FF4081 6 | 7 | -------------------------------------------------------------------------------- /gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- 1 | #Sat May 19 14:14:30 CST 2018 2 | distributionBase=GRADLE_USER_HOME 3 | distributionPath=wrapper/dists 4 | zipStoreBase=GRADLE_USER_HOME 5 | zipStorePath=wrapper/dists 6 | distributionUrl=https\://services.gradle.org/distributions/gradle-4.1-all.zip 7 | -------------------------------------------------------------------------------- /app/src/main/res/mipmap-anydpi-v26/ic_launcher.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /app/src/main/res/mipmap-anydpi-v26/ic_launcher_round.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /app/src/main/res/values/styles.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 10 | 11 | 12 | -------------------------------------------------------------------------------- /app/src/test/java/com/example/whensunset/tensorflowdemos/ExampleUnitTest.java: -------------------------------------------------------------------------------- 1 | package com.example.whensunset.tensorflowdemos; 2 | 3 | import org.junit.Test; 4 | 5 | import static org.junit.Assert.*; 6 | 7 | /** 8 | * Example local unit test, which will execute on the development machine (host). 9 | * 10 | * @see Testing documentation 11 | */ 12 | public class ExampleUnitTest { 13 | @Test 14 | public void addition_isCorrect() throws Exception { 15 | assertEquals(4, 2 + 2); 16 | } 17 | } -------------------------------------------------------------------------------- /app/src/main/res/layout/activity_main.xml: -------------------------------------------------------------------------------- 1 | 2 | 8 | 9 | 13 | 14 | 18 | 19 | 20 | -------------------------------------------------------------------------------- /gradle.properties: -------------------------------------------------------------------------------- 1 | # Project-wide Gradle settings. 2 | 3 | # IDE (e.g. Android Studio) users: 4 | # Gradle settings configured through the IDE *will override* 5 | # any settings specified in this file. 6 | 7 | # For more details on how to configure your build environment visit 8 | # http://www.gradle.org/docs/current/userguide/build_environment.html 9 | 10 | # Specifies the JVM arguments used for the daemon process. 11 | # The setting is particularly useful for tweaking memory settings. 12 | org.gradle.jvmargs=-Xmx1536m 13 | 14 | # When configured, Gradle will run in incubating parallel mode. 15 | # This option should only be used with decoupled projects. More details, visit 16 | # http://www.gradle.org/docs/current/userguide/multi_project_builds.html#sec:decoupled_projects 17 | # org.gradle.parallel=true 18 | -------------------------------------------------------------------------------- /app/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- 1 | 2 | 4 | 5 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | 21 | -------------------------------------------------------------------------------- /app/proguard-rules.pro: -------------------------------------------------------------------------------- 1 | # Add project specific ProGuard rules here. 2 | # You can control the set of applied configuration files using the 3 | # proguardFiles setting in build.gradle. 4 | # 5 | # For more details, see 6 | # http://developer.android.com/guide/developing/tools/proguard.html 7 | 8 | # If your project uses WebView with JS, uncomment the following 9 | # and specify the fully qualified class name to the JavaScript interface 10 | # class: 11 | #-keepclassmembers class fqcn.of.javascript.interface.for.webview { 12 | # public *; 13 | #} 14 | 15 | # Uncomment this to preserve the line number information for 16 | # debugging stack traces. 17 | #-keepattributes SourceFile,LineNumberTable 18 | 19 | # If you keep the line number information, uncomment this to 20 | # hide the original source file name. 21 | #-renamesourcefileattribute SourceFile 22 | -------------------------------------------------------------------------------- /app/src/androidTest/java/com/example/whensunset/tensorflowdemos/ExampleInstrumentedTest.java: -------------------------------------------------------------------------------- 1 | package com.example.whensunset.tensorflowdemos; 2 | 3 | import android.content.Context; 4 | import android.support.test.InstrumentationRegistry; 5 | import android.support.test.runner.AndroidJUnit4; 6 | 7 | import org.junit.Test; 8 | import org.junit.runner.RunWith; 9 | 10 | import static org.junit.Assert.*; 11 | 12 | /** 13 | * Instrumented test, which will execute on an Android device. 14 | * 15 | * @see Testing documentation 16 | */ 17 | @RunWith(AndroidJUnit4.class) 18 | public class ExampleInstrumentedTest { 19 | @Test 20 | public void useAppContext() throws Exception { 21 | // Context of the app under test. 22 | Context appContext = InstrumentationRegistry.getTargetContext(); 23 | 24 | assertEquals("com.example.whensunset.tensorflowdemos", appContext.getPackageName()); 25 | } 26 | } 27 | -------------------------------------------------------------------------------- /app/build.gradle: -------------------------------------------------------------------------------- 1 | apply plugin: 'com.android.application' 2 | 3 | android { 4 | compileSdkVersion 26 5 | defaultConfig { 6 | applicationId "com.example.whensunset.tensorflowdemos" 7 | minSdkVersion 15 8 | targetSdkVersion 26 9 | versionCode 1 10 | versionName "1.0" 11 | testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner" 12 | } 13 | buildTypes { 14 | release { 15 | minifyEnabled false 16 | proguardFiles getDefaultProguardFile('proguard-android.txt'), 'proguard-rules.pro' 17 | } 18 | } 19 | } 20 | 21 | dependencies { 22 | implementation fileTree(dir: 'libs', include: ['*.jar']) 23 | implementation 'com.android.support:appcompat-v7:26.1.0' 24 | implementation 'com.android.support.constraint:constraint-layout:1.1.0' 25 | testImplementation 'junit:junit:4.12' 26 | androidTestImplementation 'com.android.support.test:runner:1.0.2' 27 | androidTestImplementation 'com.android.support.test.espresso:espresso-core:3.0.2' 28 | 29 | compile 'org.tensorflow:tensorflow-android:+' 30 | compile 'io.reactivex.rxjava2:rxjava:2.1.10' 31 | compile 'io.reactivex.rxjava2:rxandroid:2.0.2' 32 | compile 'com.tbruyelle.rxpermissions2:rxpermissions:0.9.3@aar' 33 | } 34 | -------------------------------------------------------------------------------- /app/src/main/res/drawable-v24/ic_launcher_foreground.xml: -------------------------------------------------------------------------------- 1 | 7 | 12 | 13 | 19 | 22 | 25 | 26 | 27 | 28 | 34 | 35 | -------------------------------------------------------------------------------- /gradlew.bat: -------------------------------------------------------------------------------- 1 | @if "%DEBUG%" == "" @echo off 2 | @rem ########################################################################## 3 | @rem 4 | @rem Gradle startup script for Windows 5 | @rem 6 | @rem ########################################################################## 7 | 8 | @rem Set local scope for the variables with windows NT shell 9 | if "%OS%"=="Windows_NT" setlocal 10 | 11 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 12 | set DEFAULT_JVM_OPTS= 13 | 14 | set DIRNAME=%~dp0 15 | if "%DIRNAME%" == "" set DIRNAME=. 16 | set APP_BASE_NAME=%~n0 17 | set APP_HOME=%DIRNAME% 18 | 19 | @rem Find java.exe 20 | if defined JAVA_HOME goto findJavaFromJavaHome 21 | 22 | set JAVA_EXE=java.exe 23 | %JAVA_EXE% -version >NUL 2>&1 24 | if "%ERRORLEVEL%" == "0" goto init 25 | 26 | echo. 27 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 28 | echo. 29 | echo Please set the JAVA_HOME variable in your environment to match the 30 | echo location of your Java installation. 31 | 32 | goto fail 33 | 34 | :findJavaFromJavaHome 35 | set JAVA_HOME=%JAVA_HOME:"=% 36 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe 37 | 38 | if exist "%JAVA_EXE%" goto init 39 | 40 | echo. 41 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% 42 | echo. 43 | echo Please set the JAVA_HOME variable in your environment to match the 44 | echo location of your Java installation. 45 | 46 | goto fail 47 | 48 | :init 49 | @rem Get command-line arguments, handling Windowz variants 50 | 51 | if not "%OS%" == "Windows_NT" goto win9xME_args 52 | if "%@eval[2+2]" == "4" goto 4NT_args 53 | 54 | :win9xME_args 55 | @rem Slurp the command line arguments. 56 | set CMD_LINE_ARGS= 57 | set _SKIP=2 58 | 59 | :win9xME_args_slurp 60 | if "x%~1" == "x" goto execute 61 | 62 | set CMD_LINE_ARGS=%* 63 | goto execute 64 | 65 | :4NT_args 66 | @rem Get arguments from the 4NT Shell from JP Software 67 | set CMD_LINE_ARGS=%$ 68 | 69 | :execute 70 | @rem Setup the command line 71 | 72 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar 73 | 74 | @rem Execute Gradle 75 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS% 76 | 77 | :end 78 | @rem End local scope for the variables with windows NT shell 79 | if "%ERRORLEVEL%"=="0" goto mainEnd 80 | 81 | :fail 82 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of 83 | rem the _cmd.exe /c_ return code! 84 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1 85 | exit /b 1 86 | 87 | :mainEnd 88 | if "%OS%"=="Windows_NT" endlocal 89 | 90 | :omega 91 | -------------------------------------------------------------------------------- /gradlew: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | ############################################################################## 4 | ## 5 | ## Gradle start up script for UN*X 6 | ## 7 | ############################################################################## 8 | 9 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. 10 | DEFAULT_JVM_OPTS="" 11 | 12 | APP_NAME="Gradle" 13 | APP_BASE_NAME=`basename "$0"` 14 | 15 | # Use the maximum available, or set MAX_FD != -1 to use that value. 16 | MAX_FD="maximum" 17 | 18 | warn ( ) { 19 | echo "$*" 20 | } 21 | 22 | die ( ) { 23 | echo 24 | echo "$*" 25 | echo 26 | exit 1 27 | } 28 | 29 | # OS specific support (must be 'true' or 'false'). 30 | cygwin=false 31 | msys=false 32 | darwin=false 33 | case "`uname`" in 34 | CYGWIN* ) 35 | cygwin=true 36 | ;; 37 | Darwin* ) 38 | darwin=true 39 | ;; 40 | MINGW* ) 41 | msys=true 42 | ;; 43 | esac 44 | 45 | # Attempt to set APP_HOME 46 | # Resolve links: $0 may be a link 47 | PRG="$0" 48 | # Need this for relative symlinks. 49 | while [ -h "$PRG" ] ; do 50 | ls=`ls -ld "$PRG"` 51 | link=`expr "$ls" : '.*-> \(.*\)$'` 52 | if expr "$link" : '/.*' > /dev/null; then 53 | PRG="$link" 54 | else 55 | PRG=`dirname "$PRG"`"/$link" 56 | fi 57 | done 58 | SAVED="`pwd`" 59 | cd "`dirname \"$PRG\"`/" >/dev/null 60 | APP_HOME="`pwd -P`" 61 | cd "$SAVED" >/dev/null 62 | 63 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar 64 | 65 | # Determine the Java command to use to start the JVM. 66 | if [ -n "$JAVA_HOME" ] ; then 67 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then 68 | # IBM's JDK on AIX uses strange locations for the executables 69 | JAVACMD="$JAVA_HOME/jre/sh/java" 70 | else 71 | JAVACMD="$JAVA_HOME/bin/java" 72 | fi 73 | if [ ! -x "$JAVACMD" ] ; then 74 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME 75 | 76 | Please set the JAVA_HOME variable in your environment to match the 77 | location of your Java installation." 78 | fi 79 | else 80 | JAVACMD="java" 81 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 82 | 83 | Please set the JAVA_HOME variable in your environment to match the 84 | location of your Java installation." 85 | fi 86 | 87 | # Increase the maximum file descriptors if we can. 88 | if [ "$cygwin" = "false" -a "$darwin" = "false" ] ; then 89 | MAX_FD_LIMIT=`ulimit -H -n` 90 | if [ $? -eq 0 ] ; then 91 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then 92 | MAX_FD="$MAX_FD_LIMIT" 93 | fi 94 | ulimit -n $MAX_FD 95 | if [ $? -ne 0 ] ; then 96 | warn "Could not set maximum file descriptor limit: $MAX_FD" 97 | fi 98 | else 99 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT" 100 | fi 101 | fi 102 | 103 | # For Darwin, add options to specify how the application appears in the dock 104 | if $darwin; then 105 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\"" 106 | fi 107 | 108 | # For Cygwin, switch paths to Windows format before running java 109 | if $cygwin ; then 110 | APP_HOME=`cygpath --path --mixed "$APP_HOME"` 111 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"` 112 | JAVACMD=`cygpath --unix "$JAVACMD"` 113 | 114 | # We build the pattern for arguments to be converted via cygpath 115 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null` 116 | SEP="" 117 | for dir in $ROOTDIRSRAW ; do 118 | ROOTDIRS="$ROOTDIRS$SEP$dir" 119 | SEP="|" 120 | done 121 | OURCYGPATTERN="(^($ROOTDIRS))" 122 | # Add a user-defined pattern to the cygpath arguments 123 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then 124 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)" 125 | fi 126 | # Now convert the arguments - kludge to limit ourselves to /bin/sh 127 | i=0 128 | for arg in "$@" ; do 129 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -` 130 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option 131 | 132 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition 133 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"` 134 | else 135 | eval `echo args$i`="\"$arg\"" 136 | fi 137 | i=$((i+1)) 138 | done 139 | case $i in 140 | (0) set -- ;; 141 | (1) set -- "$args0" ;; 142 | (2) set -- "$args0" "$args1" ;; 143 | (3) set -- "$args0" "$args1" "$args2" ;; 144 | (4) set -- "$args0" "$args1" "$args2" "$args3" ;; 145 | (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;; 146 | (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;; 147 | (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;; 148 | (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;; 149 | (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;; 150 | esac 151 | fi 152 | 153 | # Split up the JVM_OPTS And GRADLE_OPTS values into an array, following the shell quoting and substitution rules 154 | function splitJvmOpts() { 155 | JVM_OPTS=("$@") 156 | } 157 | eval splitJvmOpts $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS 158 | JVM_OPTS[${#JVM_OPTS[*]}]="-Dorg.gradle.appname=$APP_BASE_NAME" 159 | 160 | exec "$JAVACMD" "${JVM_OPTS[@]}" -classpath "$CLASSPATH" org.gradle.wrapper.GradleWrapperMain "$@" 161 | -------------------------------------------------------------------------------- /app/src/main/res/drawable/ic_launcher_background.xml: -------------------------------------------------------------------------------- 1 | 2 | 7 | 10 | 15 | 20 | 25 | 30 | 35 | 40 | 45 | 50 | 55 | 60 | 65 | 70 | 75 | 80 | 85 | 90 | 95 | 100 | 105 | 110 | 115 | 120 | 125 | 130 | 135 | 140 | 145 | 150 | 155 | 160 | 165 | 170 | 171 | -------------------------------------------------------------------------------- /app/src/main/java/com/example/whensunset/tensorflowdemos/MainActivity.java: -------------------------------------------------------------------------------- 1 | package com.example.whensunset.tensorflowdemos; 2 | 3 | import android.Manifest; 4 | import android.content.Context; 5 | import android.graphics.Bitmap; 6 | import android.graphics.BitmapFactory; 7 | import android.os.Bundle; 8 | import android.os.Environment; 9 | import android.support.v7.app.AppCompatActivity; 10 | import android.text.TextUtils; 11 | import android.util.Log; 12 | import android.widget.ImageView; 13 | 14 | import com.tbruyelle.rxpermissions2.Permission; 15 | import com.tbruyelle.rxpermissions2.RxPermissions; 16 | 17 | import org.tensorflow.Operation; 18 | import org.tensorflow.contrib.android.TensorFlowInferenceInterface; 19 | 20 | import java.io.File; 21 | import java.io.FileOutputStream; 22 | import java.io.IOException; 23 | import java.io.InputStream; 24 | import java.util.Iterator; 25 | 26 | import io.reactivex.functions.Consumer; 27 | import io.reactivex.functions.Predicate; 28 | 29 | public class MainActivity extends AppCompatActivity { 30 | 31 | private static final String MODEL_FILE = "file:///android_asset/real.pb"; 32 | private static final String INPUT_NODE = "padsss:0"; 33 | private static final String OUTPUT_NODE = "squeezesss:0"; 34 | private TensorFlowInferenceInterface inferenceInterface; 35 | private float[] floatValues; 36 | private float[] floatValuess; 37 | private int[] intValues; 38 | private int[] intValuess; 39 | private ImageView mImageView; 40 | private ImageView mImageView1; 41 | private int mInWidth = 800; 42 | private int mInHeight = 600; 43 | 44 | private int mOutWidth = 780; 45 | private int mOutHeight = 580; 46 | 47 | String CACHE_DIRECTORY; 48 | String FONT_EDIT_VIEW_IMAGE; 49 | @Override 50 | protected void onCreate(Bundle savedInstanceState) { 51 | super.onCreate(savedInstanceState); 52 | setContentView(R.layout.activity_main); 53 | 54 | // 申请权限 55 | RxPermissions rxPermissions = new RxPermissions(this); 56 | rxPermissions.requestEach( 57 | Manifest.permission.READ_EXTERNAL_STORAGE , 58 | Manifest.permission.WRITE_EXTERNAL_STORAGE) 59 | .filter(new Predicate() { 60 | @Override 61 | public boolean test(Permission permission) throws Exception { 62 | return permission.granted; 63 | } 64 | }).subscribe(new Consumer() { 65 | @Override 66 | public void accept(Permission permission) throws Exception { 67 | //权限申请成功后 将assets中准备好的图片写到sd卡中,以便后面使用 68 | CACHE_DIRECTORY = getCacheDirectory(getBaseContext(), "").getPath(); 69 | FONT_EDIT_VIEW_IMAGE = CACHE_DIRECTORY + "/test.jpg"; 70 | assetToFile("test.jpg", FONT_EDIT_VIEW_IMAGE); 71 | 72 | //开始处理图片 73 | make(); 74 | } 75 | }); 76 | 77 | 78 | } 79 | 80 | private void make() { 81 | // 从文件中读取需要处理的图片为 bitmap 82 | BitmapFactory.Options options = new BitmapFactory.Options(); 83 | options.inPreferredConfig = Bitmap.Config.ARGB_8888; 84 | options.outHeight = mInHeight; 85 | options.outWidth = mInWidth; 86 | Bitmap bitmap = BitmapFactory.decodeFile(FONT_EDIT_VIEW_IMAGE, options); 87 | mImageView = (ImageView) findViewById(R.id.image); 88 | mImageView.setImageBitmap(bitmap); 89 | 90 | // 因为 神经网络需要传入float数组,所以需要将bitmap中所有的像素的真实值传入到 float数组中 91 | floatValues = new float[mInHeight * mInWidth * 3]; 92 | floatValuess = new float[mOutHeight * mOutWidth * 3]; 93 | intValues = new int[mInHeight * mInWidth]; 94 | intValuess = new int[mOutHeight * mOutWidth]; 95 | 96 | // 将 mInHeight * mInWidth 这么大的图片像素传入到intValues中 97 | bitmap.getPixels(intValues, 0, bitmap.getWidth(), 0, 0, mInWidth, mInHeight); 98 | 99 | // intValues中的每一个值都是一个 将 A R G B四个通道整合之后的值,所以这里需要将四个通道分离,然后将其中的R G B三个通道写入到floatValues中 100 | // 所以floatValues的大小是 mInHeight * mInWidth * 3 101 | for (int i = 0; i < intValues.length; ++i) { 102 | final int val = intValues[i]; 103 | floatValues[i * 3] = ((val >> 16) & 0xFF); 104 | floatValues[i * 3 + 1] = ((val >> 8) & 0xFF); 105 | floatValues[i * 3 + 2] = (val & 0xFF); 106 | } 107 | 108 | // 创建一个TensorFlow在Java下的实例,这里只要将 神经网络文件 放到assets中,然后让Tensorflow自动读取就好了 109 | inferenceInterface = new TensorFlowInferenceInterface(getAssets(), MODEL_FILE); 110 | 111 | // 获取一下 神经网络的每个节点 然后输出一下名字,有助于确认输入和输出节点 112 | Iterator operationIterator = inferenceInterface.graph().operations(); 113 | while (operationIterator.hasNext()){ 114 | Operation operation = operationIterator.next(); 115 | Log.d("MainActivitysss", operation.name()); 116 | } 117 | 118 | // 将刚刚获取到的floatValues数组当做输入节点输入到Tensorflow实例中,参数依次是 节点名字,数据,dims:表示将floatValues转换成 1 * mInHeight * mInWidth * 3的张量 119 | inferenceInterface.feed(INPUT_NODE, floatValues, 1, mInHeight, mInWidth, 3); 120 | // 输入输出节点的名字,并运行,这里是阻塞当前线程的,所以在正式项目中需要在其他线程运行,这里我就简单一点在主线程运行。 121 | inferenceInterface.run(new String[] {OUTPUT_NODE}, true); 122 | // 运行完毕之后,取出经过神经网络处理的数据 123 | inferenceInterface.fetch(OUTPUT_NODE, floatValuess); 124 | 125 | // 将floatValuess 整合成Bitmap中需要的像素值 126 | for (int i = 0; i < intValuess.length; ++i) { 127 | intValuess[i] = 128 | 0xFF000000 129 | | (((int) (floatValuess[i * 3])) << 16) 130 | | (((int) (floatValuess[i * 3 + 1])) << 8) 131 | | ((int) (floatValuess[i * 3 + 2])); 132 | } 133 | 134 | // 将构建好的像素值 存回Bitmap中 135 | Bitmap b = Bitmap.createBitmap(mOutWidth, mOutHeight, Bitmap.Config.ARGB_8888); 136 | b.setPixels(intValuess, 0, b.getWidth(), 0, 0, b.getWidth(), b.getHeight()); 137 | mImageView1 = (ImageView) findViewById(R.id.image1); 138 | mImageView1.setImageBitmap(b); 139 | } 140 | 141 | public void assetToFile(String imageName, String filePath){ 142 | InputStream ims = null; 143 | FileOutputStream fileOutputStream = null; 144 | try { 145 | // get input stream 146 | ims = getAssets().open(imageName); 147 | File file = new File(filePath); 148 | if(file.exists()){ 149 | file.delete(); 150 | } 151 | fileOutputStream = new FileOutputStream(filePath); 152 | byte[] buffer = new byte[512]; 153 | int count = 0; 154 | while((count = ims.read(buffer)) > 0){ 155 | fileOutputStream.write(buffer, 0 ,count); 156 | } 157 | fileOutputStream.flush(); 158 | } catch (IOException e) { 159 | throw new RuntimeException("将asset写入文件失败"); 160 | } finally { 161 | try { 162 | if (fileOutputStream != null && ims != null) { 163 | fileOutputStream.close(); 164 | ims.close(); 165 | } 166 | } catch (IOException e) { 167 | e.printStackTrace(); 168 | } 169 | } 170 | 171 | } 172 | 173 | public static File getCacheDirectory(Context context, String type) { 174 | File appCacheDir = getExternalCacheDirectory(context,type); 175 | if (appCacheDir == null){ 176 | appCacheDir = getInternalCacheDirectory(context,type); 177 | } 178 | 179 | if (appCacheDir == null){ 180 | Log.e("getCacheDirectory","getCacheDirectory fail ,the reason is mobile phone unknown exception !"); 181 | }else { 182 | if (!appCacheDir.exists()&&!appCacheDir.mkdirs()){ 183 | Log.e("getCacheDirectory","getCacheDirectory fail ,the reason is make directory fail !"); 184 | } 185 | } 186 | return appCacheDir; 187 | } 188 | 189 | private static File getExternalCacheDirectory(Context context,String type) { 190 | File appCacheDir = null; 191 | if( Environment.MEDIA_MOUNTED.equals(Environment.getExternalStorageState())) { 192 | if (TextUtils.isEmpty(type)){ 193 | appCacheDir = context.getExternalCacheDir(); 194 | }else { 195 | appCacheDir = context.getExternalFilesDir(type); 196 | } 197 | 198 | if (appCacheDir == null){// 有些手机需要通过自定义目录 199 | appCacheDir = new File(Environment.getExternalStorageDirectory(),"Android/data/"+context.getPackageName()+"/cache/"+type); 200 | } 201 | 202 | if (appCacheDir == null){ 203 | Log.e("getExternalDirectory","getExternalDirectory fail ,the reason is sdCard unknown exception !"); 204 | }else { 205 | if (!appCacheDir.exists()&&!appCacheDir.mkdirs()){ 206 | Log.e("getExternalDirectory","getExternalDirectory fail ,the reason is make directory fail !"); 207 | } 208 | } 209 | }else { 210 | Log.e("getExternalDirectory","getExternalDirectory fail ,the reason is sdCard nonexistence or sdCard mount fail !"); 211 | } 212 | return appCacheDir; 213 | } 214 | 215 | private static File getInternalCacheDirectory(Context context,String type) { 216 | File appCacheDir = null; 217 | if (TextUtils.isEmpty(type)){ 218 | appCacheDir = context.getCacheDir();// /data/data/app_package_name/cache 219 | }else { 220 | appCacheDir = new File(context.getFilesDir(),type);// /data/data/app_package_name/files/type 221 | } 222 | 223 | if (!appCacheDir.exists()&&!appCacheDir.mkdirs()){ 224 | Log.e("getInternalDirectory","getInternalDirectory fail ,the reason is make directory fail !"); 225 | } 226 | return appCacheDir; 227 | } 228 | 229 | 230 | } 231 | --------------------------------------------------------------------------------