-
Notifications
You must be signed in to change notification settings - Fork 136
Open
Description
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
frankwmoyer, narayanan-infiswift, ucalyptus and CPLTOM
Metadata
Metadata
Assignees
Labels
No labels