ruby-dnn 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ccb761989aa380096c4b16db46da21a4fb74e95bb8b15ed4564f4da6818bf04f
-  data.tar.gz: 2764cc949b09a42b2bbb7ccee7504e62055b33318cf5f3beb4977eb50c9b86ff
+  metadata.gz: 5ab676b5d5bb17163dfc274ff37f3164c2008d5fd17e8c411085202f123c206d
+  data.tar.gz: '099f3fce9dc95457e23e31b44fa4557151adb367d19e5189988eb0b7d63044a2'
 SHA512:
-  metadata.gz: da160fcb2a10367a916dbb020b10ecf6b7cc8eba0e1aa30d68e0f788c06a5a5ff29122a758308666384921f49866a493230aae527e343bec1b3470f6de849fa2
-  data.tar.gz: 6016b580542781fae7e1b07e2cb3aac219070a6428d0d02505fe833a1a54564947dbb196afeea76122fb964f761c7b23aca2513162d5b14be64113cddeac4f2b
+  metadata.gz: d4576bf98e350e9dcb9ec7e6e787da23e19cf504b6cba2c4892385aa946d383439d50456e1e4acf16b3202f28faa05da29f9e8594d2106cb126d0a1f28d44731
+  data.tar.gz: 8162cf4f8ef0f5f560c5e52a28aa6c9278d59b74835a00101c971435da9204c9ce0255aaf8ee024315f53dcf6c7917aeee37c18a4c0f8a07da119fa218f5a2fd

data/API-Reference.ja.md

@@ -2,7 +2,7 @@
 ruby-dnnのAPIリファレンスです。このリファレンスでは、APIを利用するうえで必要となるクラスとメソッドしか記載していません。
 そのため、プログラムの詳細が必要な場合は、ソースコードを参照してください。
 
-対応バージョン:0.2.0
+対応バージョン:0.2.2
 
 # module DNN
 ruby-dnnの名前空間をなすモジュールです。
@@ -178,7 +178,7 @@ SFloat
 ### return
 なし。
 
-## def builded?
+## def built?
 レイヤーがビルド済みであるか否かを取得します。
 ### arguments
 なし。
@@ -318,6 +318,19 @@ maxプーリングを行うレイヤーです。
 ゼロパディングを行います。
 
 
+# class UnPool2D < Layer
+逆プーリングを行うレイヤーです。
+
+## 【Instance methods】
+## def initialize(unpool_width, unpool_height)
+コンストラクタ。
+### arguments
+* Integer unpool_width  
+逆プーリングを行う横の長さ。
+* Integer unpool_height
+逆プーリングを行う縦の長さ。
+
+
 # class Flatten
 N次元のデータを平坦化します。
 

data/README.md

@@ -8,7 +8,7 @@ Currently, you can get 99% accuracy with MNIST and 74% with CIFAR 10.
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'dnn'
+gem 'ruby-dnn'
 ```
 
 And then execute:
@@ -17,7 +17,7 @@ And then execute:
 
 Or install it yourself as:
 
-    $ gem install dnn
+    $ gem install ruby-dnn
 
 ## Usage
 

data/lib/dnn/core/layers.rb

@@ -6,17 +6,18 @@ module DNN
       include Numo
 
       def initialize
-        @builded = false
+        @built = false
       end
 
-      #Initialize layer when model is compiled.
+      #Build the layer.
       def build(model)
         @builded = true
         @model = model
       end
-
-      def builded?
-        @builded
+      
+      #Does the layer have already been built?
+      def built?
+        @built
       end
 
       #Forward propagation.
@@ -343,7 +344,7 @@ module DNN
         col = im2col(x, @out_width, @out_height, @pool_width, @pool_height, @strides)
         col = col.reshape(x.shape[0] * @out_width * @out_height * x.shape[3], @pool_width * @pool_height)
         @max_index = col.max_index(1)
-        col.max(1).reshape(x.shape[0], @out_width, @out_height, x.shape[3])#.transpose(0, 3, 1, 2)
+        col.max(1).reshape(x.shape[0], @out_width, @out_height, x.shape[3])
       end
 
       def backward(dout)
@@ -369,8 +370,56 @@ module DNN
         }
       end
     end
+
+
+    class UnPool2D < Layer
+      include Convert
+
+      def initialize(unpool_width, unpool_height)
+        super()
+        @unpool_width = unpool_width
+        @unpool_height = unpool_height
+      end
+
+      def self.load_hash(hash)
+        UnPool2D.new(hash[:unpool_width], hash[:unpool_height])
+      end
+
+      def build(model)
+        super
+        @origin_width = prev_layer.shape[0]
+        @origin_height = prev_layer.shape[1]
+        @out_width = @origin_width * @unpool_width
+        @out_height = @origin_height * @unpool_height
+        @num_channel = prev_layer.shape[2]
+      end
+
+      def forward(x)
+        unpool_size = @unpool_width * @unpool_height
+        x2 = SFloat.zeros(x.shape[0], x.shape[1], @unpool_width, x.shape[2], @unpool_height, x.shape[3])
+        x2[true, true, 0, true, 0, true] = x
+        x2.reshape(x.shape[0], @out_width, @out_height, x.shape[3])
+      end
+
+      def backward(dout)
+        dout = dout.reshape(dout.shape[0], @origin_width, @unpool_width, @origin_height, @unpool_height, dout.shape[3])
+        dout[true, true, 0, true, 0, true].clone
+      end
+
+      def shape
+        [@out_width, @out_height, @num_channel]
+      end
+
+      def to_hash
+        {
+          name: self.class.name,
+          unpool_width: @unpool_width,
+          unpool_height: @unpool_height,
+        }
+      end
+    end
     
-    
+
     class Flatten < Layer
       def forward(x)
         @shape = x.shape

data/lib/dnn/core/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-07-12 00:00:00.000000000 Z
+date: 2018-07-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray