github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upd

Browse source
This commit is contained in:
Colombo 2020-11-22 18:26:54 +04:00
parent c516454566
commit e9e7344424
4 changed files with 38 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

3
samplelib/Sample.py
View file

@ -79,6 +79,9 @@ class Sample(object):
self._filename_offset_size = None
def has_xseg_mask(self):
return self.xseg_mask is not None or self.xseg_mask_compressed is not None
def get_xseg_mask(self):
if self.xseg_mask_compressed is not None:
xseg_mask = cv2.imdecode(self.xseg_mask_compressed, cv2.IMREAD_UNCHANGED)