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

Merge pull request #10 from MachineEditor/preview_filenames

Browse source 
Preview filenames - fixes
This commit is contained in:
Ognjen 2021-12-07 17:36:49 +01:00 • committed by GitHub
commit 9cef60ce91
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 3 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) )

2
samplelib/SampleGeneratorFace.py
View file

@ -115,9 +115,9 @@ class SampleGeneratorFace(SampleGeneratorBase):
samples, index_host, ct_samples, ct_index_host = param samples, index_host, ct_samples, ct_index_host = param
bs = self.batch_size bs = self.batch_size
filenames = []
while True: while True:
batches = None batches = None
filenames = []
indexes = index_host.multi_get(bs) indexes = index_host.multi_get(bs)
ct_indexes = ct_index_host.multi_get(bs) if ct_samples is not None else None ct_indexes = ct_index_host.multi_get(bs) if ct_samples is not None else None