From 8e14f5f073dc054933f744978699b440213e3b7c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Sat, 8 Sep 2012 15:24:14 +0200 Subject: [PATCH] Add missing 0th layer of laplacian pyramid --- skimage/transform/pyramids.py | 1 + skimage/transform/tests/test_pyramids.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/skimage/transform/pyramids.py b/skimage/transform/pyramids.py index 470218a5..4dd6f7b1 100644 --- a/skimage/transform/pyramids.py +++ b/skimage/transform/pyramids.py @@ -262,6 +262,7 @@ def build_laplacian_pyramid(image, max_layer=-1, downscale=2, sigma=None, cols = image.shape[1] prev_layer_image = image - _smooth(image, sigma, mode, cval) + yield prev_layer_image # build downsampled images until max_layer is reached or downsampled image # has size of 1 in one direction diff --git a/skimage/transform/tests/test_pyramids.py b/skimage/transform/tests/test_pyramids.py index d9e6c599..9aa236ff 100644 --- a/skimage/transform/tests/test_pyramids.py +++ b/skimage/transform/tests/test_pyramids.py @@ -33,7 +33,6 @@ def test_build_laplacian_pyramid(): pyramid = build_laplacian_pyramid(image, downscale=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)