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

fixes

Browse source
This commit is contained in:
Colombo 2020-01-26 12:56:21 +04:00
parent 76ca79216e
commit c485e1718a
4 changed files with 10 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

7
models/Model_Quick96/Model.py
View file

@ -413,18 +413,15 @@ class QModel(ModelBase):
#override
def onGetPreview(self, samples):
n_samples = min(4, self.get_batch_size() )
( (warped_src, target_src, target_srcm),
(warped_dst, target_dst, target_dstm) ) = \
[ [sample[0:n_samples] for sample in sample_list ]
for sample_list in samples ]
(warped_dst, target_dst, target_dstm) ) = samples
S, D, SS, DD, DDM, SD, SDM = [ np.clip( nn.to_data_format(x,"NHWC", self.model_data_format), 0.0, 1.0) for x in ([target_src,target_dst] + self.AE_view (target_src, target_dst) ) ]
DDM, SDM, = [ np.repeat (x, (3,), -1) for x in [DDM, SDM] ]
target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )]
n_samples = min(4, self.get_batch_size() )
result = []
st = []
for i in range(n_samples):