deep_image_fusion2/fusion_strategy_w_pca.m at master · nuriyeyldrm/deep_image_fusion2 · GitHub

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
function [gen, ave1, ave2, ave3, ave4] = fusion_strategy_w_pca(features_a, features_b,...
    source_a, source_b, img1, img2, img_1, img_2, unit)

[m,n] = size(features_a);
[m1,n1] = size(source_a);

ave_temp1 = zeros(m1,n1);
ave_temp2 = zeros(m1,n1);
ave_temp3 = zeros(m1,n1);
ave_temp4 = zeros(m1,n1);

weight_ave_temp1 = zeros(m1,n1);
weight_ave_temp2 = zeros(m1,n1);
weight_ave_temp3 = zeros(m1,n1);
weight_ave_temp4 = zeros(m1,n1);

for i=2:m-1
    for j=2:n-1
        A1 = sum(sum(features_a(i-1:i+1,j-1:j+1)))/9;
        A2 = sum(sum(features_b(i-1:i+1,j-1:j+1)))/9;
        A3 = sum(sum(img1(i-1:i+1,j-1:j+1)))/9;
        A4 = sum(sum(img2(i-1:i+1,j-1:j+1)))/9;

        % weight average
        weight_ave_temp1(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit))...
            = A1/(A1+A2+A3+A4);
        weight_ave_temp2(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit))...
            = A2/(A1+A2+A3+A4);
        weight_ave_temp3(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit))...
            = A3/(A1+A2+A3+A4);
        weight_ave_temp4(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit))...
            = A4/(A1+A2+A3+A4);

        ave_temp1(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit)) = A1;
        ave_temp2(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit)) = A2;
        ave_temp3(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit)) = A3;
        ave_temp4(((i-2)*unit+1):((i-1)*unit),((j-2)*unit+1):((j-1)*unit)) = A4;
    end
end

weight_ave_temp1 = weight_ave_temp1(1:m1,1:n1);
weight_ave_temp2 = weight_ave_temp2(1:m1,1:n1);
weight_ave_temp3 = weight_ave_temp3(1:m1,1:n1);
weight_ave_temp4 = weight_ave_temp4(1:m1,1:n1);

gen = source_a.*weight_ave_temp1 + source_b.*weight_ave_temp2 + img_1.*weight_ave_temp3...
    + img_2.*weight_ave_temp4;

ave1 = ave_temp1;
ave2 = ave_temp2;
ave3 = ave_temp3;
ave4 = ave_temp4;
end