Блог

Я пытаюсь запустить код, предоставленный opencv / samples / python2 для Camshift, используя потоки.Я создал два объекта для класса App (), которые вызывают метод run.

     ##
from threading import Thread
from multiprocessing import Process
import numpy as np
import cv2
import video


class App(object):
    def __init__(self, video_src):
         self.cam = video.create_capture(video_src)
         ret, self.frame = self.cam.read()
         cv2.namedWindow('camshift')
         cv2.setMouseCallback('camshift', self.onmouse)

          self.selection = None
          self.drag_start = None
          self.tracking_state = 0
          self.show_backproj = False

   def onmouse(self, event, x, y, flags, param):
         x, y = np.int16([x, y]) # BUG
         if event == cv2.EVENT_LBUTTONDOWN:
              self.drag_start = (x, y)
              self.tracking_state = 0
         if self.drag_start:
              if flags amp; cv2.EVENT_FLAG_LBUTTON:
                   h, w = self.frame.shape[:2]
                   xo, yo = self.drag_start
                   x0, y0 = np.maximum(0, np.minimum([xo, yo], [x, y]))
                   x1, y1 = np.minimum([w, h], np.maximum([xo, yo], [x, y]))
                   self.selection = None
                   if x1-x0 > 0 and y1-y0 > 0:
                        self.selection = (x0, y0, x1, y1)
             else:
                   self.drag_start = None
                  if self.selection is not None:
                         self.tracking_state = 1

    def show_hist(self):
        bin_count = self.hist.shape[0]
        bin_w = 24
        img = np.zeros((256, bin_count*bin_w, 3), np.uint8)
        for i in xrange(bin_count):
             h = int(self.hist[i])
             cv2.rectangle(img, (i*bin_w 2, 255), ((i 1)*bin_w-2, 255-h),                 (int(180.0*i/bin_count), 255, 255), -1)
        img = cv2.cvtColor(img, cv2.COLOR_HSV2BGR)
        cv2.imshow('hist', img)

   def run(self):
     while True:
        ret, self.frame = self.cam.read()
        vis = self.frame.copy()
        hsv = cv2.cvtColor(self.frame, cv2.COLOR_BGR2HSV)
        mask = cv2.inRange(hsv, np.array((0., 60., 32.)), np.array((180., 255., 255.)))

        if self.selection:
            x0, y0, x1, y1 = self.selection
            self.track_window = (x0, y0, x1-x0, y1-y0)
            hsv_roi = hsv[y0:y1, x0:x1]
            mask_roi = mask[y0:y1, x0:x1]
            hist = cv2.calcHist( [hsv_roi], [0], mask_roi, [16], [0, 180] )
            cv2.normalize(hist, hist, 0, 255, cv2.NORM_MINMAX);
            self.hist = hist.reshape(-1)
            #self.show_hist()
            vis_roi = vis[y0:y1, x0:x1]

            cv2.bitwise_not(vis_roi, vis_roi)
            vis[mask == 0] = 0

        if self.tracking_state == 1:
            self.selection = None
            prob = cv2.calcBackProject([hsv], [0], self.hist, [0, 180], 1)
            prob amp;= mask
            term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
            track_box, self.track_window = cv2.CamShift(prob, self.track_window, term_crit)
            print track_box[0]
            if self.show_backproj:
                vis[:] = prob[...,np.newaxis]
            try: cv2.ellipse(vis, track_box, (0, 0, 255), 2)
            except: print track_box

        cv2.imshow('camshift', vis)

        ch = 0xFF amp; cv2.waitKey(5)
        if ch == 27:
            break
        if ch == ord('b'):
            self.show_backproj = not self.show_backproj
    cv2.destroyAllWindows()


if __name__ == '__main__':
     import sys
     try: video_src = sys.argv[1]
     except: video_src = 0
     print __doc__
     left=App(1)# 1=device id for left camera
     right=App(2)#2=device id for right camera
     threadLeft=Process(target=left.run())
     threadRight=Process(target=right.run())
     threadRight.start()
     threadLeft.start()
  

При выполнении появляются два окна, одно для правого, а другое для левого.Однако onmouse() не вызывается, когда я перетаскиваю мышь в любом из окон.Кажется, что окна зависают.После чего, при закрытии любого из них, шаблон, который я ранее создал в окне для отслеживания любого объекта, автоматически используется во втором незакрытом окне.

Итак, есть ли какой-либо другой метод, с помощью которого я могу вызвать этот код для моих двух камер.Любая помощь приветствуется.Спасибо

Чтобы создать процесс, передайте функцию target . В приведенном выше коде left.run() выполняется, а не передается процессу.

Измените код на этот:

 threadLeft = Process(target=left.run)
threadRight = Process(target=right.run)