mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-04 14:31:44 +08:00
Vighnesh Birodkar
20 lines
472 B
Python
20 lines
472 B
Python
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()
|