diff options
Diffstat (limited to 'tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java')
-rw-r--r-- | tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java b/tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java index e498c9e28f..2f16ded6c2 100644 --- a/tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java +++ b/tensorflow/examples/android/src/org/tensorflow/demo/Classifier.java @@ -17,7 +17,6 @@ package org.tensorflow.demo; import android.graphics.Bitmap; import android.graphics.RectF; - import java.util.List; /** @@ -44,10 +43,8 @@ public interface Classifier { */ private final Float confidence; - /** - * Optional location within the source image for the location of the recognized object. - */ - private final RectF location; + /** Optional location within the source image for the location of the recognized object. */ + private RectF location; public Recognition( final String id, final String title, final Float confidence, final RectF location) { @@ -73,6 +70,10 @@ public interface Classifier { return new RectF(location); } + public void setLocation(RectF location) { + this.location = location; + } + @Override public String toString() { String resultString = ""; |