AISHACK Morph

ErosionDilation.py

def _dilate(self, img, str_img=[]):
  '''Expands the poindary of img
  '''
  w, h = img.shape
  res = np.asarray([[0 for _ in range(h)] for _ in range(w)])
  # Iterating the anchor of the structure image over the original image
  for i, im_row in enumerate(img):
      for j, im_ele in enumerate(im_row):
          if img[i][j]:
              for k in range(-1, 2):
                  for l in range(-1, 2):
                      if (i+k) >= w or (i+k) < 0 or (l+j) >= h or (l+j) < 0:
                          continue
                      else:
                          res[i+k][j+l] = 255

  return(res)

def _erode(self, img, str_img=[]):
  '''Contracts the boundary of img
  '''
  w, h = img.shape
  se_w, se_h = str_img.shape
  res = np.asarray([[0 for _ in range(h)] for _ in range(w)])
  for i, im_row in enumerate(img):
      for j, im_ele in enumerate(im_row):
          flag = True
          for k in range(-1, 2):
              for l in range(-1, 2):
                  if (i+k) >= w or (i+k) < 0 or (l+j) >= h or (l+j) < 0:
                      continue
                  if str_img[k][l] != 0 and img[i+k][j+l] == 0:
                      flag = False
          if flag:
              res[i][j] = 255

  return(np.asarray(res))