Matlab：检查目录中的所有图像是否为灰度？

Question

我想检查当前目录中的所有 jpg 图像是否为灰度或包含彩色像素...我试过了：

figdirectory = pwd;
fullpath = sprintf('%s/*.jpg', figdirectory);
d = dir(fullpath);

% Loop
pages = [];
for i = 1:length(d)
    f = d(i).name;
    fname_input = sprintf('%s/%s', figdirectory, f);

    A = imread(fname_input); 
    B = rgb2gray(A);

    if(A-B == 0)
        hascolor = 0;
    else
        hascolor = 1;
    end
    pages = [pages; hascolor];        
end 

但这给了我关于 A 和 B 的矩阵维度的错误。为什么 A 有第三个维度？ 谢谢！

Answer 1

这样做：

0) 假设您的输入纯粹是“图像”而不是“体积”。

1) [r c d] = 大小(im);

2) 如果 d 大于 1，则为彩色图像

3) 否则，它必须是灰度图像。

编辑： 您可以再添加一个条件来可靠地区分灰度 img 和彩色图像。假设你的彩色图像有 3 个通道，

如果 d 等于 3，则检查所有 3 个通道是否都相等

im(:,:,1)==im(:,:,2) && im(:,:,1)==im(:,:,3) 然后你有一个灰度图像否则颜色img

Answer 2

这是一个基于 IMFINFO 的简单解决方案：

%# some test images shipped with the Image Processing Toolbox
fNames = {
    'circles.png'   %# binary
    'shadow.tif'    %# indexed color
    'coins.png'     %# gray
    'peppers.png'   %# RGB
};

isGrayscale = false(size(fNames));
for i=1:numel(fNames)
    imgInfo = imfinfo(fNames{i});
    if strcmp(imgInfo.ColorType,'truecolor')
        isGrayscale(i) = false;
    elseif strcmp(imgInfo.ColorType,'grayscale')
        isGrayscale(i) = true;
    elseif strcmp(imgInfo.ColorType,'indexed')
            %# indexed images colormap (the three channels should be all same)
        isGrayscale(i) = all(all( diff(imgInfo.Colormap,[],2)==0 ,2),1);
    end
end

您案例的第一部分可以是：

dirName = 'C:\path\to\myimages';
files = dir( fullfile(dirName,'*.jpg') );
fNames = {files.name}';

---

编辑：

@Adrian：关于你发的图片，就保存的图片格式而言，是彩色图片。现在所有 R/G/B 通道都相同的事实只是一个特例......

无论如何，如果您希望能够检测到此类情况，请将上述代码的'truecolor' 部分更改为：

#% ...
if strcmp(imgInfo.ColorType,'truecolor')
    img = imread(fNames{i});
    isGrayscale(i) = isequal(img(:,:,1),img(:,:,2),img(:,:,3));
elseif strcmp(imgInfo.ColorType,'grayscale')
#% ...

Answer 3

您可以使用IMFINFO 执行此任务，这样您就不必将图像加载到内存中。

figdirectory = pwd;
fullpath = sprintf('%s/*.jpg', figdirectory);
d = dir(fullpath);

nImages = length(d);

%# imageType is a cell array with either 'grayscale', 'truecolor', or 'indexed', 
%# depending on the kind of image you have. 

imageType = cell(nImages,1);

for iImg = 1:nImages
info = imfinfo(d(iImg).name);
imageType{iImg} = info.ColorType;
end

%# isGrayscale is true for grayscale images, false otherwise 
%# (though note that there might be mapped images that map to a grayscale colormap).
isGrayscale = cellfun(@(x)strcmp(x,'grayscale'),imageType);

编辑

%# the indexed images have to be loaded in order for you to check 
%# for grayscale maps

indexedIdx = find(cellfun(@(x)strcmp(x,'indexed'),imageType));

for iImg = indexedIdx(:)'

   [~,map] = imread(fullfile(figDirectory,d(iImg).name));

   %# It's a grayscale image if rgb map values are all equal
   isGrayscale(iImg) = all(all(bsxfun(@eq,map,map(:,1)),2),1);
end

%# finally, it is possible that the images are *stored* as truecolor
%# but containing, in fact, a grayscale image

truecolorIdx = find(cellfun(@(x)strcmp(x,'truecolor'),imageType));

for iImg = truecolorIdx(:)'

   img = imread(fullfile(figDirectory,d(iImg).name));

   %# It's a grayscale image if rgb map values are all equal
   isGrayscale(iImg) = all(all(all(bsxfun(@eq,img(:,:,1),img),1),2),3);
end

Answer 4

来自 imread http://www.mathworks.co.uk/help/techdoc/ref/imread.html 的手册

如果文件包含灰度图像，则 A 是 M×N 数组。如果 文件包含一个真彩色图像，A 是一个 M×N×3 数组。

所以 A 将具有用于彩色图像的第三维。

如果您想测试灰度，则可以将图像转换为 hsv。

B = rgb2hsv(A)

来自手册，http://www.mathworks.co.uk/help/techdoc/ref/hsv2rgb.html

当 B(:,2) 为 0 时，颜色是不饱和的（即灰色阴影）。

对于像发布的那样的真彩色图像，如果 unique(B(:,:,2)) 为零，因此

A = imread(fname_input);        
B = rgb2hsv(A);

%# if saturation levels are zero then it's a greyscale image
if(unique(B(:,:,2)) == 0)
    hascolor = 0;
else         
    hascolor = 1; 
end