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

fix

Browse source
This commit is contained in:
Colombo 2020-01-05 14:21:34 +04:00
parent ea33541177
commit 842a48964f
2 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
samplelib/PackedFaceset.py
View file

@ -136,6 +136,7 @@ class PackedFaceset():
samples_configs = pickle.loads ( f.read(sizeof_samples_bytes) ) samples_configs = pickle.loads ( f.read(sizeof_samples_bytes) )
samples = [] samples = []
for sample_config in samples_configs: for sample_config in samples_configs:
sample_config = pickle.loads(pickle.dumps (sample_config))
samples.append ( Sample (**sample_config) ) samples.append ( Sample (**sample_config) )
offsets = [ struct.unpack("Q", f.read(8) )[0] for _ in range(len(samples)+1) ] offsets = [ struct.unpack("Q", f.read(8) )[0] for _ in range(len(samples)+1) ]

4
samplelib/SampleHost.py
View file

@ -1,6 +1,5 @@
import multiprocessing import multiprocessing
import operator import operator
import pickle
import traceback import traceback
from pathlib import Path from pathlib import Path
@ -66,9 +65,6 @@ class SampleHost:
if sample_type == SampleType.FACE_TEMPORAL_SORTED: if sample_type == SampleType.FACE_TEMPORAL_SORTED:
result = SampleHost.upgradeToFaceTemporalSortedSamples(result) result = SampleHost.upgradeToFaceTemporalSortedSamples(result)
result_dumped = pickle.dumps(result)
del result
result = pickle.loads(result_dumped)
samples[sample_type] = mp_utils.ListHost(result) samples[sample_type] = mp_utils.ListHost(result)
list_host = samples[sample_type] list_host = samples[sample_type]