From 53ecfe7a27832c5aebf0327d0e2dbaf7c7e26477 Mon Sep 17 00:00:00 2001 From: seranus Date: Tue, 7 Dec 2021 17:34:35 +0100 Subject: [PATCH 1/2] fixed file name labels --- samplelib/SampleGeneratorFace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samplelib/SampleGeneratorFace.py b/samplelib/SampleGeneratorFace.py index 68229c2..d7dec3a 100644 --- a/samplelib/SampleGeneratorFace.py +++ b/samplelib/SampleGeneratorFace.py @@ -115,9 +115,9 @@ class SampleGeneratorFace(SampleGeneratorBase): samples, index_host, ct_samples, ct_index_host = param bs = self.batch_size - filenames = [] while True: batches = None + filenames = [] indexes = index_host.multi_get(bs) ct_indexes = ct_index_host.multi_get(bs) if ct_samples is not None else None From b2d800e11235c595be624257908908d30d2727cf Mon Sep 17 00:00:00 2001 From: seranus Date: Tue, 7 Dec 2021 17:35:23 +0100 Subject: [PATCH 2/2] dst masks showing wrong fillenames --- models/Model_Quick96/Model.py | 2 +- models/Model_SAEHD/Model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/models/Model_Quick96/Model.py b/models/Model_Quick96/Model.py index a03d673..a215cb5 100644 --- a/models/Model_Quick96/Model.py +++ b/models/Model_Quick96/Model.py @@ -319,7 +319,7 @@ class QModel(ModelBase): DM = D[i]*target_dstm[i] if filenames is not None and len(filenames) > 0: SM = label_face_filename(SM, filenames[0][i]) - DM = label_face_filename(DM, filenames[0][i]) + DM = label_face_filename(DM, filenames[1][i]) ar = SM, SS[i], DM, DD[i]*DDM[i], SD[i]*(DDM[i]*SDM[i]) st_m.append ( np.concatenate ( ar, axis=1) ) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 817ce9d..51569ed 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -994,7 +994,7 @@ class SAEHDModel(ModelBase): DM = D[i]*target_dstm[i] if filenames is not None and len(filenames) > 0: SM = label_face_filename(SM, filenames[0][i]) - DM = label_face_filename(DM, filenames[0][i]) + DM = label_face_filename(DM, filenames[1][i]) ar = SM, SS[i]*SSM[i], DM, DD[i]*DDM[i], SD[i]*SD_mask st_m.append ( np.concatenate ( ar, axis=1) )