search

NJU-Jet / SR_Mobile_Quantization

Winner solution of mobile AI (CVPRW 2021).
148 stars 41 forks source link

How to convert tflite model to onnx model? #1

Open xiaoxiongli opened 3 years ago

xiaoxiongli commented 3 years ago

it is good work and congratulation!

I have to two questions that: 1) which framwork do you use: tensorflow or tensorflow_lite? 2) if you use tensorflow_lite, How to convert the output tflite model to onnx model?

please help... thank you very much! double click 666 ^_^

NJU-Jet commented 3 years ago

Hello:)

  1. We use tensorflow during training, and after training we convert the saved model to .tflite model.
  2. I uploaded .pb file in experiment directory just now, so you can convert it to onnx model.
simplew2011 commented 1 year ago

python -m tf2onnx.convert --saved-model experiment/base7_D4C28_bs16ps64_lr1e-3/best_status/ --output model.onnx --opset 11 --verbose --inputs-as-nchw serving_default_input_1:0

error： image

simplew2011 commented 1 year ago

python -m tf2onnx.convert --saved-model experiment/base7_D4C28_bs16ps64_lr1e-3/best_status/ --output model.onnx --opset 11 --verbose --inputs-as-nchw serving_default_input_1:0

error： image

change tf 2.3.1 is ok.