generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

get rid of Lock in gabor_threads.py

Browse Source
This commit is contained in:
Alexander Mordvintsev 2012-09-17 19:30:20 +04:00
parent 5eac0419b2
commit 2a96f28bc1
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
samples/python2/gabor_threads.py
View File

@ -14,7 +14,6 @@ gabor_threads.py [image filename]
import numpy as np import numpy as np
import cv2 import cv2
from threading import Lock
from multiprocessing.pool import ThreadPool from multiprocessing.pool import ThreadPool
@ -36,13 +35,11 @@ def process(img, filters):
def process_threaded(img, filters, threadn = 8): def process_threaded(img, filters, threadn = 8):
accum = np.zeros_like(img) accum = np.zeros_like(img)
accum_lock = Lock()
def f(kern): def f(kern):
fimg = cv2.filter2D(img, cv2.CV_8UC3, kern) return cv2.filter2D(img, cv2.CV_8UC3, kern)
with accum_lock:
np.maximum(accum, fimg, accum)
pool = ThreadPool(processes=threadn) pool = ThreadPool(processes=threadn)
pool.map(f, filters) for fimg in pool.imap_unordered(f, filters):
np.maximum(accum, fimg, accum)
return accum return accum
if __name__ == '__main__': if __name__ == '__main__':