HuQiong
/
PaddleRS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Fix] Fix changestar init

Browse Source
own
Bobholamovic 3 years ago
parent d0f7293481
commit ca1051e5eb
1 changed files with 2 additions and 2 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 4
      paddlers/custom_models/cd/changestar.py

4
paddlers/custom_models/cd/changestar.py
Unescape View File

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

Write Preview
Loading…
Cancel
Save