|
@@ -88,6 +88,8 @@ class DSIFN(nn.Layer):
|
|
|
self.bn_sa5 = make_norm(16)
|
|
|
self.o5_conv4 = Conv1x1(16, num_classes)
|
|
|
|
|
|
+ self.init_weight()
|
|
|
+
|
|
|
def forward(self, t1, t2):
|
|
|
# Extract bi-temporal features.
|
|
|
with paddle.no_grad():
|
|
@@ -173,6 +175,10 @@ class DSIFN(nn.Layer):
|
|
|
|
|
|
return out5, out4, out3, out2, out1
|
|
|
|
|
|
+ def init_weight(self):
|
|
|
+ # Do nothing
|
|
|
+ pass
|
|
|
+
|
|
|
|
|
|
class VGG16FeaturePicker(nn.Layer):
|
|
|
def __init__(self, indices=(3,8,15,22,29)):
|