From 775cbdf20f980ea653fe2125bc4735d3eeb63412 Mon Sep 17 00:00:00 2001 From: zjut Date: Mon, 18 Nov 2024 09:27:16 +0800 Subject: [PATCH] =?UTF-8?q?refactor(net):=20=E4=BF=AE=E6=94=B9=20DetailFea?= =?UTF-8?q?tureExtraction=20=E5=92=8C=20DetailFeatureExtractionSAR=20?= =?UTF-8?q?=E7=B1=BB=E4=B8=AD=E7=9A=84=20DetailNode=20=E4=BD=BF=E7=94=A8?= =?UTF-8?q?=E6=96=B9=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 将 DetailFeatureExtraction 类中的 DetailNode 使用方式从 useBlock=2 改为 useBlock=1 - 将 DetailFeatureExtractionSAR 类中的 DetailNode 使用方式从 useBlock=1 改为 useBlock=2 --- net.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/net.py b/net.py index debb154..2e2bc40 100644 --- a/net.py +++ b/net.py @@ -286,7 +286,7 @@ class DetailFeatureFusion(nn.Module): class DetailFeatureExtraction(nn.Module): def __init__(self, num_layers=3): super(DetailFeatureExtraction, self).__init__() - INNmodules = [DetailNode(useBlock=2) for _ in range(num_layers)] + INNmodules = [DetailNode(useBlock=1) for _ in range(num_layers)] self.net = nn.Sequential(*INNmodules) def forward(self, x): @@ -299,7 +299,7 @@ class DetailFeatureExtraction(nn.Module): class DetailFeatureExtractionSAR(nn.Module): def __init__(self, num_layers=3): super(DetailFeatureExtractionSAR, self).__init__() - INNmodules = [DetailNode(useBlock=1) for _ in range(num_layers)] + INNmodules = [DetailNode(useBlock=2) for _ in range(num_layers)] self.net = nn.Sequential(*INNmodules) def forward(self, x):