Added implementation and doc string

This commit is contained in:
Vighnesh Birodkar
2015-03-30 13:22:22 +05:30
parent 3e9506c6b3
commit d30ed25968
5 changed files with 308 additions and 1 deletions
+19
View File
@@ -0,0 +1,19 @@
from skimage import io, data
from skimage import transform
from skimage import color, filters
from matplotlib import pyplot as plt
def custom_sobel(img):
if img.ndim == 3:
img = color.rgb2gray(img)
return filters.sobel(img)
img = data.coins()
out = transform.seam_carve(img, 'vertical', 80, energy_func = custom_sobel)
out = transform.seam_carve(out, 'horizontal', 70, energy_func = custom_sobel)
io.imshow(out)
plt.figure()
io.imshow(img)
io.show()