Use consistent dtype for all levels of pyramid

2026-08-04 13:14:23 +08:00 · 2012-09-07 18:47:50 +02:00
parent 2780ae63e2
commit bccbc36b91
2 changed files with 9 additions and 2 deletions
@@ -33,6 +33,7 @@ def test_build_laplacian_pyramid():
    pyramid = build_laplacian_pyramid(image, factor=2)

    for layer, out in enumerate(pyramid):
+        layer += 1
        layer_shape = (rows / 2 ** layer, cols / 2 ** layer, dim)
        assert_array_equal(out.shape, layer_shape)