部分源代码

下面是上述仿真的部分代码，供参考。

%  a原始数据数组
%  aa存储像素从1到255
%  bb存储aa中每个像素所对应的个数
%  cc将a化成一列后，所形成的数组
%  dd,cc中小于最大个数的像素值，减1处理后的结果
% copyright by lilizong
a=imread('c:oat.png');
figure,imshow(a),title('嵌入水印前原图像');
for i=1:255
    aa(i)=i;  %用于存储像素值
end
for i=1:255
    bb(i)=0;  %用于存储每个像素的个数
end
[m,n]=size(a);
k=m*n;   %a的大小
cc=a(:);
tt=1;   %循环遍历向量
%求出某个像素的个数

for i=1:k
    for j=1:255
        if(cc(i)==aa(j))
            bb(j)=bb(j)+1;
        end
    end
end
% 求出哪个像素的个数是最大

max=bb(1);
for i=1:255
    if(bb(i)>max)
        max=bb(i);
        maxi=i;
    end
end
% 将所有小于max的像素的值都减1
ddd=zeros(m,n);
dd=ddd(:);  %产生和cc同样大小的数组，用于存放cc处理后结果；

for i=1:k
    if(cc(i)<max)
        dd(i)=cc(i)-1;
    end
end
%读入水印，并将水印转换成一列，
watermark=imread('c:watermark1.png');
figure,imshow(watermark),title('水印图像');
[w1,w2]=size(watermark);
watermark_line=watermark(:);
oo=w1*w2;
%开始嵌入水印
tt=1;  %让tt加1来遍历整个水印图像
for i=1:k
    if(dd(i)==max)
        if(tt>oo)
            tt=1;  %如果超出水印的范围从头开始
            if(watermark_line(tt)==1)
                  dd(i)=dd(i)-1;
            end
      
        end
        tt=tt+1;
    end
end

%显示嵌入水印的图像


embed=dd;
embed=reshape(dd,m,n);
figure,imshow(embed,[]),title('嵌入水印后图像');
imwrite(embed,'c:embed2.gif');

%提取水印

for i=1:k
    if(dd(i)==max)
        watermark(tt)=0;
        tt=tt+1;
    elseif(dd(i)==(max-1))
        watermark(tt)=1;
        tt=tt+1;
    end
end
for i=1:oo
    watermark_dect(i)=watermark(i);
end
watermark_dect=reshape(watermark_dect,w1,w2);
figure,imshow(watermark_dect,[]),title('提取后水印');

%恢复原图像
dd=embed(:);
for i=1:k
    if(dd(i)==(max-1))
        dd(i)=dd(i)+1;
    elseif(dd(i)<max)
        dd(i)=dd(i)+1;
    end
end
dd=reshape(dd,m,n);
figure,imshow(dd,[]),title('恢复后图像');


%对原图和恢复图像进行比较