diff --git a/chromakey/chromakey.m b/chromakey/chromakey.m index f874783..0f11b3e 100644 --- a/chromakey/chromakey.m +++ b/chromakey/chromakey.m @@ -29,6 +29,11 @@ function chromakey (prj_name, k_e_matte, k_e_edge) k_ds_matte = 1.0; +clip_white_thresh = 0.80; +clip_white_amp = 3; + +clip_black_thresh = 0.20; +clip_black_amp = 3; [im_rgb, ref_rgb] = prj_load(prj_name); @@ -41,6 +46,12 @@ 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, clip_black_thresh, clip_black_amp); + +figure; +imshow(matte_clipped_w.*matte_clipped_b); title("matte_{clipped}") + figure; intermediate = im_rgb.*matte; imshow(intermediate); title("intermediate") @@ -57,6 +68,8 @@ imwrite(intermediate, sprintf('%s.intermediate.png',prj_name),'png','alpha', mat 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; diff --git a/chromakey/clip_black.m b/chromakey/clip_black.m new file mode 100644 index 0000000..e4c9f07 --- /dev/null +++ b/chromakey/clip_black.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 +## . + +## -*- texinfo -*- +## @deftypefn {} {@var{retval} =} clip_black (@var{input1}, @var{input2}) +## +## @seealso{} +## @end deftypefn + +## Author: Jens +## Created: 2020-10-04 + +function y = clip_black (x, thresh, k) +ind = find (x >= thresh); + +y = x/k; +y(ind) = thresh/k + (x(ind)-thresh)*(1+thresh); + +endfunction diff --git a/chromakey/clip_white.m b/chromakey/clip_white.m new file mode 100644 index 0000000..a186fcd --- /dev/null +++ b/chromakey/clip_white.m @@ -0,0 +1,33 @@ +## 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 +## . + +## -*- texinfo -*- +## @deftypefn {} {@var{retval} =} clip_white (@var{input1}, @var{input2}) +## +## @seealso{} +## @end deftypefn + +## Author: Jens +## Created: 2020-10-04 + +function y = clip_white (x, thresh, k) + +ind = find (x >= thresh); + +y = x; +y(ind) = thresh + (x(ind)-thresh)*k; + +endfunction