Contents

require "dnn"
require "dnn/image"
require "numo/linalg/autoloader"
require_relative "dcgan"

include DNN::Loaders
Image = DNN::Image

batch_size = 100

dcgan = DCGAN.load("trained/dcgan_model_epoch20.marshal")
gen = dcgan.gen

Numo::SFloat.srand(rand(1 << 31))
noise = Numo::SFloat.new(batch_size, 20).rand(-1, 1)

images = gen.predict(noise)

batch_size.times do |i|
  img = Numo::UInt8.cast(((images[i, false] + 1) * 127.5).round)
  Image.write("img/img_#{i}.jpg", img)
end

Version data entries

20 entries across 20 versions & 1 rubygems

Version	Path
ruby-dnn-1.3.0	examples/dcgan/imgen.rb
ruby-dnn-1.2.3	examples/dcgan/imgen.rb
ruby-dnn-1.2.2	examples/dcgan/imgen.rb
ruby-dnn-1.2.1	examples/dcgan/imgen.rb
ruby-dnn-1.2.0	examples/dcgan/imgen.rb
ruby-dnn-1.1.6	examples/dcgan/imgen.rb
ruby-dnn-1.1.5	examples/dcgan/imgen.rb
ruby-dnn-1.1.4	examples/dcgan/imgen.rb
ruby-dnn-1.1.3	examples/dcgan/imgen.rb
ruby-dnn-1.1.2	examples/dcgan/imgen.rb
ruby-dnn-1.1.1	examples/dcgan/imgen.rb
ruby-dnn-1.1.0	examples/dcgan/imgen.rb
ruby-dnn-1.0.0	examples/dcgan/imgen.rb
ruby-dnn-0.16.2	examples/dcgan/imgen.rb
ruby-dnn-0.16.1	examples/dcgan/imgen.rb
ruby-dnn-0.16.0	examples/dcgan/imgen.rb
ruby-dnn-0.15.3	examples/dcgan/imgen.rb
ruby-dnn-0.15.2	examples/dcgan/imgen.rb
ruby-dnn-0.15.1	examples/dcgan/imgen.rb
ruby-dnn-0.15.0	examples/dcgan/imgen.rb