Filters/sobel.py at master · samacyc/Filters · 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
# -*- coding: utf-8 -*-
"""
Created on Tue Mar 26 09:00:01 2019

@author: Samacyc
"""

from PIL import Image
from scipy import misc
from skimage import data
import cv2
import numpy as np
import scipy.misc
def produit_mat(Matrice_1,Matrice_2):
    n = np.shape(Matrice_1)[0]
    Produit = np.zeros((n,n))

    for i in range(n) :
        for j in range (n) :
              Produit[i][j] = Matrice_1[i][j]*Matrice_2[i][j]

    return Produit

def somme (matrice) :
    somme = 0
    n = matrice.shape[0]
    for i in range(n) :
        for j in range (n) :
            somme += matrice[i][j]

    return somme

def reshaping(matrice) :
    line , row = np.shape(matrice)
    new_arr = np.zeros((line+2,row+2))

    line1 , row1 = np.shape(new_arr)
    for i in range (1,line1-1):
        for j in range(1,row1-1):
            new_arr[i][j]=matrice[i-1][j-1]
    for j in range(1,row1-1):
      new_arr[0][j] = matrice[1][j-1]
      new_arr[line1-1][j] = new_arr[line1-3][j]

    for i in range(line1):
      new_arr[i][0]=new_arr[i][2]
      new_arr[i][row1-1] = new_arr[i][row1-3]
    return new_arr

def decrease (matrice) :
    line , row = np.shape(matrice)
    new_arr = np.zeros((line-2,row-2))
    line1 , row1 = np.shape(new_arr)
    for i in range(line1):
        for j in range(row1) :
            new_arr[i][j]=matrice[i+1][j+1]

    return new_arr

def copy(matrice,x,y):
    new_arr = np.zeros((3,3))
    for i in range(3) :
        for j in range(3) :
            new_arr[i][j] = matrice[x-1+i][y-1+j]

    return new_arr

def final_func(matrice,matrice1,x,y):
    helper = np.zeros((3,3))
    helper = copy(matrice,x,y)

    helper = produit_mat(helper,matrice1)
    n = somme(helper)

    return float(n)


hx = np.array([[-1/4,0,1/4],[-1/2,0,1/2],[-1/4,0,1/4]])
hy = np.array([[-1/4,-1/2,-1/4],[0,0,0],[1/4,1/2,1/4]])
img = (data.coins())
img_outx = img.copy()
img_outy = img.copy()
line , row = img.shape
img_outx = reshaping(img_outx)
img_outy = reshaping(img_outy)
print(img_outx)
line1 , row1 = np.shape(img_outx)
helper = np.zeros((3,3))
for x in range(1,line1-1):
     for y in range (1,row-1) :
         img_outx[x][y] = final_func(img_outx,hx,x,y)
         img_outy[x][y]  = final_func(img_outy,hy,x,y)
img_outx = decrease(img_outx)
img_outy = decrease(img_outy)
cv2.imshow('image_x',img_outx)
cv2.imshow('image_y',img_outx)
cv2.imwrite('sobelx.png',img_outx)
cv2.imwrite('sobely.png',img_outx)

"""
img_outx = Image.fromarray(img_outx)
img_outx = img_outx.convert('RGB')
scipy.misc.imsave('sobelx.png', img_outx)
img_outy = Image.fromarray(img_outy)
img_outy = img_outy.convert('RGB')
scipy.misc.imsave('sobely.png', img_outy)
img_outx.show()
#img_outy.show()
"""