python队列syn中的Opencv多处理

2024-04-29 12:20:28 发布

您现在位置:Python中文网/ 问答频道 /正文

我是一个在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()

编辑

多亏有了答案,我才找到问题所在。下面是一个经过修改的脚本,其中我故意减慢了循环速度,并为调试添加了一个outqueue。虽然用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()

Tags: fromimageimportifqueueasdisplaytaskqueue