CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Colombo
2020-12-02 00:55:27 +04:00
parent a03493c4f7
commit bbf3a71a96
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samplelib/SampleGeneratorFaceXSeg.py
+4 -4
View File
@@ -87,10 +87,10 @@ class SampleGeneratorFaceXSeg(SampleGeneratorBase):
if sample.seg_ie_polys.has_polys(): if sample.seg_ie_polys.has_polys():
mask = np.zeros ((h,w,1), dtype=np.float32) mask = np.zeros ((h,w,1), dtype=np.float32)
sample.seg_ie_polys.overlay_mask(mask) sample.seg_ie_polys.overlay_mask(mask)
#elif sample.has_xseg_mask(): elif sample.has_xseg_mask():
# mask = sample.get_xseg_mask() mask = sample.get_xseg_mask()
# mask[mask < 0.5] = 0.0 mask[mask < 0.5] = 0.0
# mask[mask >= 0.5] = 1.0 mask[mask >= 0.5] = 1.0
else: else:
raise Exception(f'no mask in sample {sample.filename}') raise Exception(f'no mask in sample {sample.filename}')