-
Notifications
You must be signed in to change notification settings - Fork 0
/
Steerable.py
71 lines (52 loc) · 1.99 KB
/
Steerable.py
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
# -*- coding: utf-8 -*-
"""
Created on Fri May 7 16:33:08 2021
@author: VArri
Edge Feature Detector with multiple steerable filters
"""
import cv2
# import matplotlib.pyplot as plt
import numpy as np
import scipy.signal as signal
import os
sigma=1.0
filter_size = 2 * int(4 * sigma + 0.5) + 1 # taille du filtre en fonction du sigma recherché
# Second order filter initialization
G2 = np.zeros((filter_size, filter_size, 3), np.float32) # création du filtre vide
m = filter_size//2
n = filter_size//2
for x in range(-m, m+1):
for y in range(-n, n+1):
# Second order
# le filtre est balayé de par le centre d'où la range selon x et selon y
G2[x+m, y+n,0] = 0.9213*(2*x**2-1)*np.exp(-(x**2 + y**2))
G2[x+m, y+n,1] = 1.843*x*y*np.exp(-(x**2 + y**2))
G2[x+m, y+n,2] = 0.9213*(2*y**2-1)*np.exp(-(x**2 + y**2))
os.chdir(r"C:\Users\VArri\Documents\PowerLines\images\visuel")
path = os.getcwd()
edge_dir = os.path.join(path, 'edge1000')
dirs = os.listdir(edge_dir)
for dir_image in dirs:
image_path = os.path.join(edge_dir, dir_image)
filename, file_extension = os.path.splitext(os.path.basename(dir_image))
img = cv2.imread(image_path)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#print(image_path)
#print(filename)
#print(file_extension)
#print(dir_image)
#break
Start=True
#for theta in [np.pi/2, np.pi/3, 2*np.pi/3]:
for theta in [np.pi/2]:
#print(theta)
kG2 = [np.cos(theta)**2, - 2*np.sin(theta)*np.cos(theta), np.sin(theta)**2]
G2comb = kG2[0]*G2[:,:,0] +kG2[1]*G2[:,:,1] +kG2[2]*G2[:,:,2]
Gray_i = signal.convolve2d(gray, G2comb)
if Start:
E_i = np.zeros(Gray_i.shape, np.float32)
Start = False
E_i = np.sqrt(np.square(E_i) + np.square(Gray_i))
output_dir = os.path.join(path,'steered1000', filename+'_steered.jpg')
cv2.imwrite(output_dir, E_i)
# break