def onmouse(self, event, x, y, flags, param): if self.lock: if event == cv2.EVENT_RBUTTONDOWN: self.pop_camshift() return if event == cv2.EVENT_LBUTTONDOWN: self.drag_start = (x, y) self.newcamshift=mycamshift() if self.drag_start: xmin = min(x, self.drag_start[0]) ymin = min(y, self.drag_start[1]) xmax = max(x, self.drag_start[0]) ymax = max(y, self.drag_start[1]) self.selection=(xmin, ymin, xmax, ymax) if event == cv2.EVENT_LBUTTONUP: self.fps.reset() self.drag_start = None if self.newcamshift is not None and self.newcamshift.getHist() is not None: self.newcamshift.ID=len(self.list_camshift) self.list_camshift.append(self.newcamshift) self.newcamshift=None self.selection=None
def get_light(self): temp = mycamshift() temp.prProcess_light(self.frame) temp.ID = 99 return temp
def creat_camshift_from_img(hsv): camshift = mycamshift() mask = np.ones((hsv.shape[0], hsv.shape[1]), dtype=np.uint8) camshift.preProcess(hsv, mask, (0, 0, hsv.shape[1], hsv.shape[0]), 32) return camshift