|
|
|
@ -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( |
|
|
|
|