[Fix] Fix changestar init

own
Bobholamovic 3 years ago
parent d0f7293481
commit ca1051e5eb
  1. 4
      paddlers/custom_models/cd/changestar.py

@ -28,7 +28,7 @@ class _ChangeStarBase(nn.Layer):
def __init__(self, seg_model, num_classes, mid_channels, inner_channels,
num_convs, scale_factor):
super(_ChangeStarBase, self).__init__(_ChangeStarBase, self)
super(_ChangeStarBase, self).__init__()
self.extract = seg_model
self.detect = ChangeMixin(
@ -63,7 +63,7 @@ class _ChangeStarBase(nn.Layer):
class ChangeMixin(nn.Layer):
def __init__(self, in_ch, out_ch, mid_ch, num_convs, scale_factor):
super(ChangeMixin, self).__init__(ChangeMixin, self)
super(ChangeMixin, self).__init__()
convs = [Conv3x3(in_ch, mid_ch, norm=True, act=True)]
convs += [
Conv3x3(

Loading…
Cancel
Save