whaifree/pfcfuse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(net): 修改 DetailFeatureExtraction 和 DetailFeatureExtractionSAR 类中的 DetailNode 使用方式

Browse Source 
- 将 DetailFeatureExtraction 类中的 DetailNode 使用方式从 useBlock=2 改为 useBlock=1
- 将 DetailFeatureExtractionSAR 类中的 DetailNode 使用方式从 useBlock=1 改为 useBlock=2
This commit is contained in:
zjut 2024-11-18 09:27:16 +08:00
parent 125a6bdf6f
commit ef66a0321d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
net.py
View File

@ -286,7 +286,7 @@ class DetailFeatureFusion(nn.Module):
class DetailFeatureExtraction(nn.Module): class DetailFeatureExtraction(nn.Module):
def __init__(self, num_layers=3): def __init__(self, num_layers=3):
super(DetailFeatureExtraction, self).__init__() 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) self.net = nn.Sequential(*INNmodules)
def forward(self, x): def forward(self, x):
@ -299,7 +299,7 @@ class DetailFeatureExtraction(nn.Module):
class DetailFeatureExtractionSAR(nn.Module): class DetailFeatureExtractionSAR(nn.Module):
def __init__(self, num_layers=3): def __init__(self, num_layers=3):
super(DetailFeatureExtractionSAR, self).__init__() 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) self.net = nn.Sequential(*INNmodules)
def forward(self, x): def forward(self, x):