From df88a41e1de2c18eecc884aed2e27ee31541e2b6 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 8 Oct 2026 15:37:35 +0200 Subject: [PATCH] fix test --- tests/test_face_masker.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_face_masker.py b/tests/test_face_masker.py index d6e80102..9f80c5da 100644 --- a/tests/test_face_masker.py +++ b/tests/test_face_masker.py @@ -31,7 +31,7 @@ def before_all() -> None: state_manager.init_item('face_detector_score', 0.5) state_manager.init_item('face_aligner_model', '2dfan4') state_manager.init_item('face_aligner_score', 0.5) - state_manager.init_item('face_occluder_model', 'xseg_1') + state_manager.init_item('face_occluder_model', 'many') state_manager.init_item('face_parser_model', 'bisenet_resnet_34') face_classifier.pre_check() @@ -59,6 +59,8 @@ def test_create_occlusion_mask() -> None: source_face = get_one_face(get_static_faces([ source_vision_frame ])) crop_vision_frame, _ = warp_face_by_face_landmark_5(source_vision_frame, source_face.landmark_set.get('5/68'), 'arcface_128', (256, 256)) + state_manager.set_item('face_occluder_model', 'xseg_1') + assert create_occlusion_mask(crop_vision_frame).sum() == pytest.approx(25111.992, rel = 1e-3) state_manager.set_item('face_occluder_model', 'many')