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

Decreased amount of RAM used by Sample Generator.

Browse source
This commit is contained in:
Colombo 2020-04-05 13:52:32 +04:00
parent 8e9e346c9d
commit 33b0aadb4e
5 changed files with 221 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

8
samplelib/SampleLoader.py
View file

@ -6,6 +6,7 @@ from pathlib import Path
import samplelib.PackedFaceset
from core import pathex
from core.mplib import MPSharedList
from core.interact import interact as io
from core.joblib import Subprocessor
from DFLIMG import *
@ -33,6 +34,9 @@ class SampleLoader:
@staticmethod
def load(sample_type, samples_path, subdirs=False):
"""
Return MPSharedList of samples
"""
samples_cache = SampleLoader.samples_cache
if str(samples_path) not in samples_cache.keys():
@ -56,12 +60,12 @@ class SampleLoader:
if result is None:
result = SampleLoader.load_face_samples( pathex.get_image_paths(samples_path, subdirs=subdirs) )
samples[sample_type] = result
samples[sample_type] = MPSharedList(result)
elif sample_type == SampleType.FACE_TEMPORAL_SORTED:
result = SampleLoader.load (SampleType.FACE, samples_path)
result = SampleLoader.upgradeToFaceTemporalSortedSamples(result)
samples[sample_type] = result
samples[sample_type] = MPSharedList(result)
return samples[sample_type]