def main():
parser = argparse.ArgumentParser(description="arguments of Neuron Network Optimizer")
parser.add_argument('-n', '--network', default="data/ResNet-50-deploy.prototxt")
parser.add_argument('-m', '--model', default='data/ResNet-50-model.caffemodel')
args = parser.parse_args()
network_file = args.network
json_file = ProtoReader.read_proto(network_file)
model_file = args.model
model = parse_caffemodel(model_file)
network = Network()
network.construct_from_json(json_file)
network.init_caffe_model(model)
in_data = network.input
in_data.set_content(np.zeros(in_data.get_shape()))
for layer in network.layers:
print 'start execute layer %s' % layer.name
layer.execute()
layer.print_info()
print ""
idx = len(network.layers) - 1
print "output:"
print network.layers[idx].top[0].get_content()
