ruby-dnn 0.6.3 → 0.6.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83bf68ae55435acfd33ccebf5798252ad670c7047f31de52c5b1951e0d0064f0
-  data.tar.gz: 63edb99dcafd2abb6b66f09a5fe44ada8e50f2674b56af2d3b3100980ba6e841
+  metadata.gz: e7500fad8921cbb1efca407a3dac06aee005257e96403654a0047acd1f05bd91
+  data.tar.gz: 6b2177271b189de689130e75848b92fe747c9b995cc90d16fe2d656102753fb7
 SHA512:
-  metadata.gz: 78baa0705d6062f066d132f46c500277daf03957ddaa4e7530c9f75b8bfcc08d0d4ac6ffd3130d8cbf7e01b32ce6a1e60ab1b046af9ed187c2b658d6ee71bb8e
-  data.tar.gz: ae2930b1fcccd676f3e3273e0031c63888fe264d069c2e6c16aeb9c7def9a8bf5e05e2de2ee9265d2ce1e633c964e0b91843b3e2c0fa845428c77db09c9a05cd
+  metadata.gz: dba263057f0ac42d515bfac8ba26c33898329d3ab7e29132f498110eaad22660a2ac8e1b2e9f537ed226c5da07561332ca327123857941598ea7f666fb2d2392
+  data.tar.gz: 3a771401278c68385c9e4a163caee00ffa380ac71ca545ccb500b503c6a92afefb8ca93aa4216dd6783cf10c5187eaa353b916618879e8275b3997a479926117

data/API-Reference.ja.md

@@ -2,7 +2,7 @@
 ruby-dnnのAPIリファレンスです。このリファレンスでは、APIを利用するうえで必要となるクラスとメソッドしか記載していません。
 そのため、プログラムの詳細が必要な場合は、ソースコードを参照してください。
 
-最終更新バージョン:0.6.2
+最終更新バージョン:0.6.4
 
 # module DNN
 ruby-dnnの名前空間をなすモジュールです。
@@ -12,17 +12,20 @@ ruby-dnnの名前空間をなすモジュールです。
 ## VERSION
 ruby-dnnのバージョン。
 
+# class Model
+ ニューラルネットワークのモデルを作成するクラスです。
+
 ## 【Properties】
 
 ## attr_accessor :layer
 モデルに追加されたレイヤーの配列を取得します。
 
+## attr_accessor :trainable
+falseを設定すると、パラメータの学習を禁止します。
+
 ## attr_reader :optimize
 モデルのオプティマイザーを取得します。
 
-# class Model
- ニューラルネットワークのモデルを作成するクラスです。
-
 ## 【Singleton methods】
 
 ## def self.load(file_name)

data/lib/dnn/core/layers.rb

@@ -45,13 +45,15 @@ module DNN
     
     # This class is a superclass of all classes with learning parameters.
     class HasParamLayer < Layer
-      attr_reader :params # The parameters of the layer.
-      attr_reader :grads  # Differential value of parameter of layer.
+      attr_accessor :trainable # Setting false prevents learning of parameters.
+      attr_reader :params      # The parameters of the layer.
+      attr_reader :grads       # Differential value of parameter of layer.
     
       def initialize
         super
         @params = {}
         @grads = {}
+        @trainable = true
       end
     
       def build(model)
@@ -61,7 +63,7 @@ module DNN
     
       # Update the parameters.
       def update
-        @model.optimizer.update(self)
+        @model.optimizer.update(self) if @trainable
       end
     
       private

data/lib/dnn/core/model.rb

@@ -3,8 +3,9 @@ require "json"
 module DNN
   # This class deals with the model of the network.
   class Model
-    attr_accessor :layers
-    attr_reader :optimizer
+    attr_accessor :layers    # All layers possessed by the model
+    attr_accessor :trainable # Setting false prevents learning of parameters.
+    attr_reader :optimizer   # Optimizer possessed by the model
 
     def self.load(file_name)
       Marshal.load(File.binread(file_name))
@@ -20,6 +21,7 @@ module DNN
   
     def initialize
       @layers = []
+      @trainable = true
       @optimizer = nil
       @training = false
       @compiled = false
@@ -140,7 +142,7 @@ module DNN
       forward(x, true)
       loss = @layers[-1].loss(y)
       backward(y)
-      @layers.each { |layer| layer.update if layer.respond_to?(:update) }
+      @layers.each { |layer| layer.update if @trainable && layer.is_a?(HasParamLayer) }
       loss
     end
   

data/lib/dnn/ext/rb_stb_image/rb_stb_image.c

@@ -37,7 +37,6 @@ VALUE rb_stbi_write_png(VALUE self, VALUE rb_filename, VALUE rb_w, VALUE rb_h, V
   int result;
 
   result = stbi_write_png(filename, w, h, comp, data, stride_in_bytes);
-  stbi_image_free(data);
   return INT2FIX(result);
 }
 
@@ -51,7 +50,6 @@ VALUE rb_stbi_write_bmp(VALUE self, VALUE rb_filename, VALUE rb_w, VALUE rb_h, V
   int result;
 
   result = stbi_write_bmp(filename, w, h, comp, data);
-  stbi_image_free(data);
   return INT2FIX(result);
 }
 
@@ -66,7 +64,6 @@ VALUE rb_stbi_write_jpg(VALUE self, VALUE rb_filename, VALUE rb_w, VALUE rb_h, V
   int result;
 
   result = stbi_write_jpg(filename, w, h, comp, data, quality);
-  stbi_image_free(data);
   return INT2FIX(result);
 }
 

data/lib/dnn/lib/image_io.rb

@@ -20,12 +20,12 @@ module DNN
       h, w, ch = img.shape
       bin = img.to_binary
       case file_name
-      when /\.png$/
+      when /\.png$/i
         stride_in_bytes = w * ch
         Stb.stbi_write_png(file_name, w, h, ch, bin, stride_in_bytes)
-      when /\.bmp$/
+      when /\.bmp$/i
         Stb.stbi_write_bmp(file_name, w, h, ch, bin)
-      when /\.jpg$/
+      when /\.jpg$/i, /\.jpeg/i
         Stb.stbi_write_jpg(file_name, w, h, ch, bin, quality)
       end
     rescue => ex

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.6.3"
+  VERSION = "0.6.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.6.3
+  version: 0.6.4
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-08-19 00:00:00.000000000 Z
+date: 2018-08-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray