From 1cb6feaf5ad80efcfe6aee9a621a2acb74d9ba4b Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Thu, 22 May 2014 01:02:56 -0700 Subject: [PATCH] fix draw_net python script include caffe.draw for drawing functions. --- python/draw_net.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/draw_net.py b/python/draw_net.py index cbea5d9ff4e..ba488294275 100755 --- a/python/draw_net.py +++ b/python/draw_net.py @@ -5,7 +5,7 @@ import os from google.protobuf import text_format -import caffe +import caffe, caffe.draw from caffe.proto import caffe_pb2 @@ -17,7 +17,7 @@ def main(argv): net = caffe_pb2.NetParameter() text_format.Merge(open(sys.argv[1]).read(), net) print 'Drawing net to %s' % sys.argv[2] - draw_net_to_file(net, sys.argv[2]) + caffe.draw.draw_net_to_file(net, sys.argv[2]) if __name__ == '__main__':