Skip to content

Commit 6040d92

Browse files
committed
feat: default face detect
1 parent 28e861e commit 6040d92

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

demo/ui.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,15 @@ def create_ui(
1616
):
1717
DEFAULT_LANG = "zh"
1818
DEFAULT_HUMAN_MATTING_MODEL = "modnet_photographic_portrait_matting"
19-
DEFAULT_FACE_DETECT_MODEL = "mtcnn"
19+
DEFAULT_FACE_DETECT_MODEL = "retinaface-resnet50"
2020

2121
if DEFAULT_HUMAN_MATTING_MODEL in human_matting_models:
2222
human_matting_models.remove(DEFAULT_HUMAN_MATTING_MODEL)
2323
human_matting_models.insert(0, DEFAULT_HUMAN_MATTING_MODEL)
2424

25+
if DEFAULT_FACE_DETECT_MODEL not in face_detect_models:
26+
DEFAULT_FACE_DETECT_MODEL = "mtcnn"
27+
2528
demo = gr.Blocks(title="HivisionIDPhotos")
2629

2730
with demo:

0 commit comments

Comments
 (0)