-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMain612.m
78 lines (76 loc) · 4.8 KB
/
Main612.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
clear
clc
dbstop if error
I = imread('Lena.tiff');
%I = imread('Airplane.tiff');
%I = imread('Baboon.tiff');
%I = imread('Jetplane.tiff');
%I = imread('Tiffany.tiff');
%I = imread('Man.tiff');
Origin_I = double(I);
imshow(Origin_I,[]);
subplot(221);imshow(Origin_I,[]);title('原始图像');
%% 随机生成嵌入的数据
num = 4000000; %秘密数据的长度
rand('seed',0);
Data = round(rand(1,num)*1);
%%
[row,col] = size(Origin_I);
block_size =4;
%% 图像加密密钥和数据加密密钥
Image_key=1;
Data_key=2;
%% 图像重排列并自嵌typeimage
[typeI1,typeI2,typeI3,typeI4,typeI5,typeI6,typeI7,typeI8,compress_type_len,recover_start_ub,Predict_error_I,judge_predict,compress_predict,tag_preprocess,finalem,f,Process_bitplane1,Process_bitplane2,Process_bitplane3,Process_bitplane4,Process_bitplane5,Process_bitplane6,Process_bitplane7,Process_bitplane8] = Preprocess1(Origin_I,block_size);
%% 判断每个NUB是否可嵌，并产生序列,嵌入UB
[bin_NUB_len,final_emUBdata,tag_NUB1,tag_NUB2,tag_NUB3,tag_NUB4,tag_NUB5,tag_NUB6,tag_NUB7,tag_NUB8,Second_pro_bitplane1,Second_pro_bitplane2,Second_pro_bitplane3,Second_pro_bitplane4,Second_pro_bitplane5,Second_pro_bitplane6,Second_pro_bitplane7,Second_pro_bitplane8] = Preprocess2(compress_type_len,tag_preprocess,finalem,f,block_size,Process_bitplane1,Process_bitplane2,Process_bitplane3,Process_bitplane4,Process_bitplane5,Process_bitplane6,Process_bitplane7,Process_bitplane8);
%% 图像加密
[Process_I] = eight_to_one(Second_pro_bitplane1,Second_pro_bitplane2,Second_pro_bitplane3,Second_pro_bitplane4,Second_pro_bitplane5,Second_pro_bitplane6,Second_pro_bitplane7,Second_pro_bitplane8);
[Encrypt_I] = Encrypt_image(Process_I,Image_key,final_emUBdata,f,tag_preprocess,block_size);
subplot(222);
imshow(Encrypt_I,[]);
%% UB数据嵌入
[emubD,num_emubD,num_em_everyub,Stego_bitplane1,Stego_bitplane2,Stego_bitplane3,Stego_bitplane4,Stego_bitplane5,Stego_bitplane6,Stego_bitplane7,Stego_bitplane8] = Embed_data(Encrypt_I,block_size,Data,num,tag_preprocess,final_emUBdata,f);
%% NUB数据嵌入
[emNUBD,num_NUBD,num_NUBD_every,final_Stego1,final_Stego2,final_Stego3,final_Stego4,final_Stego5,final_Stego6,final_Stego7,final_Stego8] = Embed_NUBdata(block_size,num_emubD,Data,tag_preprocess,f,Stego_bitplane1,Stego_bitplane2,Stego_bitplane3,Stego_bitplane4,Stego_bitplane5,Stego_bitplane6,Stego_bitplane7,Stego_bitplane8,tag_NUB1,tag_NUB2,tag_NUB3,tag_NUB4,tag_NUB5,tag_NUB6,tag_NUB7,tag_NUB8);
[Stego_I] = eight_to_one(final_Stego1,final_Stego2,final_Stego3,final_Stego4,final_Stego5,final_Stego6,final_Stego7,final_Stego8);
subplot(223);
imshow(Stego_I,[]);
%% 提取UB里的秘密信息
[exUB_Data,exUB_numData] = Extract_UBdata(tag_preprocess,num_emubD,final_emUBdata,block_size,final_Stego1,final_Stego2,final_Stego3,final_Stego4,final_Stego5,final_Stego6,final_Stego7,final_Stego8);
%% 提取NUB里的秘密信息
[exNUB_Data,exNUB_numData] = Extract_NUBdata(tag_preprocess,num_NUBD,block_size,tag_NUB1,tag_NUB2,tag_NUB3,tag_NUB4,tag_NUB5,tag_NUB6,tag_NUB7,tag_NUB8,final_Stego1,final_Stego2,final_Stego3,final_Stego4,final_Stego5,final_Stego6,final_Stego7,final_Stego8);
%% 比较提取与嵌入
check1 = isequal(Data(1:num_emubD),exUB_Data);
if check1 == 1
disp('UB提取数据与嵌入数据完全相同！')
else
disp('UB数据提取错误！')
end
check2 = isequal(Data(num_emubD+1:num_emubD+num_NUBD),exNUB_Data);
if check2 == 1
disp('NUB提取数据与嵌入数据完全相同！')
else
disp('NUB数据提取错误！')
end
%% 嵌入率
compress_predict_len = length(compress_predict); %判断像素是否可改变的标记
ER=((num_emubD+num_NUBD-compress_predict_len-8-16*8-16*8)/(row*col));
disp(['Embedding rate equal to : ' num2str(ER)])
%8是tag数目，2*8*8是每个位平面有一个16位二进制表示压缩后NUB的总个数（未嵌入），16*8是每个位平面压缩后typeimage的个数
% 图像要解密后，才能恢复
%% 图像恢复
%--------解密--------%
[Decrypt_I] = decrypt_image(Stego_I,Image_key,final_emUBdata,f,tag_preprocess,block_size);
%----------------恢复得到预测误差位平面----------------%
[recover_bitplane1,recover_bitplane2,recover_bitplane3,recover_bitplane4,recover_bitplane5,recover_bitplane6,recover_bitplane7,recover_bitplane8] = Recover_image1(Decrypt_I,compress_type_len,tag_preprocess,recover_start_ub,block_size,typeI1,typeI2,typeI3,typeI4,typeI5,typeI6,typeI7,typeI8,tag_NUB1,tag_NUB2,tag_NUB3,tag_NUB4,tag_NUB5,tag_NUB6,tag_NUB7,tag_NUB8,final_Stego1,final_Stego2,final_Stego3,final_Stego4,final_Stego5,final_Stego6,final_Stego7,final_Stego8);
%----------------得到原始图像--------------------%
[Recover_I,Prediction_I] = Recover_image2(judge_predict,recover_bitplane1,recover_bitplane2,recover_bitplane3,recover_bitplane4,recover_bitplane5,recover_bitplane6,recover_bitplane7,recover_bitplane8);
subplot(224);imshow(Recover_I,[]);title('恢复图像');
%% 判断恢复图像
check3 = isequal(Origin_I,Recover_I);
if check3 == 1
disp('重构图像与原始图像完全相同！')
else
disp('Warning！图像重构错误！')
end