Spaces:
Running
Running
File size: 4,837 Bytes
8fb085a |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
from pydantic import BaseModel
from typing import Optional, List
from facefusion.typing import FaceSelectorMode, FaceAnalyserOrder, FaceAnalyserAge, FaceAnalyserGender, OutputVideoEncoder, FaceDetectorModel, FaceRecognizerModel, TempFrameFormat, Padding
from facefusion.processors.frame.typings import FaceSwapperModel, FaceEnhancerModel, FrameEnhancerModel, FaceDebuggerItem
class Params(BaseModel):
user_id : str
source : Optional[str]
target : str
source_type: str
target_type: str
# execution
execution_providers: Optional[List[str]] = ['CPUExecutionProvider']
execution_thread_count: Optional[int] = 4
execution_queue_count: Optional[int] = 1
max_memory: Optional[int] = 0
# face analyser
face_analyser_order: Optional[FaceAnalyserOrder] = 'left-right'
face_analyser_age: Optional[FaceAnalyserAge] = None
face_analyser_gender: Optional[FaceAnalyserGender] = None
face_detector_model: Optional[FaceDetectorModel] = 'retinaface'
face_detector_size: Optional[str] = '640x640'
face_detector_score: Optional[float] = 0.5
face_recognizer_model: Optional[FaceRecognizerModel] = 'arcface_inswapper'
# face selector
face_selector_mode: Optional[FaceSelectorMode] = 'reference'
reference_face_position: Optional[int] = 0
reference_face_distance: Optional[float] = 0.6
reference_frame_number: Optional[int] = 0
# face mask
face_mask_blur: Optional[float] = 0.3
face_mask_padding: Optional[Padding] = (0, 0, 0, 0)
# frame extraction
trim_frame_start: Optional[int] = None
trim_frame_end: Optional[int] = None
temp_frame_format: Optional[TempFrameFormat] = 'jpg'
temp_frame_quality: Optional[int] = 100
keep_temp: Optional[bool] = False
# output creation
output_image_quality: Optional[int] = 80
output_video_encoder: Optional[OutputVideoEncoder] = 'libx264'
output_video_quality: Optional[int] = 80
keep_fps: Optional[bool] = False
skip_audio: Optional[bool] = False
# frame processors
frame_processors: List[str] = ['face_blur']
face_swapper_model: Optional[FaceSwapperModel] = 'inswapper_128'
face_enhancer_model: Optional[FaceEnhancerModel] = 'gfpgan_1.4'
face_enhancer_blend: Optional[int] = 80
frame_enhancer_model: Optional[FrameEnhancerModel] = 'real_esrgan_x2plus'
frame_enhancer_blend: Optional[int] = 80
face_debugger_items: Optional[List[FaceDebuggerItem]] = ['kps', 'face-mask']
import facefusion.globals as globals
import facefusion.processors.frame.globals as frame_processors_globals
def print_globals():
print(f'execution_providers: {globals.execution_providers}')
print(f'execution_thread_count: {globals.execution_thread_count}')
print(f'execution_queue_count: {globals.execution_queue_count}')
print(f'max_memory: {globals.max_memory}')
print(f'face_analyser_order: {globals.face_analyser_order}')
print(f'face_analyser_age: {globals.face_analyser_age}')
print(f'face_analyser_gender: {globals.face_analyser_gender}')
print(f'face_detector_model: {globals.face_detector_model}')
print(f'face_detector_size: {globals.face_detector_size}')
print(f'face_detector_score: {globals.face_detector_score}')
print(f'face_recognizer_model: {globals.face_recognizer_model}')
print(f'face_selector_mode: {globals.face_selector_mode}')
print(f'reference_face_position: {globals.reference_face_position}')
print(f'reference_face_distance: {globals.reference_face_distance}')
print(f'reference_frame_number: {globals.reference_frame_number}')
print(f'face_mask_blur: {globals.face_mask_blur}')
print(f'face_mask_padding: {globals.face_mask_padding}')
print(f'trim_frame_start: {globals.trim_frame_start}')
print(f'trim_frame_end: {globals.trim_frame_end}')
print(f'temp_frame_format: {globals.temp_frame_format}')
print(f'temp_frame_quality: {globals.temp_frame_quality}')
print(f'keep_temp: {globals.keep_temp}')
print(f'output_image_quality: {globals.output_image_quality}')
print(f'output_video_encoder: {globals.output_video_encoder}')
print(f'output_video_quality: {globals.output_video_quality}')
print(f'keep_fps: {globals.keep_fps}')
print(f'skip_audio: {globals.skip_audio}')
print(f'frame_processors: {globals.frame_processors}')
print(f'face_swapper_model: {frame_processors_globals.face_swapper_model}')
print(f'face_enhancer_model: {frame_processors_globals.face_enhancer_model}')
print(f'face_enhancer_blend: {frame_processors_globals.face_enhancer_blend}')
print(f'frame_enhancer_model: {frame_processors_globals.frame_enhancer_model}')
print(f'frame_enhancer_blend: {frame_processors_globals.frame_enhancer_blend}')
print(f'face_debugger_items: {frame_processors_globals.face_debugger_items}') |