Digital-Image-Processing/Compass2.py at master · KKMonika/Digital-Image-Processing · 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
import cv2
import numpy as np
from matplotlib import pyplot as plt


img = cv2.imread('Lenna.png')
img =np.float32(img)

h1 = np.array(([1,1,0],[1, 0, -1], [0, -1 , -1]), dtype = 'float32')
fi1 = cv2.filter2D(img, -1, h1)
fi1 =abs(fi1)
fi1 = fi1/np.amax(fi1[:])

h2 = np.array(([1,1,1],[0, 0, 0], [-1, -1 , -1]), dtype = 'float32')
fi2 = cv2.filter2D(img, -1, h2)
fi2 =abs(fi2)
fi2 = fi2/np.amax(fi2[:])

h3 = np.array(([0,1,1],[-1, 0, 1], [-1, -1 , 0]), dtype = 'float32')
fi3 = cv2.filter2D(img, -1, h3)
fi3 =abs(fi3)
fi3 = fi3/np.amax(fi3[:])

h4 = np.array(([1,0,-1],[1, 0, -1], [1, 0 , -1]), dtype = 'float32')
fi4 = cv2.filter2D(img, -1, h4)
fi4 =abs(fi4)
fi4 = fi4/np.amax(fi4[:])

h5 = np.array(([0,-1,-1],[1, 0, -1], [1, 1 , 0]), dtype = 'float32')
fi5 = cv2.filter2D(img, -1, h5)
fi5 =abs(fi5)
fi5 = fi5/np.amax(fi5[:])

h6 = np.array(([-1,-1,-1],[0, 0, 0], [1, 1 , 1]), dtype = 'float32')
fi6 = cv2.filter2D(img, -1, h6)
fi6 =abs(fi6)
fi6 = fi6/np.amax(fi6[:])

h7 = np.array(([-1,-1,0],[-1, 0, 1], [0, 1 , 1]), dtype = 'float32')
fi7 = cv2.filter2D(img, -1, h7)
fi7 =abs(fi7)
fi7 = fi7/np.amax(fi7[:])

h8 = np.array(([-1,0,1],[-1, 0, 1], [-1, 0 , 1]), dtype = 'float32')
fi8 = cv2.filter2D(img, -1, h8)
fi8 =abs(fi8)
fi8 = fi8/np.amax(fi8[:])


edgeSum1 = cv2.add(np.power(fi1,2), np.power(fi2,2))
edgeSum2 = cv2.add(np.power(fi3, 2), np.power(fi4, 2))
edgeSum3 = cv2.add(np.power(fi5, 2), np.power(fi6, 2))
edgeSum4 = cv2.add(np.power(fi7, 2), np.power(fi8, 2))

sum1 = cv2.add(np.power(edgeSum1,2), np.power(edgeSum2, 2))
sum2 = cv2.add(np.power(edgeSum3,2), np.power(edgeSum4, 2))

sum = cv2.add(np.power(sum1,2), np.power(sum2,2))
ret, thresh1 = cv2.threshold(sum, 0.05, 1, cv2.THRESH_BINARY)

plt.figure(1)
plt.subplot(421), plt.imshow(fi1)
plt.subplot(422), plt.imshow(fi2)
plt.subplot(423), plt.imshow(fi3)
plt.subplot(424), plt.imshow(fi4)
plt.subplot(425), plt.imshow(fi5)
plt.subplot(426), plt.imshow(fi6)
plt.subplot(427), plt.imshow(fi7)
plt.subplot(428), plt.imshow(fi8)

plt.figure(2), plt.imshow(thresh1), plt.title("EDGES")

plt.show()