-
Notifications
You must be signed in to change notification settings - Fork 96
/
colorize.m
63 lines (55 loc) · 1.88 KB
/
colorize.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
%COLORIZE Colorize a greyscale image
%
% OUT = COLORIZE(IM, MASK, COLOR) is a color image where each pixel in OUT
% is set to the corresponding element of the greyscale image IM or a specified
% COLOR according to whether the corresponding value of MASK is true or false
% respectively. The color is specified as a 3-vector (R,G,B).
%
% OUT = COLORIZE(IM, FUNC, COLOR) as above but a the mask is the return value
% of the function handle FUNC applied to the image IM, and returns a per-pixel
% logical result, eg. @isnan.
%
% Examples::
%
% Display image with values < 100 in blue
% out = colorize(im, im<100, [0 0 1])
%
% Display image with NaN values shown in red
% out = colorize(im, @isnan, [1 0 0])
%
% Notes::
% - With no output arguments the image is displayed.
%
% See also IMONO, ICOLOR, IPIXSWITCH.
% Copyright (C) 1993-2011, by Peter I. Corke
%
% This file is part of The Machine Vision Toolbox for Matlab (MVTB).
%
% MVTB is free software: you can redistribute it and/or modify
% it under the terms of the GNU Lesser General Public License as published by
% the Free Software Foundation, either version 3 of the License, or
% (at your option) any later version.
%
% MVTB 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 Lesser General Public License for more details.
%
% You should have received a copy of the GNU Leser General Public License
% along with MVTB. If not, see <http://www.gnu.org/licenses/>.
function co = colorize(img, mask, color)
grey = idouble(img);
g = grey(:);
z = [g g g];
if isa(mask, 'function_handle')
mask = mask(img);
end
k = find(mask(:));
z(k,:) = ones(numrows(k),1)*color(:)';
z = reshape(z, [size(grey) 3]);
if nargout > 0
co = z;
else
image(z);
shg
end