56 lines
1.1 KiB
Python
56 lines
1.1 KiB
Python
|
|
|
|
|
|
seecoder_decoder_config = {
|
|
'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
|
|
}
|
|
|
|
seecoder_query_config = {
|
|
'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
|
|
}
|
|
|
|
# Not used
|
|
seecoder_query_pa_config = {
|
|
**seecoder_query_config,
|
|
'with_fea2d_pos': True
|
|
}
|
|
|
|
swin_config = {
|
|
'embed_dim': 128,
|
|
'depths': [ 2, 2, 18, 2 ],
|
|
'num_heads': [ 4, 8, 16, 32 ],
|
|
'window_size': 7,
|
|
'ape': False,
|
|
'drop_path_rate': 0.3,
|
|
'patch_norm': True,
|
|
}
|
|
|
|
swin_large_config = {
|
|
**swin_config,
|
|
'embed_dim': 192,
|
|
'depths': [ 2, 2, 18, 2 ],
|
|
'num_heads': [ 6, 12, 24, 48 ],
|
|
'window_size': 12,
|
|
'ape': False,
|
|
'drop_path_rate': 0.3,
|
|
'patch_norm': True,
|
|
} |