mirror of
https://github.com/ruby-opencv/ruby-opencv
synced 2023-03-27 23:22:12 -04:00
43 lines
868 B
Ruby
43 lines
868 B
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
# inpaint.rb
|
||
|
|
require "rubygems"
|
||
|
|
require "opencv"
|
||
|
|
|
||
|
|
include OpenCV
|
||
|
|
|
||
|
|
owindow = GUI::Window.new "original"
|
||
|
|
mwindow = GUI::Window.new "mask"
|
||
|
|
iwindow = GUI::Window.new "inpaint"
|
||
|
|
|
||
|
|
image = IplImage::load "inpaint.png"
|
||
|
|
b, g, r = image.split
|
||
|
|
original_mask = r.threshold_binary_inverse(0x00, 0xFF) & b.threshold_binary_inverse(0x00, 0xFF)
|
||
|
|
mask = original_mask.copy
|
||
|
|
|
||
|
|
num_dilate = 3
|
||
|
|
radius = 5
|
||
|
|
dilate_bar = mwindow.set_trackbar("dilate", 10, num_dilate){|v|
|
||
|
|
num_dilate = v
|
||
|
|
mask = original_mask.dilate(nil, num_dilate)
|
||
|
|
mwindow.show mask
|
||
|
|
}
|
||
|
|
|
||
|
|
radius_bar = mwindow.set_trackbar("radius", 30, radius){|v|
|
||
|
|
radius = v
|
||
|
|
}
|
||
|
|
|
||
|
|
owindow.show image
|
||
|
|
mwindow.show mask
|
||
|
|
|
||
|
|
while key = GUI::wait_key
|
||
|
|
case key.chr
|
||
|
|
when "\e" # esc
|
||
|
|
exit
|
||
|
|
when "n"
|
||
|
|
iwindow.show image.inpaint_ns(mask, radius)
|
||
|
|
when "t"
|
||
|
|
iwindow.show image.inpaint_telea(mask, radius)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|