github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-14 10:46:59 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added 5.XSeg) data_dst/data_src mask for XSeg trainer - remove.bat

Browse source 
removes labeled xseg polygons from the extracted frames
This commit is contained in:
Colombo 2020-04-06 21:45:46 +04:00
commit 5b4d023712
3 changed files with 69 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

6
DFLIMG/DFLJPG.py
View file

@ -245,6 +245,9 @@ class DFLJPG(object):
return None
def set_image_to_face_mat(self, image_to_face_mat): self.dfl_dict['image_to_face_mat'] = image_to_face_mat
def has_seg_ie_polys(self):
return self.dfl_dict.get('seg_ie_polys',None) is not None
def get_seg_ie_polys(self):
d = self.dfl_dict.get('seg_ie_polys',None)
if d is not None:
@ -266,6 +269,9 @@ class DFLJPG(object):
self.dfl_dict['seg_ie_polys'] = seg_ie_polys
def has_xseg_mask(self):
return self.dfl_dict.get('xseg_mask',None) is not None
def get_xseg_mask(self):
mask_buf = self.dfl_dict.get('xseg_mask',None)
if mask_buf is None: