github.com
  
  
    

VainF/DeepLabV3Plus-Pytorch


  DeepLabv3, DeepLabv3+ and pretrained weights on VOC &amp; Cityscapes

  
  
    
    
  
  


Just like the model here.
  
      github.com
  
  
    VainF/DeepLabV3Plus-Pytorch/blob/0c67dce524b2eb94dc3587ff2832e28f11440cae/main.py#L253

# Set up modelmodel_map = {    'deeplabv3_resnet50': network.deeplabv3_resnet50,    'deeplabv3plus_resnet50': network.deeplabv3plus_resnet50,    'deeplabv3_resnet101': network.deeplabv3_resnet101,    'deeplabv3plus_resnet101': network.deeplabv3plus_resnet101,    'deeplabv3_mobilenet': network.deeplabv3_mobilenet,    'deeplabv3plus_mobilenet': network.deeplabv3plus_mobilenet}model = model_map[opts.model](num_classes=opts.num_classes, output_stride=opts.output_stride)if opts.separable_conv and 'plus' in opts.model:    network.convert_to_separable_conv(model.classifier)utils.set_bn_momentum(model.backbone, momentum=0.01)# Set up metricsmetrics = StreamSegMetrics(opts.num_classes)# Set up optimizeroptimizer = torch.optim.SGD(params=[    {'params': model.backbone.parameters(), 'lr': 0.1*opts.lr},