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

.

Browse source
This commit is contained in:
Colombo 2020-12-02 00:54:47 +04:00
parent a35076a171
commit a03493c4f7
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
samplelib/SampleGeneratorFaceXSeg.py
View file

@ -87,10 +87,10 @@ class SampleGeneratorFaceXSeg(SampleGeneratorBase):
if sample.seg_ie_polys.has_polys():
mask = np.zeros ((h,w,1), dtype=np.float32)
sample.seg_ie_polys.overlay_mask(mask)
elif sample.has_xseg_mask():
mask = sample.get_xseg_mask()
mask[mask < 0.5] = 0.0
mask[mask >= 0.5] = 1.0
#elif sample.has_xseg_mask():
# mask = sample.get_xseg_mask()
# mask[mask < 0.5] = 0.0
# mask[mask >= 0.5] = 1.0
else:
raise Exception(f'no mask in sample {sample.filename}')