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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
function [selectedBbox,selectedScore,varargout]=selectStrongestBbox(bBox,score,varargin)
// Selecting strongest bounding boxes
//
// Calling Sequence
// [selectedBbox, selectedScore]= selectStrongestBbox(bBox,score);
// [selectedBbox, selectedScore]= selectStrongestBbox(bBox,score,Name,Value);
// [selectedBbox, selectedScore, selectionIndex]= selectStrongestBbox(bBox,score,Name,Value);
//
// Parameters
// bBox: Each row represents one bounding box specified as [x y width height];
// score: Confidence Score of bounding box
// varargin: optinal (Name,Value) pair arguments
// Optional arguments can be
// <itemizedlist>
// <listitem><para>RatioType: method to compute the ratio of the intersection area between two boxes and it is specified as string, possile values are 'Union', 'Min' </para></listitem>
// <listitem><para>overlapThreshold: It specifies maximum overlap ratio, if overlap ration is more than this corresponding bounding box will be removed, possible range is 0 to 1</para></listitem>
// <itemizedlist>
//
// Description
// Returns strongest bounding boxes as per the given RatioType and OverlapThreshold and additionally it returns the index of the selected boxes
//
// Examples
// box(1)=[100 200 150 140];
// box(2)=[100 180 140 160];
// score(1)=0.03;
// score(2)=0.05;
// [sb ss]=selectStrongestBbox(box,score);
[lhs,rhs]=argn(0)
if rhs<2 then
error(msprintf(" Not enough input arguments"))
elseif rhs>6 then
error(msprintf(" Too many input arguments to the function"))
elseif lhs<2 then
error(msprintf(" Not enough output arguments"))
elseif lhs>3 then
error(msprintf(" Too many output arguments"))
end
[bBoxRows bBoxCols]=size(bBox);
[scoreRows scoreCols]=size(score);
if ~bBoxCols==4 then
error(msprintf("bounding box matrix must be M*4"))
elseif ~scoreCols==1 then
error(msprintf(" score matrix must be M*1"))
elseif ~bBoxRows==scoreRows then
error(msprintf(" The number of bounding boxes and scores should be same"))
elseif ~isreal(bBox)
error(msprintf(" Wrong input argument,complex matrix is not expected"))
end
for i=1:bBoxRows
if bBox(i,3)<0 | bBox(i,4)<0
error(msprintf(" The width and height of the bounded box must be positive"))
end
end
ratioType=1;
overlapThreshold=0.5;
for i=1:2:rhs-2
if strcmpi(varargin(i),"RatioType")==0 then
i=i+1;
if strcmpi(varargin(i),'union')== 0 then
ratioType=1;//1-union
elseif strcmpi(varargin(i),'min')==0 then
ratioType=0;//0-min
else
error(msprintf(" wrong value for the ratio type,it must be union or min"))
end
elseif strcmpi(varargin(i),'OverlapThreshold')==0 then
i=i+1;
if 0<=varargin(i) & varargin(i)<=1 then
overlapThreshold=varargin(i)
else
error(msprintf("wrong value for the overlapThreshold,it must be in between 0 and 1"))
end
else
error(msprintf(_(" Wrong value for input argument")));
end
end
for i=1:bBoxRows
index(i)=i;
end
score_temp=score(:,1);
for i=1:bBoxRows
for j=i+1:bBoxRows
if score_temp(i,1)<score_temp(j,1)
temp=score_temp(i,1);
score_temp(i,1)=score_temp(j,1);
score_temp(j,1)=temp;
temp=index(i);
index(i)=index(j);
index(j)=temp;
end
end
end
bBox_temp=bBox(index,:);
selection=ones(size(bBox_temp,1),1);
area = bBox_temp(:,3).*bBox_temp(:,4);
x1 = bBox_temp(:,1);
x2 = bBox_temp(:,1)+bBox_temp(:,3);
y1 = bBox_temp(:,2);
y2 = bBox_temp(:,2)+bBox_temp(:,4);
for i = 1:bBoxRows
if selection(i)
for j = (i+1):bBoxRows
if selection(j)
width = min(x2(i), x2(j)) - max(x1(i), x1(j));
height = min(y2(i), y2(j)) - max(y1(i), y1(j));
intersectionArea = width * height;
if ratioType
overlapRatio = intersectionArea/(area(i)+area(j)-intersectionArea);
else
overlapRatio = intersectionArea/min(area(i), area(j));
end
if overlapRatio > overlapThreshold
selection(j) = 0;
end
end
end
end
end
k=1;
for i=1:bBoxRows
if selection(i)
selectionIndex(k)=i;
indexOriginal(k)=index(i);
k=k+1;
end
end
//disp(indexOriginal);
[selectedIndexRows selectedIndexRows]=size(selectionIndex)
for i=1:selectedIndexRows
for j=i+1:selectedIndexRows
if indexOriginal(i)>indexOriginal(j)
temp=indexOriginal(i);
indexOriginal(i)=indexOriginal(j)
indexOriginal(j)=temp;
end
end
end
//disp("original indexes");
//disp(indexOriginal);
selectedBbox=bBox(indexOriginal,:);
selectedScore=score(indexOriginal,:);
varargout(1)=indexOriginal;
endfunction
|
