mirror of https://github.com/opencv/opencv.git
parent
513080ea12
commit
eb28bef2db
1 changed files with 63 additions and 0 deletions
@ -0,0 +1,63 @@ |
|||||||
|
import numpy as np |
||||||
|
import cv2 |
||||||
|
|
||||||
|
from Queue import Queue |
||||||
|
from threading import Thread |
||||||
|
from collections import deque |
||||||
|
|
||||||
|
class Worker(Thread): |
||||||
|
def __init__(self, tasks): |
||||||
|
Thread.__init__(self) |
||||||
|
self.tasks = tasks |
||||||
|
self.daemon = True |
||||||
|
self.start() |
||||||
|
|
||||||
|
def run(self): |
||||||
|
while True: |
||||||
|
func, args, kargs = self.tasks.get() |
||||||
|
try: func(*args, **kargs) |
||||||
|
except Exception, e: print e |
||||||
|
self.tasks.task_done() |
||||||
|
|
||||||
|
class ThreadPool: |
||||||
|
def __init__(self, num_threads): |
||||||
|
self.tasks = Queue(num_threads) |
||||||
|
for _ in range(num_threads): Worker(self.tasks) |
||||||
|
|
||||||
|
def add_task(self, func, *args, **kargs): |
||||||
|
self.tasks.put((func, args, kargs)) |
||||||
|
|
||||||
|
def wait_completion(self): |
||||||
|
self.tasks.join() |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
results = deque() |
||||||
|
|
||||||
|
def process_frame(i, frame): |
||||||
|
global results |
||||||
|
res = cv2.medianBlur(frame, 15) |
||||||
|
results.append((i, res)) |
||||||
|
|
||||||
|
pool = ThreadPool(4) |
||||||
|
cap = cv2.VideoCapture(0) |
||||||
|
frame_count = 0 |
||||||
|
last_frame = None |
||||||
|
last_count = -1 |
||||||
|
while True: |
||||||
|
ret, frame = cap.read() |
||||||
|
pool.add_task(process_frame, frame_count, frame.copy()) |
||||||
|
frame_count += 1 |
||||||
|
while len(results) > 0: |
||||||
|
i, frame = results.popleft() |
||||||
|
if i > last_count: |
||||||
|
last_count, last_frame = i, frame |
||||||
|
if last_frame is not None: |
||||||
|
cv2.imshow('res', last_frame) |
||||||
|
if cv2.waitKey(1) == 27: |
||||||
|
break |
||||||
|
|
||||||
|
pool.wait_completion() |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in new issue