Deep Learning Style Transformations in Ruby

#ruby #machinelearning #tutorial

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

Requirement

ONNXRuntime and its Ruby binding created by Andrew Kane.
Liquid error: internal
Magro and NArray.
Fast Neural Style Transfer Model in ONNX Model Zoo.

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)

Run

Input image size should be a multiple of 4

ruby st.rb in.jpg out.jpg

Generated images

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

candy

mosaic

pointilism

rain_princess

udnie

Beautiful.

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.