def createglobalmask(total_blobs, bg_models): size = bg_models[0].bg_img.shape height = size[0] width = size[1] total_masks = [] for bg_blobs in total_blobs: global_mask = np.zeros((height, width), dtype=np.uint8) for blob in bg_blobs: blob.drawglobalmask(global_mask) total_masks.append( cv2.morphologyEx( global_mask, cv2.MORPH_CLOSE, cv2.getStructuringElement(cv2.MORPH_CROSS, (5, 10)))) """ cv2.imshow( "demo", cv2.morphologyEx( global_mask, cv2.MORPH_CLOSE, cv2.getStructuringElement( cv2.MORPH_CROSS, (5, 10)))) """ return total_masks
def __applymaskmorphologicaloperation(self): x, y, w, h = self.bound_rect # TODO --> print self.mean mean_x mean_y kernel col = int(h / 10) row = int(w / 10) # in case that kernel is 0 if col is 0: col = 1 if row is 0: row = 1 mask = self.mask kernel = np.ones((col, row), np.uint8) mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel, iterations=1) self.mask = cv2.erode(mask, kernel, iterations=1)
def __applymaskmorphologicaloperation(self): x, y, w, h = self.bound_rect # TODO --> print self.mean mean_x mean_y kernel col = int(h/10) row = int(w/10) # in case that kernel is 0 if col is 0: col = 1 if row is 0: row = 1 mask = self.mask kernel = np.ones((col, row), np.uint8) mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel, iterations=1) self.mask = cv2.erode(mask, kernel, iterations=1)
def createglobalmask(total_blobs, bg_models): size = bg_models[0].bg_img.shape height = size[0] width = size[1] total_masks = [] for bg_blobs in total_blobs: global_mask = np.zeros((height, width), dtype=np.uint8) for blob in bg_blobs: blob.drawglobalmask(global_mask) total_masks.append( cv2.morphologyEx( global_mask, cv2.MORPH_CLOSE, cv2.getStructuringElement( cv2.MORPH_CROSS, (5, 10)))) """ cv2.imshow( "demo", cv2.morphologyEx( global_mask, cv2.MORPH_CLOSE, cv2.getStructuringElement( cv2.MORPH_CROSS, (5, 10)))) """ return total_masks