/nlp_tflite

nlp模型转tensorflow lite,并使用c++调用

Primary LanguageC++

c++调用tflite

一,进入到char-rnn下面训练模型,并将模型转换成tflite。详见char-rnn/README.md

二,编译libtensorflowlite.so文件(mac下测试,linux类似)

三,进入到cpp_tflite中,引入头文件和so文件

  • 1,将tflite的头文件放置到include下(cd tensorflow/tensorflow) (find ./lite -name "*.h" | tar -cf headers.tar -T -)
  • 2,将上面的headders.tar压缩包移动到自己项目下的cpp_tflite/include中并解压,解压后可以直接去除
  • 3,将tensorflow/bazel-bin/tensorflow/lite/libtensorflowlite.so和tensorflow/bazel-bin/external/flatbuffers/libflatbuffers.a 放到cpp_tflite/lib文件夹中
  • 4,进入到tensorflow/tensorflow/lite/tools/make下执行download_dependencies.sh,或者直接下载 flatbuffers压缩包,然后取出include的内容放置到cpp_tflite/include中即可。
  • 5,编写CMakeLists.txt,链接到so文件
  • 6,mkdir build
  • 7,cd build
  • 8,cmake ..
  • 9,make
  • 10,./nlp_tflite