text: #!/usr/bin/env python
  # -*- coding: utf-8 -*-
  
  from methods.alphapose.mxnet.AlphaPoseMXNet import AlphaPoseMXNet
  from methods.simple_baselines.mxnet.SimpleBaselinesMXNet import SimpleBaselinesMXNet
  from methods.detectron_coco_keypoints.DetectronCocoKeypoints import DetectronCocoKeypoints
  from methods.hrnet.HRNet import HRNet
  
  def get_model(model_str_name):
      if model_str_name == 'AlphaPoseMXNet':
          return AlphaPoseMXNet()
      elif model_str_name == 'DetectronCocoKeypoints':
          return DetectronCocoKeypoints()
      elif model_str_name == 'SimpleBaselinesMXNet':
          return SimpleBaselinesMXNet()
      elif model_str_name == 'HRNet':
          return HRNet()
      else:
          print(f'{model_str_name} is not a valid pose estimation model.')
          raise NotImplementedError
  