From bbf3a71a961ad79cab726b7d08ed406683bc5619 Mon Sep 17 00:00:00 2001 From: Colombo Date: Wed, 2 Dec 2020 00:55:27 +0400 Subject: [PATCH] . --- samplelib/SampleGeneratorFaceXSeg.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/samplelib/SampleGeneratorFaceXSeg.py b/samplelib/SampleGeneratorFaceXSeg.py index 5cdffb1..9c7cf6e 100644 --- a/samplelib/SampleGeneratorFaceXSeg.py +++ b/samplelib/SampleGeneratorFaceXSeg.py @@ -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}')