Compare commits
2 Commits
260e3aa760
...
775cbdf20f
Author | SHA1 | Date | |
---|---|---|---|
|
775cbdf20f | ||
|
5555d0d39c |
4
net.py
4
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=0) 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() for _ in range(num_layers)]
|
||||
INNmodules = [DetailNode(useBlock=2) for _ in range(num_layers)]
|
||||
self.net = nn.Sequential(*INNmodules)
|
||||
|
||||
def forward(self, x):
|
||||
|
Loading…
Reference in New Issue
Block a user