search

yysdck / SFFNet

31 stars 0 forks source link

Problem about MDAF.py #4

Open BaolanChen opened 4 months ago

BaolanChen commented 4 months ago

I have some doubts about a section of code in MDAF.py.

x1 may be changed to x2 in the ”attn_211 = self.conv2_1_1(x1)“

   def forward(self, x1,x2):
        b, c, h, w = x1.shape
        x1 = self.norm1(x1)
        x2 = self.norm2(x2)

        attn_111 = self.conv1_1_1(x1)
        attn_112 = self.conv1_1_2(x1)
        attn_113 = self.conv1_1_3(x1)
        attn_121 = self.conv1_2_1(x1)
        attn_122 = self.conv1_2_2(x1)
        attn_123 = self.conv1_2_3(x1)

        attn_211 = self.conv2_1_1(x1)
        attn_212 = self.conv2_1_2(x1)
        attn_213 = self.conv2_1_3(x1)
        attn_221 = self.conv2_2_1(x1)
        attn_222 = self.conv2_2_2(x1)
        attn_223 = self.conv2_2_3(x1)