File size: 919 Bytes
7ad9dfd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from .resnet import *
import logging
logger = logging.getLogger('base')
def create_CD_model(opt):
# Our CDMamba model
from models.STNR import STNR as stnr
if opt['model']['name'] == 'STNR':
cd_model = stnr(spatial_dims=opt['model']['spatial_dims'], in_channels=opt['model']['in_channels'], init_filters=opt['model']['init_filters'], out_channels=opt['model']['n_classes'],
mode=opt['model']['mode'], conv_mode=opt['model']['conv_mode'], up_mode=opt['model']['up_mode'], up_conv_mode=opt['model']['up_conv_mode'], norm=opt['model']['norm'],
blocks_down=opt['model']['blocks_down'], blocks_up=opt['model']['blocks_up'], resdiual=opt['model']['resdiual'], diff_abs=opt['model']['diff_abs'], stage=opt['model']['stage'],
mamba_act=opt['model']['mamba_act'], local_query_model=opt['model']['local_query_model'])
|