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
