mask_image.m

来自「Information Fusion of Flash and Non-Flas」· M 代码 · 共 28 行

function mask_image(mask, imagename,resultname);
uint8_img = imread(imagename);
img = double(uint8_img);
R = img(:,:,1);
[M N] = size(R);
G = img(:,:,2);
B = img(:,:,3);

%output black masked image
imgout = zeros(M,N,3);
imgout(:,:,1) = R.*mask;
imgout(:,:,2) = G.*mask;
imgout(:,:,3) = B.*mask;
outdata = uint8(round(imgout));
maskname = ['results/mask_' resultname '.png'];
imwrite(outdata, maskname,'png');

%output alpha masked image
emask = erode(mask,ones(3));
h = fspecial('gaussian',[5 5],0.7);
%h = fspecial('gaussian');
amask = filter2(h,emask);
figure(10);
imagesc(amask);
colormap(gray);
title('alpha mask');
maskname = ['results/alpha_' resultname '.png'];
imwrite(uint8_img, maskname,'png','Alpha',amask);