diff --git a/tensorflow/examples/android/src/org/tensorflow/demo/ClassifierActivity.java b/tensorflow/examples/android/src/org/tensorflow/demo/ClassifierActivity.java index 5985010a4ff..717baeb90c7 100644 --- a/tensorflow/examples/android/src/org/tensorflow/demo/ClassifierActivity.java +++ b/tensorflow/examples/android/src/org/tensorflow/demo/ClassifierActivity.java @@ -30,7 +30,6 @@ import android.os.Trace; import android.util.Size; import android.util.TypedValue; import android.view.Display; -import java.io.IOException; import java.util.List; import java.util.Vector; import org.tensorflow.demo.OverlayView.DrawCallback; @@ -108,12 +107,25 @@ public class ClassifierActivity extends CameraActivity implements OnImageAvailab borderedText = new BorderedText(textSizePx); classifier = new TensorFlowImageClassifier(); + try { - classifier.initializeTensorFlow( - getAssets(), MODEL_FILE, LABEL_FILE, NUM_CLASSES, INPUT_SIZE, IMAGE_MEAN, IMAGE_STD, - INPUT_NAME, OUTPUT_NAME); - } catch (final IOException e) { - LOGGER.e(e, "Exception!"); + final int initStatus = + classifier.initializeTensorFlow( + getAssets(), + MODEL_FILE, + LABEL_FILE, + NUM_CLASSES, + INPUT_SIZE, + IMAGE_MEAN, + IMAGE_STD, + INPUT_NAME, + OUTPUT_NAME); + if (initStatus != 0) { + LOGGER.e("TF init status != 0: %d", initStatus); + throw new RuntimeException(); + } + } catch (final Exception e) { + throw new RuntimeException("Error initializing TensorFlow!", e); } resultsView = (ResultsView) findViewById(R.id.results); diff --git a/tensorflow/examples/android/src/org/tensorflow/demo/DetectorActivity.java b/tensorflow/examples/android/src/org/tensorflow/demo/DetectorActivity.java index 309cbbdee64..b7ee3fbef84 100644 --- a/tensorflow/examples/android/src/org/tensorflow/demo/DetectorActivity.java +++ b/tensorflow/examples/android/src/org/tensorflow/demo/DetectorActivity.java @@ -33,7 +33,6 @@ import android.os.Trace; import android.util.Size; import android.util.TypedValue; import android.view.Display; -import java.io.IOException; import java.util.LinkedList; import java.util.List; import java.util.Vector; @@ -108,19 +107,25 @@ public class DetectorActivity extends CameraActivity implements OnImageAvailable tracker = new MultiBoxTracker(getResources().getDisplayMetrics()); detector = new TensorFlowMultiBoxDetector(); + try { - detector.initializeTensorFlow( - getAssets(), - MODEL_FILE, - LOCATION_FILE, - NUM_LOCATIONS, - INPUT_SIZE, - IMAGE_MEAN, - IMAGE_STD, - INPUT_NAME, - OUTPUT_NAMES); - } catch (final IOException e) { - LOGGER.e(e, "Exception!"); + final int initStatus = + detector.initializeTensorFlow( + getAssets(), + MODEL_FILE, + LOCATION_FILE, + NUM_LOCATIONS, + INPUT_SIZE, + IMAGE_MEAN, + IMAGE_STD, + INPUT_NAME, + OUTPUT_NAMES); + if (initStatus != 0) { + LOGGER.e("TF init status != 0: %d", initStatus); + throw new RuntimeException(); + } + } catch (final Exception e) { + throw new RuntimeException("Error initializing TensorFlow!", e); } previewWidth = size.getWidth();