<reponame>leehsiu/pyopenpose<gh_stars>0
from .densepose_model import DensePoseModel
from .generalized_rcnn import GeneralizedRCNN
from .openpose_model import OpenPoseModel

def build_model(cfg,model_type):
    if model_type=='OpenPose':
        model = OpenPoseModel(cfg)
    elif model_type=='DensePose':
        model = DensePoseModel(cfg)
    return model
from .densepose_model import DensePoseModel
from .generalized_rcnn import GeneralizedRCNN
from .openpose_model import OpenPoseModel

def build_model(cfg,model_type):
    if model_type=='OpenPose':
        model = OpenPoseModel(cfg)
    elif model_type=='DensePose':
        model = DensePoseModel(cfg)
    return model