Skip to main content

Set of utilities to work easier with Bender.

Project description

Set of utilities to work easier with Bender.

Currently there’s support for TensorFlow and Caffe, but we are working on more stuff!

Works on Python 2.7.+ and 3.+, with Tensorflow 1.2+.

To install:

pip install benderthon

TensorFlow is required too. The simplest way to install it is:

pip install tensorflow

There are other ways, see Installing Tensorflow. Benderthon does not install it by default to let the usage of a custom installation.


Utility to convert TensorFlow checkpoints into minimal frozen graphs.


To take the checkpoint in checkpoint_path.ckpt, whose output is yielded by the node named Tanh, and save it to graph_with_weights.pb:

benderthon tf-freeze checkpoint_path.ckpt graph_with_weights.pb Tanh


The file contains a network example for MNIST dataset with 2 convolutional layers and 2 dens layers. If you run it, it will generate checkpoints files with prefix checkpoints/mnist.ckpt:


Then you can get a minimal protobuf version with the weights frozen:

benderthon tf-freeze checkpoints/mnist.ckpt output/mnist.pb Prediction

The generated file occupies half the original checkpoints (26MB to 13MB).

You can also get only the graph, which occupies just 13kB:

benderthon tf-freeze --no-weights checkpoints/mnist.ckpt output/mnist_only_graph.pb Prediction

To save the weights in a separate path for later processing:

benderthon tf-freeze --only-weights checkpoints/mnist.ckpt weights/ Prediction


This module cannot be accessed from the command line utility, it should be used from Python code, importing benderthon.caffe_freeze.

You need caffeflow package installed first:

pip install -e git://


This utility is under development and the API is not stable. So, do not heavily rely on it.

To install locally you should do ./ install, but first have pandoc and pypandoc installed.


Copyright 2017 Xmartlabs SRL.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
See the License for the specific language governing permissions and
limitations under the License.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Built Distribution

benderthon-0.3.0-py2.py3-none-any.whl (14.0 kB view hashes)

Uploaded py2 py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page