[Fix] add init_weight method of DSIFN

own
Bobholamovic 3 years ago
parent 439e1d3964
commit fd4118afe3
  1. 6
      paddlers/models/cd/models/dsifn.py

@ -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)):

Loading…
Cancel
Save