david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
samplelib/PackedFaceset.py
+1
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) ]
samplelib/SampleHost.py
-4
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]