diff options
| -rw-r--r-- | python/caffe/draw.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/python/caffe/draw.py b/python/caffe/draw.py index f8bf5722..cfa3fc5b 100644 --- a/python/caffe/draw.py +++ b/python/caffe/draw.py @@ -10,7 +10,16 @@ Caffe network visualization: draw the NetParameter protobuffer. """ from caffe.proto import caffe_pb2 -import pydot + +""" +pydot is not supported under python 3 and pydot2 doesn't work properly. +pydotplus works nicely (pip install pydotplus) +""" +try: + # Try to load pydotplus + import pydotplus as pydot +except ImportError: + import pydot # Internal layer and blob styles. LAYER_STYLE_DEFAULT = {'shape': 'record', |