jens
130f83a27f
- matte is clipped matte white git-svn-id: http://moon:8086/svn/matlab/trunk@150 801c6759-fa7c-4059-a304-17956f83a07c
86 lines
2.5 KiB
Matlab
86 lines
2.5 KiB
Matlab
## 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
|
|
|
|
# chromakey('adventure_rose', 20, 10)
|
|
# chromakey('cineon_before', 200, 100)
|
|
|
|
function chromakey (prj_name, k_e_matte, k_e_edge)
|
|
|
|
k_ds_matte = 1.0;
|
|
clip_white_thresh = 0.60;
|
|
clip_white_amp = 5;
|
|
|
|
clip_black_thresh = 0.40;
|
|
clip_black_amp = 5;
|
|
|
|
[im_rgb, ref_rgb] = prj_load(prj_name);
|
|
|
|
close all;
|
|
figure;
|
|
[matte, dev, conf] = supermatte(im_rgb, ref_rgb, k_e_matte);
|
|
imshow(matte); title("matte")
|
|
figure;
|
|
imshow(dev); title("dev")
|
|
figure;
|
|
imshow(conf); title("conf")
|
|
|
|
matte_clipped_w = clip_white(matte, clip_white_thresh, clip_white_amp);
|
|
matte_clipped_b = clip_black(matte_clipped_w, clip_black_thresh, clip_black_amp);
|
|
matte = matte_clipped_w;
|
|
|
|
figure;
|
|
imshow(matte_clipped_w.*matte_clipped_b); title("matte_{clipped}")
|
|
|
|
figure;
|
|
intermediate = im_rgb.*matte;
|
|
imshow(intermediate); title("intermediate")
|
|
|
|
figure;
|
|
edges = 1-supermatte(intermediate, ref_rgb, k_e_edge);
|
|
imshow(edges); title("edges")
|
|
|
|
figure;
|
|
final = max(0, intermediate - k_ds_matte.*edges.*ref_rgb);
|
|
imshow(final); title("final")
|
|
|
|
imwrite(intermediate, sprintf('%s.intermediate.png',prj_name),'png','alpha', matte);
|
|
imwrite(edges, sprintf('%s.edges.png',prj_name),'png');
|
|
imwrite(final, sprintf('%s.final.png',prj_name),'png','alpha', matte);
|
|
imwrite(matte, sprintf('%s.alpha.png',prj_name),'png');
|
|
imwrite(matte_clipped_w, sprintf('%s.alpha_clipped_w.png',prj_name),'png');
|
|
imwrite(matte_clipped_b, sprintf('%s.alpha_clipped_b.png',prj_name),'png');
|
|
|
|
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
|