github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 13:09:56 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

dst masks showing wrong fillenames

Browse source
This commit is contained in:
seranus 2021-12-07 17:35:23 +01:00
commit b2d800e112
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/Model_Quick96/Model.py
View file

@ -319,7 +319,7 @@ class QModel(ModelBase):
DM = D[i]*target_dstm[i] DM = D[i]*target_dstm[i]
if filenames is not None and len(filenames) > 0: if filenames is not None and len(filenames) > 0:
SM = label_face_filename(SM, filenames[0][i]) 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]) ar = SM, SS[i], DM, DD[i]*DDM[i], SD[i]*(DDM[i]*SDM[i])
st_m.append ( np.concatenate ( ar, axis=1) ) st_m.append ( np.concatenate ( ar, axis=1) )

2
models/Model_SAEHD/Model.py
View file

@ -994,7 +994,7 @@ class SAEHDModel(ModelBase):
DM = D[i]*target_dstm[i] DM = D[i]*target_dstm[i]
if filenames is not None and len(filenames) > 0: if filenames is not None and len(filenames) > 0:
SM = label_face_filename(SM, filenames[0][i]) 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 ar = SM, SS[i]*SSM[i], DM, DD[i]*DDM[i], SD[i]*SD_mask
st_m.append ( np.concatenate ( ar, axis=1) ) st_m.append ( np.concatenate ( ar, axis=1) )