- added supermatte

git-svn-id: http://moon:8086/svn/matlab/trunk@143 801c6759-fa7c-4059-a304-17956f83a07c

chromakey/chromakey.m

@@ -28,77 +28,41 @@ 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 = [0.3 0.4 0.65];	# cineon_before 
+ref = [0.2 0.6 0.95];	# adventure_rose 
 
 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
+k_ds_matte = 1.0;
+k_e_matte = 20.0
+k_e_edge = 10.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));
+matte = supermatte(im_rgb, ref_rgb, k_e_matte);
 imshow(matte); title("matte")
 
 figure;
-final = max(0, im_rgb - k_ds*matte.*ref_rgb).*matte;
-imshow(final); title("final")
+intermediate = im_rgb.*matte;
+imshow(intermediate); title("intermediate")
 
 figure;
-edges = coloredges(final, ref_rgb, k_e_edge);
+edges = 1-supermatte(intermediate, ref_rgb, k_e_edge);
 imshow(edges); title("edges")
 
 figure;
-final_cc = im_rgb - edges.*ref_rgb;
-imshow(final_cc); title("final_cc")
+final = max(0, intermediate - k_ds_matte.*edges.*ref_rgb);
+imshow(final); title("final")
 
+imwrite(intermediate, 'intermediate.png','png','alpha', matte);
 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;

chromakey/supermatte.m

@@ -0,0 +1,32 @@
+## 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} =} supermatte (@var{input1}, @var{input2})
+##
+## @seealso{}
+## @end deftypefn
+
+## Author: Jens <jens@orion>
+## Created: 2020-10-04
+
+function matte = supermatte (im_rgb, ref_rgb, k_e)
+
+im_minus_ref = im_rgb - ref_rgb;
+var_im_minus_ref = var(im_minus_ref, 0, 3);
+matte = 1-exp(-k_e*var_im_minus_ref);
+
+endfunction