Merge pull request #1459 from vighneshbirodkar/seam_carving

FEAT: Seam Carving
This commit is contained in:
Josh Warner
2015-06-18 12:32:19 -05:00
9 changed files with 439 additions and 1 deletions
@@ -0,0 +1,23 @@
from skimage import transform
import numpy as np
from numpy import testing
def test_seam_carving():
img = np.array([[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 1, 0, 0],
[0, 1, 0, 0, 0],
[1, 0, 0, 0, 0]], dtype=np.float)
energy = 1 - img
out = transform.seam_carve(img, energy, 'vertical', 1, border=0)
testing.assert_allclose(out, 0)
img = img.T
out = transform.seam_carve(img, energy, 'horizontal', 1, border=0)
testing.assert_allclose(out, 0)
if __name__ == '__main__':
np.testing.run_module_suite()