func (t OverlapTest) Overlap(a, b image.Rectangle) bool {
if detect.Cover(a, b) >= t.MaxCover {
return true
}
if detect.IOU(a, b) >= t.MaxIOU {
return true
}
if detect.Cover(a, b) >= t.MaxCoverBoth && detect.Cover(b, a) >= t.MaxCoverBoth {
return true
}
return false
}
func TestSuppress(t *testing.T) {
cases := []struct {
MaxInter float64
MaxNum int
In, Out []detect.Det
}{
// Clear margin between windows.
{
0, 10,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
},
// Same and limit to four outputs.
{
0, 4,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
},
// Same and limit to three outputs.
{
0, 3,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(10, 0, 15, 5)},
{2, image.Rect(0, 10, 5, 15)},
},
},
// Touching but not overlapping.
{
0, 10,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(5, 0, 10, 5)},
{2, image.Rect(0, 5, 5, 10)},
{1, image.Rect(5, 5, 10, 10)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(5, 0, 10, 5)},
{2, image.Rect(0, 5, 5, 10)},
{1, image.Rect(5, 5, 10, 10)},
},
},
// All slightly overlapping.
{
0, 10,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(4, 0, 9, 5)},
{2, image.Rect(0, 4, 5, 9)},
{1, image.Rect(4, 4, 9, 9)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
},
},
// B and C overlapping A and D. Output A and D.
{
0, 10,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(3, 0, 15, 15)},
{2, image.Rect(0, 3, 15, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{1, image.Rect(10, 10, 15, 15)},
},
},
// Same, limit to two outputs.
{
0, 2,
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{3, image.Rect(3, 0, 15, 15)},
{2, image.Rect(0, 3, 15, 15)},
{1, image.Rect(10, 10, 15, 15)},
},
[]detect.Det{
{4, image.Rect(0, 0, 5, 5)},
{1, image.Rect(10, 10, 15, 15)},
},
},
// Test intersection threshold.
{
0.5, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(1, 0, 6, 5)},
{3, image.Rect(2, 0, 7, 5)},
{2, image.Rect(3, 0, 8, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{2, image.Rect(3, 0, 8, 5)},
},
},
{
0.1, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(1, 0, 6, 5)},
{3, image.Rect(2, 0, 7, 5)},
{2, image.Rect(3, 0, 8, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
},
},
{
0.3, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(1, 0, 6, 5)},
{3, image.Rect(2, 0, 7, 5)},
{2, image.Rect(3, 0, 8, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
},
{
0.6 + 0.005, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(1, 0, 6, 5)},
{3, image.Rect(2, 0, 7, 5)},
{2, image.Rect(3, 0, 8, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{3, image.Rect(2, 0, 7, 5)},
{1, image.Rect(4, 0, 9, 5)},
},
},
// Same test but vertical.
{
0.6 + 0.005, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(0, 1, 5, 6)},
{3, image.Rect(0, 2, 5, 7)},
{2, image.Rect(0, 3, 5, 8)},
{1, image.Rect(0, 4, 5, 9)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{3, image.Rect(0, 2, 5, 7)},
{1, image.Rect(0, 4, 5, 9)},
},
},
// Same test but diagonal.
{
0.5 * 0.5, 10,
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{4, image.Rect(1, 1, 6, 6)},
{3, image.Rect(2, 2, 7, 7)},
{2, image.Rect(3, 3, 8, 8)},
{1, image.Rect(4, 4, 9, 9)},
},
[]detect.Det{
{5, image.Rect(0, 0, 5, 5)},
{2, image.Rect(3, 3, 8, 8)},
},
},
// One overlaps the other but not vice versa.
{
0.75, 10,
[]detect.Det{
{2, image.Rect(0, 0, 10, 5)},
{1, image.Rect(3, 0, 8, 5)},
},
[]detect.Det{
{2, image.Rect(0, 0, 10, 5)},
{1, image.Rect(3, 0, 8, 5)},
},
},
}
for _, x := range cases {
// Test if existing detection a covers candidate detection b.
overlap := func(a, b image.Rectangle) bool { return detect.Cover(b, a) > x.MaxInter }
out := detect.Suppress(x.In, x.MaxNum, overlap)
if len(out) != len(x.Out) {
t.Error("different length")
t.Log(x)
t.Log(out)
continue
}
for i := range x.Out {
if !x.Out[i].Rect.Eq(out[i].Rect) {
t.Errorf("differ at index %d", i)
t.Log(x)
t.Logf("want: %v", out)
t.Logf("got: %v", x.Out)
break
}
}
}
}