func TestCorrStrideBLAS_vsNaive(t *testing.T) {
const eps = 1e-9
for _, q := range strideCases {
f := randImage(q.ImSize.X, q.ImSize.Y)
g := randImage(q.TmplSize.X, q.TmplSize.Y)
naive, err := slide.CorrStrideNaive(f, g, q.K)
if err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
blas, err := slide.CorrStrideBLAS(f, g, q.K)
if err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
if err := errIfNotEqImage(naive, blas, eps); err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
}
}
func TestCorrStrideNaive_vsDecimate(t *testing.T) {
const eps = 1e-9
for _, q := range strideCases {
f := randImage(q.ImSize.X, q.ImSize.Y)
g := randImage(q.TmplSize.X, q.TmplSize.Y)
h, err := slide.CorrNaive(f, g)
if err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
want := slide.Decimate(h, q.K)
got, err := slide.CorrStrideNaive(f, g, q.K)
if err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
if err := errIfNotEqImage(want, got, eps); err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.ImSize, q.TmplSize, q.K, err)
continue
}
}
}
func TestCorrStrideNaive(t *testing.T) {
const eps = 1e-9
const X = 10
cases := []struct {
F, G, H *rimg64.Image
K int
}{
{
F: rimg64.FromRows([][]float64{
{1, 2, 3, 4, 5},
{2, 5, 4, 1, 3},
{5, 4, 3, 2, 1},
}),
G: rimg64.FromRows([][]float64{
{1, -1},
}),
K: 2,
H: rimg64.FromRows([][]float64{
{1 - 2, 3 - 4},
{5 - 4, 3 - 2},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 2, 3, 4, 5},
{2, 5, 4, 1, 3},
{5, 4, 3, 2, 1},
}),
G: rimg64.FromRows([][]float64{
{1},
{-1},
}),
K: 2,
H: rimg64.FromRows([][]float64{
{1 - 2, 3 - 4, 5 - 3},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 2, 3, 4, 5},
{2, 5, 4, 1, 3},
{5, 4, 3, 2, 1},
{2, 2, 2, 2, 2},
}),
G: rimg64.FromRows([][]float64{
{1, 2, 3},
{-3, -2, -1},
}),
K: 2,
H: rimg64.FromRows([][]float64{
{(1*1 + 2*2 + 3*3) + ((-3)*2 + (-2)*5 + (-1)*4), (1*3 + 2*4 + 3*5) + ((-3)*4 + (-2)*1 + (-1)*3)},
{(1*5 + 2*4 + 3*3) + ((-3)*2 + (-2)*2 + (-1)*2), (1*3 + 2*2 + 3*1) + ((-3)*2 + (-2)*2 + (-1)*2)},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 2, 3, 4, 5},
{2, 5, 4, 1, 3},
{5, 4, 3, 2, 1},
{2, 2, 2, 2, 2},
}),
G: rimg64.FromRows([][]float64{
{1, 2, 3},
{-3, -2, -1},
}),
K: 3,
H: rimg64.FromRows([][]float64{
{(1*1 + 2*2 + 3*3) + ((-3)*2 + (-2)*5 + (-1)*4)},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 2, 3, 4, 5},
{2, 5, 4, 1, 3},
{5, 4, 3, 2, 1},
{2, 2, 2, 2, 2},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{-3, -2},
}),
K: 3,
H: rimg64.FromRows([][]float64{
{(1*1 + 2*2) + ((-3)*2 + (-2)*5), (1*4 + 2*5) + ((-3)*1 + (-2)*3)},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 1, 0, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 1, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 1},
{0, 0, 0, 0, 0, 0, 0, 1},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 2,
H: rimg64.FromRows([][]float64{
{1, 0, 3, 0},
{0, 0, 0, 0},
{0, 1, 0, 4},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 0, X, 0, 1, X, 0, 0},
{0, 0, X, 0, 0, X, 1, 0},
{X, X, X, X, X, X, X, X},
{0, 0, X, 1, 0, X, 0, 1},
{0, 1, X, 0, 1, X, 0, 1},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 3,
H: rimg64.FromRows([][]float64{
{1, 2, 3},
{4, 5, 6},
}),
},
{
F: rimg64.FromRows([][]float64{
{0, 0, X, X, 0, 1, X, X},
{0, 1, X, X, 0, 0, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{0, 0, X, X, 1, 0, X, X},
{1, 0, X, X, 0, 1, X, X},
{X, X, X, X, X, X, X, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 4,
H: rimg64.FromRows([][]float64{
{4, 2},
{3, 5},
}),
},
{
F: rimg64.FromRows([][]float64{
{0, 1, X, X, X, 0, 0, X},
{0, 0, X, X, X, 0, 1, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{0, 1, X, X, X, 1, 0, X},
{0, 1, X, X, X, 0, 0, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 5,
H: rimg64.FromRows([][]float64{
{2, 4},
{6, 1},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 1, X, X, X, X, 0, 0},
{0, 0, X, X, X, X, 1, 1},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 6,
H: rimg64.FromRows([][]float64{
{3, 7},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 1, X, X, X, X, X, X},
{1, 1, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 7,
H: rimg64.FromRows([][]float64{
{10},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 1, X, X, X, X, X, X},
{0, 1, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
{X, X, X, X, X, X, X, X},
}),
G: rimg64.FromRows([][]float64{
{1, 2},
{3, 4},
}),
K: 10000,
H: rimg64.FromRows([][]float64{
{7},
}),
},
{
F: rimg64.FromRows([][]float64{
{1, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 1, 0, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 0},
{0, 0, 1, 0, 0, 0, 0, 0},
{0, 0, 0, 0, 0, 1, 0, 0},
{0, 0, 0, 0, 0, 0, 0, 0},
}),
G: rimg64.FromRows([][]float64{
{1, 2, 3},
{4, 5, 6},
{7, 8, 9},
}),
K: 2,
H: rimg64.FromRows([][]float64{
{1, 6, 4},
{9, 7, 0},
{3, 1, 5},
}),
},
}
for _, q := range cases {
h, err := slide.CorrStrideNaive(q.F, q.G, q.K)
if err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.F.Size(), q.G.Size(), q.K, err)
continue
}
if err := errIfNotEqImage(q.H, h, eps); err != nil {
t.Errorf("im %v, tmpl %v, stride %d: %v", q.F.Size(), q.G.Size(), q.K, err)
continue
}
}
}
