mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 19:05:02 -04:00
b3ae6130ed
* Preparing GAN plugin * Adding multithreading for extract * Adding support for mmod human face detector * Adding face filter argument * Added process number argument to multiprocessing extractor. Fixed progressbar for multiprocessing. * Added tiff as image type. compression artefacts hurt my feelings. * Cleanup
16 lines
382 B
Python
16 lines
382 B
Python
import multiprocessing as mp
|
|
|
|
method = None
|
|
|
|
def pool_process(method_to_run, data, processes=None):
|
|
global method
|
|
if processes is None:
|
|
processes = mp.cpu_count()
|
|
method = method_to_run
|
|
pool = mp.Pool(processes=processes)
|
|
|
|
for i in pool.imap_unordered(runner, data):
|
|
yield i if i is not None else 0
|
|
|
|
def runner(item):
|
|
return method(item)
|