diff --git a/facefusion/choices.py b/facefusion/choices.py index 1157de15..43d05968 100755 --- a/facefusion/choices.py +++ b/facefusion/choices.py @@ -1,13 +1,13 @@ from typing import List -from facefusion.typing import VideoMemoryStrategy, FaceSelectorMode, FaceAnalyserOrder, FaceAnalyserAge, FaceAnalyserGender, FaceMaskType, FaceMaskRegion, TempFrameFormat, OutputVideoEncoder, OutputVideoPreset +from facefusion.typing import VideoMemoryStrategy, FaceSelectorMode, FaceAnalyserOrder, FaceAnalyserAge, FaceAnalyserGender, FaceDetectorModel, FaceMaskType, FaceMaskRegion, TempFrameFormat, OutputVideoEncoder, OutputVideoPreset from facefusion.common_helper import create_int_range, create_float_range video_memory_strategies : List[VideoMemoryStrategy] = [ 'strict', 'moderate', 'tolerant' ] face_analyser_orders : List[FaceAnalyserOrder] = [ 'left-right', 'right-left', 'top-bottom', 'bottom-top', 'small-large', 'large-small', 'best-worst', 'worst-best' ] face_analyser_ages : List[FaceAnalyserAge] = [ 'child', 'teen', 'adult', 'senior' ] face_analyser_genders : List[FaceAnalyserGender] = [ 'male', 'female' ] -face_detector_models : List[str] = [ 'retinaface', 'yoloface', 'yunet' ] +face_detector_models : List[FaceDetectorModel] = [ 'retinaface', 'yoloface', 'yunet' ] face_detector_sizes : List[str] = [ '160x160', '320x320', '480x480', '512x512', '640x640', '768x768', '960x960', '1024x1024' ] face_selector_modes : List[FaceSelectorMode] = [ 'reference', 'one', 'many' ] face_mask_types : List[FaceMaskType] = [ 'box', 'occlusion', 'region' ]