dilation.py

# Dilation

import numpy as np
import matplotlib.pyplot as plt
from scipy import misc

def dilation(image, structuring_element):
    # Find the number of rows and columns of both
    # image and the structuring element
    (image_row, image_col) = image.shape
    (element_row, element_col) = structuring_element.shape

    # Assign the original image as the dilated image initially
    dilated_image = np.copy(image)

    for i in range(image_row - element_row + 1):
        for j in range(image_col - element_col + 1):
            # Define the overlapping mask
            mask = image[i:i+element_row, j:j+element_col]
            # Check if there are any overlaps
            # If there are overlaps, overlap > 0
            # Otherwise overlap will be zero
            overlap = np.sum(structuring_element * mask)
            if overlap > 0:
                dilated_image[i, j] = 255
            else:
                dilated_image[i, j] = 0

    return dilated_image

def threshold(image):
    new_image = np.zeros(image.shape)
    (row, col) = image.shape
    for i in range(row):
        for j in range(col):
            if image[i, j] > 128:
                new_image[i, j] = 255
    return new_image

def show_images(image1, image2, title1, title2):
    f, a = plt.subplots(1, 2)
    a[0].imshow(image1, cmap='gray')
    a[0].set_title(title1)
    a[1].imshow(image2, cmap='gray')
    a[1].set_title(title2)
    plt.show()

structuring_element = np.array([
        [1, 1, 1],
        [1, 1, 1],
        [1, 1, 1]
    ])

image = misc.imread('255j.jpg')
image = threshold(image)
dil = dilation(image, structuring_element)

show_images(image, dil, 'Original Image', 'Dilated Image')