commit | 4428b12a34e374fe7e320cd93b22c34ac74b6a60 | [log] [tgz] |
---|---|---|
author | Koan-Sin Tan <koansin.tan@gmail.com> | Sun Aug 16 14:49:44 2020 +0800 |
committer | Koan-Sin Tan <koansin.tan@gmail.com> | Sun Aug 16 14:49:44 2020 +0800 |
tree | 8494428d149767e416a253a70b48f7686b682a6f | |
parent | 0d93baff069942da48f792d3ed20785e37bd673f [diff] |
allow gpu delegate to run quant model in tflite demo
diff --git a/tensorflow/lite/java/demo/app/src/main/java/com/example/android/tflitecamerademo/Camera2BasicFragment.java b/tensorflow/lite/java/demo/app/src/main/java/com/example/android/tflitecamerademo/Camera2BasicFragment.java index bb62b44..562f6c5 100644 --- a/tensorflow/lite/java/demo/app/src/main/java/com/example/android/tflitecamerademo/Camera2BasicFragment.java +++ b/tensorflow/lite/java/demo/app/src/main/java/com/example/android/tflitecamerademo/Camera2BasicFragment.java
@@ -368,12 +368,7 @@ classifier.setNumThreads(numThreads); if (device.equals(cpu)) { } else if (device.equals(gpu)) { - if (model.equals(mobilenetV1Quant)) { - showToast("gpu requires float model."); - classifier = null; - } else { classifier.useGpu(); - } } else if (device.equals(nnApi)) { classifier.useNNAPI(); }