mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
f1d115b63b
Basic usage instruction: https://i.imgur.com/w7LkId2.jpg 'whole_face' requires skill in Adobe After Effects. For using whole_face you have to extract whole_face's by using 4) data_src extract whole_face and 5) data_dst extract whole_face Images will be extracted in 512 resolution, so they can be used for regular full_face's and half_face's. 'whole_face' covers whole area of face include forehead in training square, but training mask is still 'full_face' therefore it requires manual final masking and composing in Adobe After Effects. added option 'masked_training' This option is available only for 'whole_face' type. Default is ON. Masked training clips training area to full_face mask, thus network will train the faces properly. When the face is trained enough, disable this option to train all area of the frame. Merge with 'raw-rgb' mode, then use Adobe After Effects to manually mask, tune color, and compose whole face include forehead.
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from enum import IntEnum
|
|
|
|
class FaceType(IntEnum):
|
|
#enumerating in order "next contains prev"
|
|
HALF = 0
|
|
MID_FULL = 1
|
|
FULL = 2
|
|
FULL_NO_ALIGN = 3
|
|
WHOLE_FACE = 4
|
|
HEAD = 5
|
|
HEAD_NO_ALIGN = 6
|
|
|
|
MARK_ONLY = 10, #no align at all, just embedded faceinfo
|
|
|
|
@staticmethod
|
|
def fromString (s):
|
|
r = from_string_dict.get (s.lower())
|
|
if r is None:
|
|
raise Exception ('FaceType.fromString value error')
|
|
return r
|
|
|
|
@staticmethod
|
|
def toString (face_type):
|
|
return to_string_dict[face_type]
|
|
|
|
from_string_dict = {'half_face': FaceType.HALF,
|
|
'midfull_face': FaceType.MID_FULL,
|
|
'full_face': FaceType.FULL,
|
|
'whole_face': FaceType.WHOLE_FACE,
|
|
'head' : FaceType.HEAD,
|
|
'mark_only' : FaceType.MARK_ONLY,
|
|
'full_face_no_align' : FaceType.FULL_NO_ALIGN,
|
|
'head_no_align' : FaceType.HEAD_NO_ALIGN,
|
|
}
|
|
to_string_dict = { FaceType.HALF : 'half_face',
|
|
FaceType.MID_FULL : 'midfull_face',
|
|
FaceType.FULL : 'full_face',
|
|
FaceType.WHOLE_FACE : 'whole_face',
|
|
FaceType.HEAD : 'head',
|
|
FaceType.MARK_ONLY :'mark_only',
|
|
FaceType.FULL_NO_ALIGN : 'full_face_no_align',
|
|
FaceType.HEAD_NO_ALIGN : 'head_no_align'
|
|
}
|