From 2429d28737e82a8660ef3e4bdee225640763a809 Mon Sep 17 00:00:00 2001 From: Colombo Date: Sat, 4 Jan 2020 23:51:33 +0400 Subject: [PATCH] optimize memory usage --- samplelib/SampleHost.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/samplelib/SampleHost.py b/samplelib/SampleHost.py index ad16dfa..b240d07 100644 --- a/samplelib/SampleHost.py +++ b/samplelib/SampleHost.py @@ -1,5 +1,7 @@ +import gc import multiprocessing import operator +import pickle import traceback from pathlib import Path @@ -58,6 +60,11 @@ class SampleHost: if result is None: result = SampleHost.load_face_samples( Path_utils.get_image_paths(samples_path) ) + result_dmp = pickle.dumps(result) + del result + gc.collect() + result = pickle.loads(result_dmp) + samples[sample_type] = result elif sample_type == SampleType.FACE_TEMPORAL_SORTED: