more stable and precise version of face transformation matrix.

fixed bleeding mask on some samples
This commit is contained in:
Colombo 2019-12-20 10:30:49 +04:00
parent 068c7d0d55
commit 64021b9c62
5 changed files with 68 additions and 57 deletions

View file

@ -54,7 +54,7 @@ class SampleGeneratorFace(SampleGeneratorBase):
if self.samples_len == 0:
raise ValueError('No training data provided.')
ct_samples = SampleLoader.load (SampleType.FACE, random_ct_samples_path) if random_ct_samples_path is not None else None
ct_samples = SampleLoader.load (SampleType.FACE, random_ct_samples_path, use_caching=use_caching) if random_ct_samples_path is not None else None
if self.debug:
self.generators_count = 1