Skip to content

Commit

Permalink
rename ia_encoder_decoder to dual_input_encoder_decoder
Browse files Browse the repository at this point in the history
  • Loading branch information
likyoo committed Nov 15, 2022
1 parent f1ac67d commit fb86fd5
Show file tree
Hide file tree
Showing 6 changed files with 10 additions and 10 deletions.
2 changes: 1 addition & 1 deletion configs/_base_/models/changer_r18.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# model settings
norm_cfg = dict(type='SyncBN', requires_grad=True)
model = dict(
type='IAEncoderDecoder',
type='DIEncoderDecoder',
pretrained=None,
backbone=dict(
type='IA_ResNetV1c',
Expand Down
2 changes: 1 addition & 1 deletion configs/_base_/models/changer_s50.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# model settings
norm_cfg = dict(type='SyncBN', requires_grad=True)
model = dict(
type='IAEncoderDecoder',
type='DIEncoderDecoder',
pretrained=None,
backbone=dict(
type='IA_ResNeSt',
Expand Down
2 changes: 1 addition & 1 deletion configs/_base_/models/fc_siam_diff.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
norm_cfg = dict(type='SyncBN', requires_grad=True)
base_channels = 16
model = dict(
type='IAEncoderDecoder',
type='DIEncoderDecoder',
pretrained=None,
backbone=dict(
type='SiamUnet_diff',
Expand Down
2 changes: 1 addition & 1 deletion configs/_base_/models/snunet_c16.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
norm_cfg = dict(type='SyncBN', requires_grad=True)
base_channels = 16
model = dict(
type='IAEncoderDecoder',
type='DIEncoderDecoder',
pretrained=None,
backbone=dict(
type='SNUNet_ECAM',
Expand Down
4 changes: 2 additions & 2 deletions opencd/models/change_detectors/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# Copyright (c) Open-CD. All rights reserved.
from .siamencoder_decoder import SiamEncoderDecoder
from .ia_encoder_decoder import IAEncoderDecoder
from .dual_input_encoder_decoder import DIEncoderDecoder

__all__ = ['SiamEncoderDecoder', 'IAEncoderDecoder']
__all__ = ['SiamEncoderDecoder', 'DIEncoderDecoder']
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@


@SEGMENTORS.register_module()
class IAEncoderDecoder(BaseSegmentor):
"""Interaction Encoder Decoder segmentors.
IAEncoderDecoder typically consists of backbone, decode_head, auxiliary_head.
class DIEncoderDecoder(BaseSegmentor):
"""Dual Input Encoder Decoder segmentors.
DIEncoderDecoder typically consists of backbone, decode_head, auxiliary_head.
Note that auxiliary_head is only used for deep supervision during training,
which could be dumped during inference.
"""
Expand All @@ -28,7 +28,7 @@ def __init__(self,
pretrained=None,
init_cfg=None,
backbone_inchannels=3):
super(IAEncoderDecoder, self).__init__(init_cfg)
super(DIEncoderDecoder, self).__init__(init_cfg)
if pretrained is not None:
assert backbone.get('pretrained') is None, \
'both backbone and segmentor set pretrained weight'
Expand Down

0 comments on commit fb86fd5

Please sign in to comment.