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