Skip to content

Android conversion #15

@e-papa

Description

@e-papa

Hello,
first of all thank you for your code.

I successfully used it to train a model and style some images using a GPU.
Then, I would like to port the model on an Android device and try it there, but unfortunately I wasn't able to proceed.

I see the pbtxt file is missing from the directory where the checkpoints are saved so I wasn't able to proceed to freeze the graph and optimize it for mobile. I tried modifying your files to make them save the pbtxt file but I'm not sure I did it correctly.
I added
tf.train.write_graph(self.sess.graph_def, ".", "graph.pbtxt")
just before
res = saver.save(self.sess,self.save_path+'/final.ckpt')
in style_transfer_trainer.py.

The saved graph file is probably way too big (~650MB), and anyway, using the summarize_graph from tensorflow I get 211 possible outputs, so unfortunately I don't know then how to freeze and optimize it.

Could you give me some advices? Thank you.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions