Module: DNN::Image

Defined in:
lib/dnn/image.rb

Defined Under Namespace

Classes: ImageError, ImageReadError, ImageShapeError, ImageWriteError

Constant Summary collapse

RGB = 
3
RGBA = 
4

Class Method Summary collapse

Class Method Details

.read(file_name, channel_type = RGB) ⇒ Object

Read image from file.

Parameters:

  • file_name (String) —

    File name to read.

  • channel_type (Integer) (defaults to: RGB) —

    Specify channel type of image.

Raises:



20
21
22
23
24
25
26
 
# File 'lib/dnn/image.rb', line 20

def self.read(file_name, channel_type = RGB)
  raise ImageReadError, "#{file_name} is not found." unless File.exist?(file_name)
  bin, w, h, n = Stb.stbi_load(file_name, channel_type)
  raise ImageReadError, "#{file_name} load failed." if bin == ""
  img = Numo::UInt8.from_binary(bin)
  img.reshape(h, w, channel_type)
end

.resize(img, out_height, out_width) ⇒ Object

Resize the image.

Parameters:

  • img (Numo::UInt8) —

    Image to resize.

  • out_height (Integer) —

    Image height to resize.

  • out_width (Integer) —

    Image width to resize.



71
72
73
74
75
76
77
 
# File 'lib/dnn/image.rb', line 71

def self.resize(img, out_height, out_width)
  img_check(img)
  in_height, in_width, ch = *img.shape
  out_bin, res = Stb.stbir_resize_uint8(img.to_binary, in_width, in_height, 0, out_width, out_height, 0, ch)
  img2 = Numo::UInt8.from_binary(out_bin).reshape(out_height, out_width, ch)
  img2
end

.to_gray_scale(img) ⇒ Object

Image convert to gray scale.

Parameters:

  • img (Numo::UInt8) —

    Image to gray scale.



92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/dnn/image.rb', line 92

def self.to_gray_scale(img)
  img_check(img)
  if img.shape[2] == RGB
    x = Numo::SFloat.cast(img)
    x = x.mean(axis: 2, keepdims: true)
  elsif img.shape[2] == RGBA
    x = Numo::SFloat.cast(img[true, true, 0..2])
    x = x.mean(axis: 2, keepdims: true).concatenate(img[true, true, 3..3], axis: 2)
  end
  Numo::UInt8.cast(x)
end

.trim(img, y, x, height, width) ⇒ Object

Trimming the image.

Parameters:

  • img (Numo::UInt8) —

    Image to resize.

  • y (Integer) —

    The begin y coordinate of the image to trimming.

  • x (Integer) —

    The begin x coordinate of the image to trimming.

  • height (Integer) —

    Image height to trimming.

  • width (Integer) —

    Image height to trimming.



85
86
87
88
 
# File 'lib/dnn/image.rb', line 85

def self.trim(img, y, x, height, width)
  img_check(img)
  img[y...(y + height), x...(x + width), true].clone
end

.write(file_name, img, quality: 100) ⇒ Object

Write image to file.

Parameters:

  • file_name (String) —

    File name to write.

  • img (Numo::UInt8) —

    Image to write.

  • quality (Integer) (defaults to: 100) —

    Image quality when jpeg write.

Raises:



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/dnn/image.rb', line 32

def self.write(file_name, img, quality: 100)
  img_check(img)
  match_data = file_name.match(%r`(.*)/.+$`)
  if match_data
    dir_name = match_data[1]
    Dir.mkdir(dir_name) unless Dir.exist?(dir_name)
  end
  h, w, ch = img.shape
  match_data = file_name.match(/\.(\w+)$/i)
  if match_data
    ext = match_data[1]
  else
    raise ImageWriteError, "File name has not extension."
  end
  case ext
  when "png"
    stride_in_bytes = w * ch
    res = Stb.stbi_write_png(file_name, w, h, ch, img.to_binary, stride_in_bytes)
  when "bmp"
    res = Stb.stbi_write_bmp(file_name, w, h, ch, img.to_binary)
  when "jpg", "jpeg"
    raise TypeError, "quality:#{quality.class} is not an instance of Integer class." unless quality.is_a?(Integer)
    raise ArgumentError, "quality should be between 1 and 100." unless quality.between?(1, 100)
    res = Stb.stbi_write_jpg(file_name, w, h, ch, img.to_binary, quality)
  when "hdr"
    float_img = Numo::SFloat.cast(img) / 255
    res = Stb.stbi_write_hdr(file_name, w, h, ch, float_img.to_binary)
  when "tga"
    res = Stb.stbi_write_tga(file_name, w, h, ch, img.to_binary)
  else
    raise ImageWriteError, "Extension '#{ext}' is not support."
  end
  raise ImageWriteError, "Image write failed." if res == 0
end