DEV Community

Cover image for Deep Learning Style Transformations in Ruby
kojix2
kojix2

Posted on

 

Deep Learning Style Transformations in Ruby

Want to do DeepLearning inference in Ruby?
Yes, you can!

Requirement

st.rb

require 'magro'
require 'onnxruntime'

model = OnnxRuntime::Model.new('')

input = Magro::IO.imread(ARGV[0])
                 .transpose(2, 0, 1)
                 .expand_dims(0)
                 .to_a

output = model.predict(input1: input)

result = Numo::Int32[*output['output1'][0]]
                    .clip(0, 255)
                    .transpose(1, 2, 0)
                    .cast_to(Numo::UInt8)

Magro::IO.imsave(ARGV[1], result)
Enter fullscreen mode Exit fullscreen mode

Run

Input image size should be a multiple of 4

ruby st.rb in.jpg out.jpg
Enter fullscreen mode Exit fullscreen mode

Generated images

https://commons.wikimedia.org/wiki/File:Arimatsushibori.JPG 1

Arimatsushibori

candy
candy.jpg

mosaic
mosaic.jpg

pointilism
pointilism

rain_princess
rain_princess.jpg

udnie
udnie.jpg

Beautiful.

Other posts


  1. Attribution-ShareAlike 3.0 Unported (CC BY-SA 3.0)  

Top comments (0)

An Animated Guide to Node.js Event Loop

Node.js doesn’t stop from running other operations because of Libuv, a C++ library responsible for the event loop and asynchronously handling tasks such as network requests, DNS resolution, file system operations, data encryption, etc.

What happens under the hood when Node.js works on tasks such as database queries? We will explore it by following this piece of code step by step.