fix reference model group 1 to group 2

brodyh · Dec 23, 2014 · f4a09dd · f4a09dd
1 parent c27a408
commit f4a09dd
Show file tree

Hide file tree

Showing 4 changed files with 92 additions and 299 deletions.
diff --git a/examples/filter_visualization_driving.ipynb b/examples/filter_visualization_driving.ipynb
diff --git a/examples/filter_visualization_driving_softmax.ipynb b/examples/filter_visualization_driving_softmax.ipynb
diff --git a/models/bvlc_reference_caffenet/train_val.prototxt b/models/bvlc_reference_caffenet/train_val.prototxt
@@ -97,7 +97,7 @@ layers {
     num_output: 256
     pad: 2
     kernel_size: 5
-    group: 1
+    group: 2
     weight_filler {
       type: "gaussian"
       std: 0.01
@@ -178,7 +178,7 @@ layers {
     num_output: 384
     pad: 1
     kernel_size: 3
-    group: 1
+    group: 2
     weight_filler {
       type: "gaussian"
       std: 0.01
@@ -208,7 +208,7 @@ layers {
     num_output: 256
     pad: 1
     kernel_size: 3
-    group: 1
+    group: 2
     weight_filler {
       type: "gaussian"
       std: 0.01

diff --git a/python/caffe/draw.py b/python/caffe/draw.py
@@ -56,7 +56,7 @@ def get_pydot_graph(caffe_net):
         pydot.Edge(pydot_nodes[edge[0]], pydot_nodes[edge[1]]))
   return pydot_graph
 
-def draw_net(caffe_net, ext='png'):
+def draw_net(caffe_net, ext='pdf'):
   """Draws a caffe net and returns the image string encoded using the given
   extension.