Files

keras2cpp

Name		Name	Last commit message	Last commit date
parent directory ..
images		images
CMakeLists.txt		CMakeLists.txt
h5_to_pb.py		h5_to_pb.py
keras.pyproj		keras.pyproj
keras2cpp.cpp		keras2cpp.cpp
keras2cpp.vcxproj		keras2cpp.vcxproj
keras_train.py		keras_train.py
load_h5_test.py		load_h5_test.py
load_pb_test.py		load_pb_test.py
my_model_ep20.h5		my_model_ep20.h5
my_model_ep20.pb		my_model_ep20.pb
readme.md		readme.md

readme.md

将Keras训练的模型部署到C++平台上的可行方案

1.编译tensorflow

Windows下使用CMake编译，ubuntu使用bazel编译，时间很长

2.keras训练

python keras_train.py
python load_h5_test.py

3.转换为pb模型

python h5_to_pb.py
python load_pb_test.py

4.C++代码使用

见keras2cpp.cpp