用matlab写一段dct算法水印嵌入和提取

以下是一个简单的MATLAB代码示例，演示了如何使用DCT实现数字水印的嵌入和提取。

% 加载图像
img = imread('lena.png');
img = rgb2gray(img);
[M, N] = size(img);

% 生成随机二进制序列
message = '1010101010101010';
message_bits = double(message(:)' == '1');

% 将二进制序列转换为DCT系数
message_dct = dct2(reshape(message_bits, 4, 4));

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

% 将DCT系数嵌入到高频分量中
alpha = 0.1;
dct(M-3:M, N-3:N) = dct(M-3:M, N-3:N) + alpha * message_dct;

% 对图像进行IDCT变换
watermarked = idct2(dct);

% 提取数字水印
extracted_dct = dct2(watermarked);
extracted_message = round(extracted_dct(M-3:M, N-3:N) / alpha);
extracted_bits = mod(extracted_message(:), 2);

% 输出结果
disp(['原始信息：', num2str(message_bits)]);
disp(['提取信息：', num2str(extracted_bits')]);

在这个示例中，我们首先加载了一张图像，并生成了一个长度为16的随机二进制序列。然后，我们将这个序列通过DCT变换转换为一组DCT系数，并将其嵌入到图像的右下角4x4的DCT系数中。这里，我们将嵌入系数的权值设置为0.1。最后，我们对加入数字水印的图像进行IDCT变换，提取出嵌入的数字水印，并对比原始信息和提取信息，以验证数字水印的正确性。