From b2d800e11235c595be624257908908d30d2727cf Mon Sep 17 00:00:00 2001 From: seranus Date: Tue, 7 Dec 2021 17:35:23 +0100 Subject: [PATCH] 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) )