TensorFlow Android相机演示 - 对灰度图像进行分类

时间:2018-02-15 19:34:12

标签: android tensorflow android-vision

我已经在灰度图像上训练了我自己的神经网络,现在我想从TensorFlow Android Camera Demo(https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android)将其插入分类器。

如何修改ClassifierActivity.java(https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/android/src/org/tensorflow/demo/ClassifierActivity.java)以便将灰度输入图像传递给我的计算图?

我需要用位图做一些事情,我只是不确定究竟是什么。

1 个答案:

答案 0 :(得分:0)

我建议您编辑TensorFlow图,以便在TensorFlow中进行灰度转换。它比在Android上更快,更轻松。这是TensorFlow中从rgb到灰度的转换: https://www.tensorflow.org/api_docs/python/tf/image/rgb_to_grayscale 而执行此操作的方法是在冻结图形以将其放入android设备之前,将其添加到图形中。我已经在Android上尝试过了,并且可以使用。