Spaces:
Running
Running
File size: 1,082 Bytes
1c25fe3 06fbec3 1c25fe3 06fbec3 1c25fe3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from hivision.creator.human_matting import *
from hivision.creator.face_detector import *
def choose_handler(creator, matting_model_option=None, face_detect_option=None):
if matting_model_option == "modnet_photographic_portrait_matting":
creator.matting_handler = extract_human_modnet_photographic_portrait_matting
elif matting_model_option == "mnn_hivision_modnet":
creator.matting_handler = extract_human_mnn_modnet
elif matting_model_option == "rmbg-1.4":
creator.matting_handler = extract_human_rmbg
# elif matting_model_option == "birefnet-portrait":
# creator.matting_handler = extract_human_birefnet_portrait
elif matting_model_option == "birefnet-v1-lite":
creator.matting_handler = extract_human_birefnet_lite
else:
creator.matting_handler = extract_human
if (
face_detect_option == "face_plusplus"
or face_detect_option == "face++ (联网API)"
):
creator.detection_handler = detect_face_face_plusplus
else:
creator.detection_handler = detect_face_mtcnn
|