def _get_simplepose_model(model_subname="", number_of_keypoints=14):
from models import simplepose_coco
if model_subname == "mobilenetv2":
model = simplepose_coco.simplepose_mobilenetv2_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet18":
model = simplepose_coco.simplepose_resnet18_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet50b":
model = simplepose_coco.simplepose_resnet50b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet101b":
model = simplepose_coco.simplepose_resnet101b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet152b":
model = simplepose_coco.simplepose_resnet152b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resneta50b":
model = simplepose_coco.simplepose_resneta50b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet101b":
model = simplepose_coco.simplepose_resneta101b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resneta152b":
model = simplepose_coco.simplepose_resneta152b_coco(
keypoints=number_of_keypoints)
else:
model = simplepose_coco.simplepose_resnet18_coco(
keypoints=number_of_keypoints)
model.return_heatmap = True
return model
def _get_simplepose_model(model_subname="",
number_of_keypoints=14,
config_extra={}):
from models import simplepose_coco
if model_subname == "mobilenetv2":
# mv2_alpha: `0.35`,`0.50`,`0.75`,`1.0`,`1.3`,`1.4`
model = simplepose_coco.simplepose_mobilenetv2_coco(
keypoints=number_of_keypoints, mv2_alpha=1.0)
elif model_subname == "mv2":
model = simplepose_coco.simplepose_mv2_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet18":
model = simplepose_coco.simplepose_resnet18_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet50b":
model = simplepose_coco.simplepose_resnet50b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet101b":
model = simplepose_coco.simplepose_resnet101b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet152b":
model = simplepose_coco.simplepose_resnet152b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resneta50b":
model = simplepose_coco.simplepose_resneta50b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resnet101b":
model = simplepose_coco.simplepose_resneta101b_coco(
keypoints=number_of_keypoints)
elif model_subname == "resneta152b":
model = simplepose_coco.simplepose_resneta152b_coco(
keypoints=number_of_keypoints)
else:
model = simplepose_coco.simplepose_resnet18_coco(
keypoints=number_of_keypoints)
model.return_heatmap = True
return model