qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-07 06:59:58 +02: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 copy
import json import json
from loky import ProcessPoolExecutor from concurrent.futures import ProcessPoolExecutor
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
import cv2 import cv2
import numpy as np import numpy as np
@ -244,7 +244,7 @@ class Eynollah:
self.num_col_lower = num_col_lower self.num_col_lower = num_col_lower
self.logger = logger if logger else getLogger('eynollah') self.logger = logger if logger else getLogger('eynollah')
# for parallelization of CPU-intensive tasks: # 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) atexit.register(self.executor.shutdown)
self.dir_models = dir_models self.dir_models = dir_models
self.model_dir_of_enhancement = dir_models + "/eynollah-enhancement_20210425" self.model_dir_of_enhancement = dir_models + "/eynollah-enhancement_20210425"