kendryte / tensorflow-workspace

Tensorflow workspaces for training your custom models of K210
66 stars 26 forks source link

Problem on using nncase to transfer tflite to C code #3

Open TMaysGGS opened 5 years ago

TMaysGGS commented 5 years ago

Hi,

I was following the instructions from https://github.com/kendryte/tensorflow-workspace/commit/d80f0d2ac23287aaf8b116c9cd3769b90c720bee#diff-7d9fc065a89438c0ca7e01e4611d0f3aR30 to generate C code from .pb file and I was using the mobilenetv1.pb file from the example. I just tried the steps from the solutions and am stuck in the following step:

.\ncc.exe -i tflite -o k210code --dataset .\dataset\ .\bin\mobilenetv1.tflite .\bin\mobilenetv1.c It shows: Fatal: output-format.

I checked the output-format options and 'k210code' has been modified to 'k210model' since v0.1.0 RC1. So I chose 'k210model' and ran it again: .\ncc.exe -i tflite -o k210model --dataset .\dataset\ .\bin\mobilenetv1.tflite .\bin\mobilenetv1.c and then I got a .kmodel file rather than the .c file.

I wonder if anyone can help me with that. Thanks a lot.

TMaysGGS

xhsonny commented 5 years ago

+1. Is there a plan to release a version of tensorflow-workspace that work with kmodel format?