vnyash
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
from typing import Literal, TypedDict
|
|
|
|
from deepfuze.typing import Face, FaceSet, AudioFrame, VisionFrame
|
|
|
|
FaceDebuggerItem = Literal['bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender']
|
|
FaceEnhancerModel = Literal['codeformer', 'gfpgan_1.2', 'gfpgan_1.3', 'gfpgan_1.4', 'gpen_bfr_256', 'gpen_bfr_512', 'gpen_bfr_1024', 'gpen_bfr_2048', 'restoreformer_plus_plus']
|
|
FaceSwapperModel = Literal['blendswap_256', 'inswapper_128', 'inswapper_128_fp16', 'simswap_256', 'simswap_512_unofficial', 'uniface_256']
|
|
FrameColorizerModel = Literal['ddcolor', 'ddcolor_artistic', 'deoldify', 'deoldify_artistic', 'deoldify_stable']
|
|
FrameEnhancerModel = Literal['clear_reality_x4', 'lsdir_x4', 'nomos8k_sc_x4', 'real_esrgan_x2', 'real_esrgan_x2_fp16', 'real_esrgan_x4', 'real_esrgan_x4_fp16', 'real_hatgan_x4', 'span_kendata_x4', 'ultra_sharp_x4']
|
|
LipSyncerModel = Literal['wav2lip_gan']
|
|
|
|
FaceDebuggerInputs = TypedDict('FaceDebuggerInputs',
|
|
{
|
|
'reference_faces' : FaceSet,
|
|
'target_vision_frame' : VisionFrame
|
|
})
|
|
FaceEnhancerInputs = TypedDict('FaceEnhancerInputs',
|
|
{
|
|
'reference_faces' : FaceSet,
|
|
'target_vision_frame' : VisionFrame
|
|
})
|
|
FaceSwapperInputs = TypedDict('FaceSwapperInputs',
|
|
{
|
|
'reference_faces' : FaceSet,
|
|
'source_face' : Face,
|
|
'target_vision_frame' : VisionFrame
|
|
})
|
|
FrameColorizerInputs = TypedDict('FrameColorizerInputs',
|
|
{
|
|
'target_vision_frame' : VisionFrame
|
|
})
|
|
FrameEnhancerInputs = TypedDict('FrameEnhancerInputs',
|
|
{
|
|
'target_vision_frame' : VisionFrame
|
|
})
|
|
LipSyncerInputs = TypedDict('LipSyncerInputs',
|
|
{
|
|
'reference_faces' : FaceSet,
|
|
'source_audio_frame' : AudioFrame,
|
|
'target_vision_frame' : VisionFrame
|
|
})
|