Xtensorflow

XTensorflow

A high level tensorflow lib which can convert defined tf model to caffe and other engine;

NetWork Definition:

from Xtensorflow import *    

input =  tf.placeholder(tf.float32, [batch_size, 256])
xnet = Xtensorflow(input, is_train=is_training, weight_decay=weight_decay, model_name=name)
fc1 = xnet.fc_layer(0, [batch_size, 256], tf.nn.relu)
fc2 = xnet.fc_layer(fc1, [batch_size, 128], tf.nn.relu)
fc3 = xnet.fc_layer(fc2, [batch_size, classnum], None)

output = xnet.get_network_output()

Convert to Caffemodel:

xnet.create_prototxt_and_caffemodel(sess,'model.prototxt','model.caffemodel')

Predefined Model Zoo：

input_imgs = tf.placeholder(tf.float32, [None, 224, 224, 3])
output, xnet = Resnet50(input_imgs, None, name='resnet50')
xnet_variables = tf.get_collection('xnet_varibale')
init = tf.variables_initializer(xnet_variables)
with tf.Session() as sess:
    sess.run(init)
    xnet.create_prototxt_and_caffemodel(sess, 'model.prototxt', 'model.caffemodel')