我使用MobileNet_v1_1.0_224 tensorflow模型进行对象检测。现在,我有了我需要转换为tflite扩展的自定义冻结图(.pb文件),这样我就可以在移动设备上使用我的模型了。我需要它们作为输入和输出参数,将冻结的图形(.pb文件)转换为tensorflow lite (.tflite)文件。
在MLKit中的Vision中,我将用我的TFLite文件替换/custom/object_labeler.tflite。我试图在绑定tflite文件但失败的代码中找到"path“。我尝试了以下操作,但是绑定my_tflite.tflite失败了。const val TAG ="ObjectDetectorProcessor"
private const val CUSTOM_MODEL_PATH = "custom_models
我需要将自定义对象检测模型导出到TensorFlow Lite,以便能够在TensorFlow设备上运行,该模型在自定义数据集上进行了微调。_tpu-8‘模型进行了精调。在检测程序中,我使用了:
##Get a tf.function for detection
@tf.function在更新到tensorflow的夜间版本(使用正常版