How to use the output model with Tensorflow lite for Mobile

[duchnguyen]

Hi,

I followed steps in description and successfully trained a new model with my images. I would like to convert the output model to the Tensorflow Lite model for Mobile usage. I followed steps here and was able to freeze the HED model to GraphDef (.pb) with output_node_names=predictions. But I don't know how to continue to convert the GraphDef model to Tensorflow Lite model (using toco) by using toco . The reason is that I don't know where to get some parameters: input_arrays, output_arrays, input_shapes, output_node_names . I also ran the tensorboard command and saw some graphs but I don't see information of those parameters.

Can you please share me know how to export the model and use it with the Tensorflow Lite?

Thanks,
Duc

[dustturtle]

any update?