Morphological-Operations/project4.py at master · pshrest21/Morphological-Operations · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import cv2
import numpy as np

olemiss_original = cv2.imread('olemiss.jpg',0)
#cv2.imshow('olemiss.jpg',olemiss)

ret, olemiss = cv2.threshold(olemiss_original, 120, 255, cv2.THRESH_BINARY)

def minkowski_subtraction(img):
    for i in range(1, len(img)-1):
        for j in range(1, len(img[i])-1):

            if (img[i-1][j-1] == 255 and img[i-1][j] == 255 and img[i-1][j+1]  == 255 and
                img[i][j-1] == 255 and img[i][j] == 255 and img[i][j+1] == 255 and
                img[i+1][j-1] == 255 and img[i+1][j] == img[i+1][j+1] == 255):
                print(i, j)
                img[i][j] = 255

            else:
                img[i][j] = 0
    return img


def erosion(img, kernel, n):
    #erosion using opencv inbuilt function cv2.erode()
    erosion = cv2.erode(img, kernel, iterations = n)
    return erosion

def dialation(img, kernel, n):
    dialation = cv2.dilate(img, kernel, iterations = n)
    return dialation

def opening(img, kernel):
    opening = cv2.morphologyEx(img, cv2.MORPH_OPEN, kernel)
    return opening

def closing(img, kernel):
    closing = cv2.morphologyEx(img, cv2.MORPH_CLOSE, kernel)
    return closing

def border1(img, kernel):
    erode = erosion(img, kernel, 1)

    for i in range(0, len(img)):
        for j in range(0, len(img[i])):
            if(img[i][j] != erode[i][j]):
                erode[i][j] = 255
            else:
                erode[i][j] = 0

    return erode


def border2(img, kernel):
    dilate = dialation(img, kernel, 1)

    for i in range(0, len(img)):
        for j in range(0, len(img[i])):
            if(img[i][j] != dilate[i][j]):
                dilate[i][j] = 255
            else:
                dilate[i][j] = 0

    return dilate

def border3(img, kernel):
    dilate = dialation(img, kernel, 2)
    erode = erosion(img, kernel, 1)

    temp = np.zeros((327, 452), dtype = np.uint8)

    for i in range(0, len(dilate)):
        for j in range(0, len(dilate[i])):
           if(dilate[i][j] != erode[i][j]):
               temp[i][j] = 255
           else:
               temp[i][j] = 0

    new_erode = erosion(temp, kernel, 1)

    return new_erode

'''
cv2.imshow('original',olemiss)
erode = minkowski_subtraction(olemiss)
cv2.imshow('erosion', erode)
'''
#set up a 3x3 structuring element
kernel = np.ones((3,3), np.uint8)

erode = erosion(olemiss, kernel, 2)
#cv2.imshow('Eroded', erode)
cv2.imwrite('eroded.jpg', erode)

dilate = dialation(olemiss, kernel, 2)
#cv2.imshow('Dilated', dilate)
cv2.imwrite('dilated.jpg', dilate)

opened = opening(olemiss, kernel)
#cv2.imshow('Opened', opened)
cv2.imwrite('opened.jpg', opened)

closed = closing(olemiss, kernel)
#cv2.imshow('Closed', closed)
cv2.imwrite('closed.jpg', closed)

#border detection
border_image1 = border1(olemiss, kernel)
#cv2.imshow('Border Image1', border_image1)
cv2.imwrite('Border Image1.jpg', border_image1)

border_image2 = border2(olemiss, kernel)
#cv2.imshow('Border Image 2', border_image2)
cv2.imwrite('Border Image2.jpg', border_image2)

border_image3 = border3(olemiss, kernel)
#cv2.imshow('Border Image 3', border_image3)
cv2.imwrite('Border Image3.jpg', border_image3)


cv2.waitKey(0)
cv2.destroyAllWindows()