[WIP] Export to Tensorflow Lite for Edge TPU

emedvedev / attention-ocr

A Tensorflow model for text recognition (CNN + seq2seq with visual attention) available as a Python package and compatible with Google Cloud ML Engine.

MIT License

1.08k stars 256 forks source link

[WIP] Export to Tensorflow Lite for Edge TPU #137

Open Luukjn opened 5 years ago

Luukjn commented 5 years ago

I have created an export function as per issue #136 However, the function does not work. It looks like the model is using functions that are not compatible with Tensorflow Lite. Alas I don't know enough about Tensorflow to attempt to fix this. It would be appreciated if someone else can look into this.

lynx97 commented 4 years ago

@Luukjn I cloned your fork with branch export-tensorflow-lite-edge-tpu but I cannot export to lite. Could you help me out? 2019-12-10 17:41:44.533208: F tensorflow/lite/toco/import_tensorflow.cc:1542] Check failed: data_type == DT_FLOAT Fatal Python error: Aborted