qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-26 23:34:13 +01:00
Code Issues Projects Releases Wiki Activity

replace loky with concurrent.futures.ProcessPoolExecutor (faster)

Browse source
This commit is contained in:
Robert Sachunsky 2025-09-21 21:35:22 +02:00
parent 0366707136
commit 758602403e
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/eynollah/eynollah.py
View file

@ -23,7 +23,7 @@ import gc
import copy
import json
from loky import ProcessPoolExecutor
from concurrent.futures import ProcessPoolExecutor
import xml.etree.ElementTree as ET
import cv2
import numpy as np
@ -244,7 +244,7 @@ class Eynollah:
self.num_col_lower = num_col_lower
self.logger = logger if logger else getLogger('eynollah')
# for parallelization of CPU-intensive tasks:
self.executor = ProcessPoolExecutor(max_workers=cpu_count(), timeout=1200)
self.executor = ProcessPoolExecutor(max_workers=cpu_count())
atexit.register(self.executor.shutdown)
self.dir_models = dir_models
self.model_dir_of_enhancement = dir_models + "/eynollah-enhancement_20210425"