mirror of
https://github.com/ruby-opencv/ruby-opencv
synced 2023-03-27 23:22:12 -04:00
57 lines
1.3 KiB
Ruby
Executable file
57 lines
1.3 KiB
Ruby
Executable file
#!/usr/bin/env ruby
|
|
# inpaint.rb
|
|
require "rubygems"
|
|
require "opencv"
|
|
include OpenCV
|
|
|
|
puts <<EOS
|
|
Inpainting sample
|
|
|
|
Usage:
|
|
'dilate' bar - Adjust mask to inpaint (Non-black areas indicate the area that needs to be inpainted).
|
|
'radius' bar - Adjust radius of a circular neighborhood of each point inpainted.
|
|
'n' key - Inpaint using Navier-Stokes based method
|
|
't' key - Inpaint using Alexandru Telea's method
|
|
'c' key - Clear the inpaint window
|
|
'ESC' key - exit"
|
|
EOS
|
|
|
|
owindow = GUI::Window.new('original')
|
|
mwindow = GUI::Window.new('mask')
|
|
iwindow = GUI::Window.new('inpaint')
|
|
|
|
image = IplImage::load('inpaint.png')
|
|
noimage = image.zero
|
|
b, g, r = image.split
|
|
original_mask = r.threshold(0x00, 0xFF, CV_THRESH_BINARY_INV) & b.threshold(0x00, 0xFF, CV_THRESH_BINARY_INV)
|
|
mask = original_mask.copy
|
|
|
|
num_dilate = 3
|
|
mwindow.set_trackbar("dilate", 10, num_dilate) { |v|
|
|
num_dilate = v
|
|
mask = original_mask.dilate(nil, num_dilate)
|
|
mwindow.show mask
|
|
}
|
|
|
|
radius = 5
|
|
mwindow.set_trackbar("radius", 10, radius) { |v|
|
|
radius = v
|
|
}
|
|
|
|
owindow.show image
|
|
mwindow.show mask
|
|
iwindow.show noimage
|
|
|
|
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)
|
|
when "c"
|
|
iwindow.show noimage
|
|
end
|
|
end
|
|
|