code: def get_seg_model(cfg, **kwargs):
      model = HighResolutionNet(cfg, **kwargs)
      model.init_weights(cfg.MODEL.PRETRAINED)
  
      return model
  
imports: import os
  import logging
  import numpy
  import torch
  import torch.nn
  import torch._utils
  import torch.nn.functional
  from .bn_helper import BatchNorm2d, BatchNorm2d_class, relu_inplace
label_desc: Use image node_id 2 for calling a global function with example usage: get_seg_model(cfg) and returns: model