File size: 1,169 Bytes
632bd54
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
class ProcessOptions:
    def __init__(

        self,

        swap_model,

        processordefines: dict,

        face_distance,

        blend_ratio,

        swap_mode,

        selected_index,

        masking_text,

        imagemask,

        num_steps,

        subsample_size,

        show_face_area,

        restore_original_mouth,

        show_mask=False,

    ):
        if swap_model is not None:
            self.swap_modelname = swap_model
            self.swap_output_size = int(swap_model.split()[-1])
        else:
            self.swap_output_size = 128
        self.processors = processordefines
        self.face_distance_threshold = face_distance
        self.blend_ratio = blend_ratio
        self.swap_mode = swap_mode
        self.selected_index = selected_index
        self.masking_text = masking_text
        self.imagemask = imagemask
        self.num_swap_steps = num_steps
        self.show_face_area_overlay = show_face_area
        self.show_face_masking = show_mask
        self.subsample_size = subsample_size
        self.restore_original_mouth = restore_original_mouth
        self.max_num_reuse_frame = 15