Files
matlab/chromakey/chromakey.m
T
2020-10-04 16:08:41 +00:00

72 lines
2.0 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
# 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;
[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")
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, 'intermediate.png','png','alpha', matte);
imwrite(edges, 'edges.png','png');
imwrite(final, 'final.png','png','alpha', matte);
imwrite(matte, 'alpha.png','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