class ResNetBackbone(object):
def __init__(self, configer):
self.configer = configer
self.resnet_models = ResNetModels(self.configer)
def __call__(self):
arch = self.configer.get('network', 'backbone')
multi_grid = None
if self.configer.exists('network', 'multi_grid'):
multi_grid = self.configer.get('network', 'multi_grid')
if arch == 'resnet34':
orig_resnet = self.resnet_models.resnet34()
arch_net = NormalResnetBackbone(orig_resnet)
arch_net.num_features = 512
elif arch == 'resnet34_dilated8':
orig_resnet = self.resnet_models.resnet34()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=8,
multi_grid=multi_grid)
arch_net.num_features = 512
elif arch == 'resnet34_dilated16':
orig_resnet = self.resnet_models.resnet34()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=16,
multi_grid=multi_grid)
arch_net.num_features = 512
elif arch == 'resnet50':
orig_resnet = self.resnet_models.resnet50()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'resnet50_dilated8':
orig_resnet = self.resnet_models.resnet50()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=8,
multi_grid=multi_grid)
elif arch == 'resnet50_dilated16':
orig_resnet = self.resnet_models.resnet50()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=16,
multi_grid=multi_grid)
elif arch == 'deepbase_resnet50':
orig_resnet = self.resnet_models.deepbase_resnet50()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'deepbase_resnet50_dilated8':
orig_resnet = self.resnet_models.deepbase_resnet50()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=8,
multi_grid=multi_grid)
elif arch == 'deepbase_resnet50_dilated16':
orig_resnet = self.resnet_models.deepbase_resnet50()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=16,
multi_grid=multi_grid)
elif arch == 'resnet101':
orig_resnet = self.resnet_models.resnet101()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'resnet101_dilated8':
orig_resnet = self.resnet_models.resnet101()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=8,
multi_grid=multi_grid)
elif arch == 'resnet101_dilated16':
orig_resnet = self.resnet_models.resnet101()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=16,
multi_grid=multi_grid)
elif arch == 'deepbase_resnet101':
orig_resnet = self.resnet_models.deepbase_resnet101()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'deepbase_resnet101_dilated8':
orig_resnet = self.resnet_models.deepbase_resnet101()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=8,
multi_grid=multi_grid)
elif arch == 'deepbase_resnet101_dilated16':
orig_resnet = self.resnet_models.deepbase_resnet101()
arch_net = DilatedResnetBackbone(orig_resnet,
dilate_scale=16,
multi_grid=multi_grid)
else:
raise Exception('Architecture undefined!')
return arch_net
def __init__(self, configer):
self.configer = configer
self.resnet_models = ResNetModels(self.configer)
class ResNetBackbone(object):
def __init__(self, configer):
self.configer = configer
self.resnet_models = ResNetModels(self.configer)
def __call__(self):
arch = self.configer.get('network', 'backbone')
if arch == 'resnet34':
orig_resnet = self.resnet_models.resnet34()
arch_net = NormalResnetBackbone(orig_resnet)
arch_net.num_features = 512
elif arch == 'resnet34_dilated8':
orig_resnet = self.resnet_models.resnet34()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=8)
arch_net.num_features = 512
elif arch == 'resnet34_dilated16':
orig_resnet = self.resnet_models.resnet34()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=16)
arch_net.num_features = 512
elif arch == 'resnet50':
orig_resnet = self.resnet_models.resnet50()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'resnet50_dilated8':
orig_resnet = self.resnet_models.resnet50()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=8)
elif arch == 'resnet50_dilated16':
orig_resnet = self.resnet_models.resnet50()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=16)
elif arch == 'resnet101':
orig_resnet = self.resnet_models.resnet101()
arch_net = NormalResnetBackbone(orig_resnet)
elif arch == 'resnet101_dilated8':
orig_resnet = self.resnet_models.resnet101()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=8)
elif arch == 'resnet101_dilated16':
orig_resnet = self.resnet_models.resnet101()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=16)
elif arch == 'caffe_resnet101':
orig_resnet = self.resnet_models.caffe_resnet101()
arch_net = NormalCaffeResnetBackbone(orig_resnet)
elif arch == 'caffe_resnet101_dilated8':
orig_resnet = self.resnet_models.caffe_resnet101()
arch_net = DilatedResnetBackbone(orig_resnet, dilate_scale=8)
elif arch == 'caffe_resnet101_dilated16':
orig_resnet = self.resnet_models.caffe_resnet101()
arch_net = DilatedCaffeResnetBackbone(orig_resnet, dilate_scale=16)
else:
raise Exception('Architecture undefined!')
return arch_net