From: Nico Galoppo <nico.galoppo@intel.com>
Date: Mon, 21 Nov 2016 19:03:52 +0000 (-0800)
Subject: Fix Python net drawing script
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=db6cf0a728cad63c93b345f2203f3ad1f5d5c2f4;p=platform%2Fupstream%2Fcaffe.git

Fix Python net drawing script
---

diff --git a/python/caffe/draw.py b/python/caffe/draw.py
index 9eecf6d..e4fd7aa 100644
--- a/python/caffe/draw.py
+++ b/python/caffe/draw.py
@@ -104,11 +104,11 @@ def get_layer_label(layer, rankdir):
                       pooling_types_dict[layer.pooling_param.pool],
                       layer.type,
                       separator,
-                      layer.pooling_param.kernel_size,
+                      layer.pooling_param.kernel_size[0] if len(layer.pooling_param.kernel_size._values) else 1,
                       separator,
-                      layer.pooling_param.stride,
+                      layer.pooling_param.stride[0] if len(layer.pooling_param.stride._values) else 1,
                       separator,
-                      layer.pooling_param.pad)
+                      layer.pooling_param.pad[0] if len(layer.pooling_param.pad._values) else 0)
     else:
         node_label = '"%s%s(%s)"' % (layer.name, separator, layer.type)
     return node_label