jens/matlab
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- added supermatte

Browse Source 
git-svn-id: http://moon:8086/svn/matlab/trunk@143 801c6759-fa7c-4059-a304-17956f83a07c
This commit is contained in:
jens
2020-10-04 08:58:53 +00:00
parent 304a41a240
commit f405c8d609
6 changed files with 44 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
chromakey/chromakey.m
+12 -48
View File
@@ -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;