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

refactoring

Browse source
This commit is contained in:
iperov 2018-12-24 18:41:36 +04:00
parent 8a223845fb
commit e4010d3eba
4 changed files with 216 additions and 146 deletions
Download patch file Download diff file Expand all files Collapse all files

8
samples/SampleProcessor.py
View file

@ -51,7 +51,7 @@ class SampleProcessor(object):
sample_rnd_seed = np.random.randint(0x80000000)
outputs = []
outputs = []
for sample_type in output_sample_types:
f = sample_type[0]
size = sample_type[1]
@ -139,13 +139,13 @@ class SampleProcessor(object):
outputs.append ( img )
if debug:
result = ()
result = []
for output in outputs:
if output.shape[2] < 4:
result += (output,)
result += [output,]
elif output.shape[2] == 4:
result += (output[...,0:3]*output[...,3:4],)
result += [output[...,0:3]*output[...,3:4],]
return result
else: