在android / java中向图形添加自定义操作

时间:2018-11-28 19:53:50

标签: android tensorflow-serving tensorflow-lite

我已经在Tensorflow中(在python环境中)实现并训练了更快的RCNN架构的修改版。我还生成了一个序列化的冻结模型(作为.pb文件)。我的实现包括一个自定义操作(使用tf.py_func函数实现)。现在,我想使用TensorFlowInferenceInterface类在android应用程序上运行模型,问题是自定义Op没有序列化,我想知道是否有办法在Android实现中重现它。 >

先谢谢了。

0 个答案:

没有答案