frugallydeep
Use Keras models in C++ with ease
Table of contents
Introduction
Would you like to build/train a model using Keras/Python? And would you like run the prediction (forward pass) on your model in C++ without linking your application against TensorFlow? Then frugallydeep is exactly for you.
frugallydeep
 is a small headeronly library written in modern and pure C++.
 is very easy to integrate and use.
 depends only on FunctionalPlus, Eigen and json  also headeronly libraries.
 supports inference (
model.predict
) not only for sequential models but also for computational graphs with a more complex topology, created with the functional API.  reimplements a (small) subset of TensorFlow, i.e. the operations needed to support prediction.
 results in a much smaller binary size than linking against TensorFlow.
 works out ofthebox also when compiled into a 32bit executable.
 utterly ignores even the most powerful GPU in your system and uses only one CPU core. ;)
 but is quite fast on one CPU core compared to TensorFlow.
Supported layer types
Layer types typically used in image recognition/generation are supported, making many popular model architectures possible (see Performance section).
Add
,Concatenate
,Subtract
,Multiply
,Average
,Maximum
AveragePooling1D/2D
,GlobalAveragePooling1D/2D
Conv1D/2D
,SeparableConv2D
Cropping1D/2D
,ZeroPadding1D/2D
BatchNormalization
,Dense
,Dropout
,Flatten
MaxPooling1D/2D
,GlobalMaxPooling1D/2D
ELU
,LeakyReLU
,ReLU
,SeLU
,PReLU
Sigmoid
,Softmax
,Softplus
,Tanh
UpSampling1D/2D
Reshape
Also supported
 multiple inputs and outputs
 nested models
 residual connections
 shared layers
 arbitrary complex model architectures / computational graphs
Currently not supported are the following:
ActivityRegularization
,
AlphaDropout
,
AveragePooling3D
,
Bidirectional
,
Conv2DTranspose
,
Conv3D
,
ConvLSTM2D
,
CuDNNGRU
,
CuDNNLSTM
,
Cropping3D
,
DepthwiseConv2D
,
Dot
,
Embedding
,
GaussianDropout
,
GaussianNoise
,
GRU
,
GRUCell
,
Lambda
,
LocallyConnected1D
,
LocallyConnected2D
,
LSTM
,
LSTMCell
,
Masking
,
MaxPooling3D
,
Permute
,
RepeatVector
,
RNN
,
SimpleRNN
,
SimpleRNNCell
,
StackedRNNCells
,
ThresholdedReLU
,
TimeDistributed
,
Upsampling3D
,
any custom layers
,
None
as input/output shape
Usage

Use Keras/Python to build (
model.compile(...)
), train (model.fit(...)
) and test (model.evaluate(...)
) your model as usual. Then save it to a single HDF5 file usingmodel.save('....h5', include_optimizer=False)
. Theimage_data_format
in your model must bechannels_last
, which is the default when using the TensorFlow backend. Models created with a differentimage_data_format
and other backends are not supported. 
Now convert it to the frugallydeep file format with
keras_export/convert_model.py

Finally load it in C++ (
fdeep::load_model(...)
) and usemodel.predict(...)
to invoke a forward pass with your data.
The following minimal example shows the full workflow:
# create_model.py
import numpy as np
from keras.layers import Input, Dense
from keras.models import Model
inputs = Input(shape=(4,))
x = Dense(5, activation='relu')(inputs)
predictions = Dense(3, activation='softmax')(x)
model = Model(inputs=inputs, outputs=predictions)
model.compile(loss='categorical_crossentropy', optimizer='nadam')
model.fit(
np.asarray([[1,2,3,4], [2,3,4,5]]),
np.asarray([[1,0,0], [0,0,1]]), epochs=10)
model.save('keras_model.h5', include_optimizer=False)
python3 keras_export/convert_model.py keras_model.h5 fdeep_model.json
// main.cpp
#include <fdeep/fdeep.hpp>
int main()
{
const auto model = fdeep::load_model("fdeep_model.json");
const auto result = model.predict(
{fdeep::tensor3(fdeep::shape3(4, 1, 1), {1, 2, 3, 4})});
std::cout << fdeep::show_tensor3s(result) << std::endl;
}
When using convert_model.py
a test case (input and corresponding output values) is generated automatically and saved along with your model. fdeep::load_model
runs this test to make sure the results of a forward pass in frugallydeep are the same as in Keras.
In order to convert images to fdeep::tensor3
the convenience function tensor3_from_bytes
is provided (cimg example, opencv example, tensor3_to_cv_mat.cpp).
In case you want to convert an Eigen::Matrix
to fdeep::tensor3
, have a look at the following two examples: copy values, reuse memory
Performance
Below you can find the average durations of multiple consecutive forward passes for some popular models ran on a single core of an Intel Core i56600 CPU @ 3.30GHz. frugallydeep was compiled (GCC ver. 5.4.0) with g++ O3 mavx
(same as TensorFlow binaries). The processes were started with CUDA_VISIBLE_DEVICES='' taskset cpulist 1 ...
to disable the GPU and to only allow usage of one CPU.
 Model  Keras + TF  frugallydeep 

 DenseNet121  0.64 s  0.24 s 
 DenseNet169  0.74 s  0.26 s 
 DenseNet201  0.95 s  0.34 s 
 InceptionV3  0.51 s  0.33 s 
 MobileNet  0.21 s  0.06 s 
 NASNetLarge  2.81 s  2.35 s 
 NASNetMobile  0.24 s  0.16 s 
 ResNet50  0.49 s  0.23 s 
 VGG16  0.64 s  0.81 s 
 VGG19  0.79 s  0.94 s 
 Xception  1.02 s  0.54 s 
Keras version: 2.1.6
TensorFlow version: 1.7.1
Requirements and Installation
A C++14compatible compiler is needed. Compilers from these versions on are fine: GCC 4.9, Clang 3.7 (libc++ 3.7) and Visual C++ 2015.
Guides for different ways to install frugallydeep can be found in INSTALL.md
.
Internals
frugallydeep uses channels_first
(depth/channels, height, width
) as its image_data_format
internally. convert_model.py
takes care of all necessary conversions.
From then on everything is handled as a float32 tensor with rank 3. Dense layers for example take its input flattened to a shape of (n, 1, 1)
. This is also the shape you will receive as the output of a final softmax
layer for example.
In case you would like to use double
instead of float
for all calculations, simply do this:
#define FDEEP_FLOAT_TYPE double
#include <fdeep/fdeep.hpp>
A frugallydeep model is threadsafe, i.e. you can call model.predict
on the same model instance from different threads simultaneously. This way you may utilize up to as many CPU cores as you have predictions to make. With model::predict_multi
there is a convenience function available to handle the parallelism for you.
Disclaimer
The API of this library still might change in the future. If you have any suggestions, find errors or want to give general feedback/criticism, I'd love to hear from you. Of course, contributions are also very welcome.
License
Distributed under the MIT License.
(See accompanying file LICENSE
or at
https://opensource.org/licenses/MIT)