Dilation.py

# Note: Image name will be stored as "Dilation_OriginalName" to avoid confict

import cv2
import glob
import numpy as np 

# DEFINE THE PATH
PATH_TO_DEST = "/content/drive/My Drive/Main Folder/Dataset/Dilation_image/"
PATH_TO_ORIGIAL_IMAGES = "/content/drive/My Drive/Main Folder/Dataset/Orig_Image/"

# if the source directory have other files than images, use extenstion of image 
# to get the files ( for example *.png )
img_files = glob.glob(PATH_TO_ORIGIAL_IMAGES+"*.*")
total = len(img_files)

# 2x2 Static kernal
kernal = np.ones((2,2),np.uint8)

for count,i in enumerate(img_files):
  image_name = i.split("/")[-1]
  print("Progress : ",count,"/",total)
  img = cv2.imread(i,0)
  _, mask = cv2.threshold(img,220,255,cv2.THRESH_BINARY_INV)
  dst = cv2.dilate(mask,kernal,iterations = 1)
  dst = cv2.bitwise_not(dst)
  cv2.imwrite(PATH_TO_DEST+"/Dilation_"+image_name,dst)