mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
33 lines
923 B
Python
33 lines
923 B
Python
from enum import IntEnum
|
|
|
|
class FaceType(IntEnum):
|
|
HALF = 0,
|
|
FULL = 1,
|
|
HEAD = 2,
|
|
AVATAR = 3, #centered nose only
|
|
MARK_ONLY = 4, #no align at all, just embedded faceinfo
|
|
QTY = 5
|
|
|
|
@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_list[face_type]
|
|
|
|
from_string_dict = {'half_face': FaceType.HALF,
|
|
'full_face': FaceType.FULL,
|
|
'head' : FaceType.HEAD,
|
|
'avatar' : FaceType.AVATAR,
|
|
'mark_only' : FaceType.MARK_ONLY,
|
|
}
|
|
to_string_list = [ 'half_face',
|
|
'full_face',
|
|
'head',
|
|
'avatar',
|
|
'mark_only'
|
|
]
|