example_vgg16.rb in menoh-1.0.1

- old
+ new

@@ -1,9 +1,9 @@
 require 'open-uri'
 require 'rmagick'
 require 'menoh'
-# TODO revise api
+
 # download dependencies
 def download_file(url, output)
   return if File.exist? output
   puts "downloading... #{url}"
   File.open(output, 'wb') do |f_output|
@@ -43,11 +43,11 @@
     {
       name: CONV1_1_IN_NAME,
       dims: [
         image_list.length,
         input_shape[:channel_num],
+        input_shape[:height],
         input_shape[:width],
-        input_shape[:height]
       ]
     }
   ],
   output_layers: [FC6_OUT_NAME, SOFTMAX_OUT_NAME]
 }