2014-09-04 30 views
-1

我已经编写了彩色图像直方图均衡的代码。该代码适用于灰度图像,但会对彩色图像引发错误。我将rgb图像转换为hsv,然后执行直方图均衡。对此有何想法?Matlab:索引必须是正整数或逻辑

代码:

% GIm=imread('tire.tif'); this is grayscale image, which works fine 
I=imread('D:\came.jpg'); 
GIm=rgb2hsv(I);% converting color image to hsv 
numofpixels=size(GIm,1)*size(GIm,2); 
figure,imshow(GIm); 
title('Original Image'); 
HIm=uint8(zeros(size(GIm,1),size(GIm,2))); 
freq=zeros(256,1); 
probf=zeros(256,1); 
probc=zeros(256,1); 
cum=zeros(256,1); 
output=zeros(256,1); 
%freq counts the occurrence of each pixel value. 
%The probability of each occurrence is calculated by probf. 
for i=1:size(GIm,1) 
    for j=1:size(GIm,2) 
     value=GIm(i,j); 
     freq(value+1)=freq(value+1)+1; % giving error here 
     probf(value+1)=freq(value+1)/numofpixels; 
    end 

end 

sum=0; 
no_bins=255; 
%The cumulative distribution probability is calculated. 
for i=1:size(probf) 
    sum=sum+freq(i); 
    cum(i)=sum; 
    probc(i)=cum(i)/numofpixels; 
    output(i)=round(probc(i)*no_bins); 
end 

for i=1:size(GIm,1) 
    for j=1:size(GIm,2) 
      HIm(i,j)=output(GIm(i,j)+1); 
    end 
end 

figure,imshow(HIm); 
title('Histogram equalization'); 
subplot(2,2,2); 
plot(GIm); 
%bar(GIm); 
title('Before Histogram equalization'); 
subplot(2,2,4); 
plot(HIm) 
%bar(HIm); 
title('After Histogram equalization'); 

错误:

Attempted to access freq(1.61378); index must be a positive integer or logical. 

Error in histogramequ (line 18) 
     freq(value+1)=freq(value+1)+1; 
+2

詹姆斯 - 仔细观察HSV的图像,GIm。在命令窗口中键入'class(GIm)'和'max(GIm(:))'。你可能会注意到,HSV数据类型(类)是** double **,最大值是1.所以所有的像素值都在[0,1]区间内,错误信息是有意义的。同时检查RGB和HSV图像的尺寸。对于灰度图像可以有三个不同的尺寸。您的代码可能需要更新才能处理。 – Geoff 2014-09-04 02:05:18

+2

我不知道你将如何创建一个直方图 - 可能是每个色相,饱和度和值分量(HSV图像)将[0,1]区间分割成固定数量的区域。例如,[0,0.1],[0.1,0.2),...,[0.8,0.9),[0.9,1]可能是您尝试“填充”的十个垃圾箱。虽然你可能需要更多的人(这只是一个例子)。 – Geoff 2014-09-04 02:12:36

回答

1

代码的主要问题是rgb2hsv()后,各像素的格式是在[0,1]双,而不是UINT8。所以你需要转换回[0,255],因此它可以用作下标。

以下代码将正常工作。

for i=1:size(GIm,1) 
    for j=1:size(GIm,2) 
     value=floor(GIm(i,j) * 255); % now value is in [0,255] 
     freq(value+1)=freq(value+1)+1; 
     probf(value+1)=freq(value+1)/numofpixels; 
    end 
end 
相关问题