python中的Opencv多处理 - 队列同步

时间:2013-06-11 20:39:05

标签: python opencv multiprocessing

我是Python中的多处理初学者。我正在为OpenCV开发一个多处理脚本,因为我的计算机没能成功处理OpenCV帧。 我的目标是在主进程中加载​​和处理帧,并使用子进程显示它们。我的问题是我不明白如何从排队的框架构建显示循环。有人可以帮忙吗?

我的代码:

#!/usr/bin/env python
from multiprocessing import Process, Queue
from Queue import Empty
from PIL import Image
import cv2
import cv2.cv as cv
import numpy as np

def image_display(taskqueue):
   cv2.namedWindow ('image_display', cv2.CV_WINDOW_AUTOSIZE)
   while True:
      if taskqueue.get()==None:
         continue
      else:
         image = taskqueue.get()
         im = Image.fromstring(image['mode'], image['size'], image['pixels'])
         num_im = np.asarray(im)
         cv2.imshow ('image_display', num_im)


if __name__ == '__main__':
   taskqueue = Queue()
   vidFile = cv2.VideoCapture('doppler.wmv')
   p = Process(target=image_display, args=(taskqueue,))
   p.start()
   while True:
      flag, image=vidFile.read()
      if flag == 0:
         break
      im = Image.fromarray(image)
      im_dict = {
      'pixels': im.tostring(),
      'size': im.size,
      'mode': im.mode,
      }
      taskqueue.put(im_dict)

p.join()
cv.DestroyAllWindows()

修改

感谢答案,我找到了问题所在。下面是一个经过修改的脚本,其中我故意减慢了我的循环,并且添加了一个用于调试的队列。虽然用vidFile.read()捕获的帧确实通过队列作为numpy数组传递,然后作为参数未经修改地传递给cv2.imshow() cv2.imshow()拒绝显示图像一个未知的原因。任何有助于解决这个问题的帮助都会受到极大的赞赏!

修改后的代码:

#!/usr/bin/env python
from multiprocessing import Process, Queue
from Queue import Empty
import cv2
import cv2.cv as cv
import numpy as np
import time

def image_display(taskqueue, outqueue):
   cv2.namedWindow ('image_display', cv2.CV_WINDOW_AUTOSIZE)
   while True:
      try:
         outqueue.put('trying')
         time.sleep(1)
         image = taskqueue.get()
         outqueue.put(image)
         cv2.imshow('image_display', image)
      except:
         continue



if __name__ == '__main__':
   taskqueue = Queue()
   outqueue = Queue()
   vidFile = cv2.VideoCapture('doppler.wmv')
   p = Process(target=image_display, args=(taskqueue, outqueue))
   p.start()
   while True:
      print outqueue.get()
      flag, image=vidFile.read()
      if flag == 0:
         break
      taskqueue.put(image)
      time.sleep(0.010)

p.join()
cv.DestroyAllWindows()

1 个答案:

答案 0 :(得分:1)

这应该有效(下面的修改说明):

#!/usr/bin/env python
from multiprocessing import Process, Queue
from Queue import Empty
from PIL import Image
import cv2
import cv2.cv as cv
import numpy as np

def image_display(taskqueue):
   cv2.namedWindow ('image_display', cv2.CV_WINDOW_AUTOSIZE)
   while True:

      image = taskqueue.get()              # Added
      if image is None:  break             # Added
      cv2.imshow ('image_display', image)  # Added
      cv2.waitKey(10)                      # Added
      continue                             # Added

      if taskqueue.get()==None:
         continue
      else:
         image = taskqueue.get()
         im = Image.fromstring(image['mode'], image['size'], image['pixels'])
         num_im = np.asarray(im)
         cv2.imshow ('image_display', num_im)


if __name__ == '__main__':
   taskqueue = Queue()
   vidFile = cv2.VideoCapture('doppler.wmv')
   p = Process(target=image_display, args=(taskqueue,))
   p.start()
   while True:
      flag, image=vidFile.read()

      taskqueue.put(image)  # Added
      import time           # Added
      time.sleep(0.010)     # Added
      continue              # Added

      if flag == 0:
         break
      im = Image.fromarray(image)
      im_dict = {
      'pixels': im.tostring(),
      'size': im.size,
      'mode': im.mode,
      }
      taskqueue.put(im_dict)

taskqueue.put(None)
p.join()
cv.DestroyAllWindows()

我尝试通过添加行(包含注释# Added的行)对代码进行最小的更改:

1)只需将图像本身(原始的NumPy数组)放在队列中即可。

2)在读取另一帧之前暂停主进程中的一点点。您需要这样做以免超出队列,因为生成的进程中的imshow()可能需要更长的时间,因为它正在调用X.您可能需要根据您的系统增加此值(以秒为单位)。

3)产生的过程必须在每waitKey()之后执行imshow()

4)主进程在完成后将特殊None图像放入队列。