3v324v23's picture
code pushed
515f781
seecoder_base:
symbol: seecoder
args: {}
seecoder:
super_cfg: seecoder_base
type: seecoder
args:
imencoder_cfg : MODEL(swin_large)
imdecoder_cfg : MODEL(seecoder_decoder)
qtransformer_cfg : MODEL(seecoder_query_transformer)
seecoder_pa:
super_cfg: seet
type: seecoder
args:
imencoder_cfg : MODEL(swin_large)
imdecoder_cfg : MODEL(seecoder_decoder)
qtransformer_cfg : MODEL(seecoder_query_transformer_position_aware)
###########
# decoder #
###########
seecoder_decoder:
super_cfg: seecoder_base
type: seecoder_decoder
args:
inchannels:
res3: 384
res4: 768
res5: 1536
trans_input_tags: ['res3', 'res4', 'res5']
trans_dim: 768
trans_dropout: 0.1
trans_nheads: 8
trans_feedforward_dim: 1024
trans_num_layers: 6
#####################
# query_transformer #
#####################
seecoder_query_transformer:
super_cfg: seecoder_base
type: seecoder_query_transformer
args:
in_channels : 768
hidden_dim: 768
num_queries: [4, 144]
nheads: 8
num_layers: 9
feedforward_dim: 2048
pre_norm: False
num_feature_levels: 3
enforce_input_project: False
with_fea2d_pos: false
seecoder_query_transformer_position_aware:
super_cfg: seecoder_query_transformer
args:
with_fea2d_pos: true