note pydot dependency of model visualization
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Fri, 25 Apr 2014 16:23:00 +0000 (09:23 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Fri, 25 Apr 2014 16:23:00 +0000 (09:23 -0700)
python/caffe/drawnet.py

index de5a876..ff18ecf 100644 (file)
@@ -1,5 +1,10 @@
 #!/usr/bin/env python
-"""Functions to draw a caffe NetParameter protobuffer.
+"""
+Caffe network visualization: draw the NetParameter protobuffer.
+
+NOTE: this requires pydot>=1.0.2, which is not included in requirements.txt
+since it requires graphviz and other prerequisites outside the scope of the
+Caffe.
 """
 
 from caffe.proto import caffe_pb2