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