mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2024-12-25 07:21:13 -08:00
61472cdaf7
removed support of extracted(aligned) PNG faces. Use old builds to convert from PNG to JPG. fanseg model file in facelib/ is renamed
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import multiprocessing
|
|
from core.interact import interact as io
|
|
|
|
class MPClassFuncOnDemand():
|
|
def __init__(self, class_handle, class_func_name, **class_kwargs):
|
|
self.class_handle = class_handle
|
|
self.class_func_name = class_func_name
|
|
self.class_kwargs = class_kwargs
|
|
|
|
self.class_func = None
|
|
|
|
self.s2c = multiprocessing.Queue()
|
|
self.c2s = multiprocessing.Queue()
|
|
self.lock = multiprocessing.Lock()
|
|
|
|
io.add_process_messages_callback(self.io_callback)
|
|
|
|
def io_callback(self):
|
|
while not self.c2s.empty():
|
|
func_args, func_kwargs = self.c2s.get()
|
|
if self.class_func is None:
|
|
self.class_func = getattr( self.class_handle(**self.class_kwargs), self.class_func_name)
|
|
self.s2c.put ( self.class_func (*func_args, **func_kwargs) )
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
with self.lock:
|
|
self.c2s.put ( (args, kwargs) )
|
|
return self.s2c.get()
|
|
|
|
def __getstate__(self):
|
|
return {'s2c':self.s2c, 'c2s':self.c2s, 'lock':self.lock}
|
|
|