jens
281c35bcc9
git-svn-id: http://moon:8086/svn/matlab/trunk@141 801c6759-fa7c-4059-a304-17956f83a07c
113 lines
2.8 KiB
Objective-C
113 lines
2.8 KiB
Objective-C
## Copyright (C) 2020 Jens
|
|
##
|
|
## This program is free software: you can redistribute it and/or modify it
|
|
## under the terms of the GNU General Public License as published by
|
|
## the Free Software Foundation, either version 3 of the License, or
|
|
## (at your option) any later version.
|
|
##
|
|
## This program is distributed in the hope that it will be useful, but
|
|
## WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
## GNU General Public License for more details.
|
|
##
|
|
## You should have received a copy of the GNU General Public License
|
|
## along with this program. If not, see
|
|
## <https://www.gnu.org/licenses/>.
|
|
|
|
## -*- texinfo -*-
|
|
## @deftypefn {} {@var{retval} =} bluescreen (@var{input1}, @var{input2})
|
|
##
|
|
## @seealso{}
|
|
## @end deftypefn
|
|
|
|
## Author: Jens <jens@orion>
|
|
## Created: 2020-10-03
|
|
|
|
function chromakey (img_name)
|
|
im = im2double(imread(img_name));
|
|
im_rgb = imresize(im, 1);
|
|
se = strel('square',3)
|
|
|
|
ref = [0.3 0.4 0.65]; # cineon_before
|
|
#ref = 0.2 0.6 0.95]; # adventure_rose
|
|
|
|
ref_c = 1 - ref
|
|
|
|
ref_rgb = toRef(ref, size(im_rgb));
|
|
ref_rgb_c = toRef(ref_c, size(im_rgb));
|
|
|
|
k_ds = 0.2;
|
|
k_e_matte = 5.0
|
|
k_e_edge = 3.0
|
|
|
|
close all;
|
|
|
|
figure;
|
|
imshow(ref_rgb);
|
|
|
|
figure;
|
|
matte_p = imnorm(im_rgb, ref_rgb);
|
|
imshow(matte_p); title("matte_p")
|
|
|
|
figure;
|
|
matte_n = imnorm(im_rgb, ref_rgb_c);
|
|
imshow(matte_n); title("matte_n")
|
|
|
|
figure;
|
|
matte = exp(-k_e_matte*(matte_n - matte_p));
|
|
imshow(matte); title("matte")
|
|
|
|
figure;
|
|
final = max(0, im_rgb - k_ds*matte.*ref_rgb).*matte;
|
|
imshow(final); title("final")
|
|
|
|
figure;
|
|
edges = coloredges(final, ref_rgb, k_e_edge);
|
|
imshow(edges); title("edges")
|
|
|
|
figure;
|
|
final_cc = im_rgb - edges.*ref_rgb;
|
|
imshow(final_cc); title("final_cc")
|
|
|
|
imwrite(final, 'final.png','png','alpha', matte);
|
|
imwrite(final_cc, 'final_cc.png','png','alpha', matte);
|
|
imwrite(matte, 'alpha.png','png');
|
|
|
|
function r = toRef(ref, d)
|
|
r = repmat(reshape(ref,1, 1, 3), d(1), d(2));
|
|
endfunction
|
|
|
|
function hsv = toHsv(in, gain_hsv)
|
|
hsv = rgb2hsv(in) .* toRef(gain_hsv, size(in));
|
|
endfunction
|
|
|
|
function s = imnorm(im, ref)
|
|
s = vecnorm (im - ref, 2, 3);
|
|
endfunction
|
|
|
|
function s = similarity(im, ref, k)
|
|
s = 1-exp(-k*vecnorm (im - ref, 2, 3));
|
|
endfunction
|
|
|
|
function s = similarity_hsv(im_rgb, ref_rgb, k, weigth)
|
|
im = toHsv(im_rgb, weigth);
|
|
ref = toHsv(ref_rgb, weigth);
|
|
s = 1-exp(-k*vecnorm (im - ref, 2, 3));
|
|
endfunction
|
|
|
|
function s = coloredges(im, ref, k)
|
|
s = exp(-k*vecnorm (im - ref, 2, 3));
|
|
endfunction
|
|
|
|
function mask = mask_clip_above(mask_in, thresh)
|
|
k = mask_in < thresh;
|
|
mask = k .* mask_in + kn;
|
|
endfunction
|
|
|
|
function mask = mask_clip_below(mask_in, thresh)
|
|
k = mask_in > thresh;
|
|
mask = k .* mask_in + 0;
|
|
endfunction
|
|
|
|
endfunction
|