用matlab给图像嵌入水印和提取水印的代码

以下是用 MATLAB 实现图像水印嵌入和提取的代码：

嵌入水印：

% 加载图像和水印
img = imread('image.png');
watermark = imread('watermark.png');

% 将水印转换为灰度图像
watermark = rgb2gray(watermark);

% 将水印调整至与图像大小相同
watermark = imresize(watermark, size(img, 1));

% 对图像进行DCT变换
dct_img = dct2(img);

% 对水印进行二值化处理
threshold = graythresh(watermark);
watermark_binary = imbinarize(watermark, threshold);

% 将水印嵌入到DCT系数中
alpha = 0.1; % 水印强度
dct_img(1:size(watermark_binary,1),1:size(watermark_binary,2)) = dct_img(1:size(watermark_binary,1),1:size(watermark_binary,2)) + alpha * watermark_binary;

% 对嵌入水印后的DCT系数进行IDCT变换
watermarked_img = idct2(dct_img);

% 保存嵌入水印后的图像
imwrite(watermarked_img, 'watermarked.png');

提取水印：

% 加载嵌入水印后的图像
watermarked_img = imread('watermarked.png');

% 对嵌入水印后的图像进行DCT变换
dct_watermarked_img = dct2(watermarked_img);

% 提取水印
watermark_size = size(watermark);
watermark_binary = zeros(watermark_size);

for i = 1:watermark_size(1)
    for j = 1:watermark_size(2)
        watermark_binary(i,j) = dct_watermarked_img(i,j) > dct_watermarked_img(i+1,j);
    end
end

% 将二值化的水印转换为灰度图像
watermark = uint8(watermark_binary) * 255;

% 保存提取的水印
imwrite(watermark, 'extracted_watermark.png');

请注意，以上代码仅提供了基本的实现思路和代码框架，具体实现可能需要根据具体需求进行调整和优化。