Set an application icon.
[master-thesis.git] / src / Tests_hough / HTCircleGradient.m
1 % Hough Transform for circles with gradient.
2 % img : gradient image
3 % radius_range is a vector : [min, max]
4 function [acc_votes, acc_radius] = HTCircleGradient(img, radius_range)
5
6     [rows, columns] = size(img);
7     
8     acc_votes = zeros(rows, columns);
9     acc_radius = zeros(rows, columns);
10     
11     phi_range = linspace(0, 2 * pi, 100);
12     phi_range(end) = [];
13     
14     indexes_non_zero = find(img)';
15     s = size(img);
16     
17     for r = radius_range(1):radius_range(2)
18         acc = zeros(rows, columns);
19         for i = indexes_non_zero            
20             [y, x] = ind2sub(s, i);  
21             for phi = phi_range
22                 x0 = round(x - r * cos(phi));
23                 y0 = round(y - r * sin(phi));
24                 if x0 < columns && x0 > 0 && y0 < rows && y0 > 0
25                     acc(y0, x0) = acc(y0, x0) + img(y, x);
26                 end
27             end
28         end
29         
30         for x = 1:columns
31             for y = 1:rows
32                 if acc(y, x) > acc_votes(y, x)
33                     acc_votes(y, x) = acc(y, x);
34                     acc_radius(y, x) = r;
35                 end
36             end
37         end        
38     end
39     
40     acc_votes = imgaussfilt(acc_votes, 1.0);
41 end
42