matlab直方图均衡，使用向量优化

matlab自带有histeq函数对图像进行直方图均衡

自己写了一个，改成向量化形式，效率提高了一点，但是比自带的还是差很多，差不多9倍

function D = my_histeq(I)
[m,n] = size(I);
S = zeros(1,256,'double');
% 得到0-255各个灰阶的计数
i = 1:m;j = 1:n;
S(I(i,j)+1) = S(I(i,j)+1) + 1;
% 得到各灰度阶的概率
S = S./(m*n);
% 得到概率累计分布函数
P = zeros(1,256,'double');
P(1) = S(1);
k = 2:256;
P(k) = P(k-1) + S(k);
% 投影到灰度阶
D = zeros(m,n,'uint8');
D(i,j) = P(I(i,j)+1)*255+0.5;

原来使用for循环的

function D = my_histeq(I)
[m,n] = size(I);
S = zeros(1,256,'double');
% 得到0-255各个灰阶的计数
for i = 1:m
     for j = 1:n
         S(I(i,j)+1) = S(I(i,j)+1) + 1;  % 注意下标
     end
end
% 得到各灰度阶的概率
S = S./(m*n);
% 得到概率累计分布函数
P = zeros(1,256,'double');
for i = 1:256
     if i==1
         P(i) = S(i);
     else
         P(i) = P(i-1) + S(i);
     end
end
D = zeros(m,n,'uint8');
for i = 1:m
     for j=1:n
         D(i,j) = P(I(i,j)+1)*255+0.5;
     end
end