Added rocket image

doc/examples/plot_seam_carving.py

@@ -13,31 +13,81 @@ using the Sobel filter to signify the importance of each pixel.
        http://www.cs.jhu.edu/~misha/ReadingSeminar/Papers/Avidan07.pdf
 
 """
-from skimage import io, data
-from skimage import transform
+from skimage import data, draw
+from skimage import transform, util
+import numpy as np
 from skimage import filters, color
 from matplotlib import pyplot as plt
 
-def nothing(img):
-    return img
 
-img = io.imread('/home/vighnesh/images/rocket.jpg')
-#img = color.rgb2gray(img)
+hl_color = np.array([0, 1, 0])
+
+img = data.rocket()
+img = util.img_as_float(img)
 eimg = filters.sobel(color.rgb2gray(img))
-#img = data.camera()
-out = transform.seam_carve(img, eimg, 'vertical', 200)
-#out = transform.seam_carve(out, 'horizontal', 70, energy_func=filters.sobel)
-resized = transform.resize(img, out.shape)
 
 plt.title('Original Image')
-io.imshow(img, plugin='matplotlib')
+plt.imshow(img)
 
-#plt.figure()
-#plt.title('Resized Image')
-#io.imshow(resized, plugin='matplotlib')
+"""
+.. image:: PLOT2RST.current_figure
+"""
 
+resized = transform.resize(img, (img.shape[0], img.shape[1] - 200))
+plt.figure()
+plt.title('Resized Image')
+plt.imshow(resized)
+
+
+"""
+.. image:: PLOT2RST.current_figure
+"""
+
+out = transform.seam_carve(img, eimg, 'vertical', 200)
 plt.figure()
 plt.title('Resized using Seam-Carving')
-io.imshow(out, plugin='matplotlib')
+plt.imshow(out)
 
-io.show()
+"""
+.. image:: PLOT2RST.current_figure
+
+As you can see, resizing as distorted the rocket and the objects around,
+whereas seam carving has reszied by removing the empty spaces in between.
+
+Object Removal
+--------------
+
+Seam Carving can also be used to remove atrifacts from images. To do that, we
+have to ensure that pixels to be removes get less importance. In the following
+code I approximately mark the rocket with a mask, and then decrease the
+importance of those pixels
+
+"""
+
+masked_img = img.copy()
+
+poly = [(404, 281), (404, 360), (359, 364), (338, 337), (145, 337), (120, 322), (145, 304), (340, 306), (362, 284)]
+pr = np.array([p[0] for p in poly])
+pc = np.array([p[1] for p in poly])
+rr, cc = draw.polygon(pr, pc)
+
+masked_img[rr,cc,:] = masked_img[rr,cc,:]*0.5 + hl_color*.5
+plt.figure()
+plt.title('Object Marked')
+
+plt.imshow(masked_img)
+"""
+.. image:: PLOT2RST.current_figure
+"""
+
+eimg[rr, cc] -= 1000
+
+plt.figure()
+plt.title('Object Removed')
+out = transform.seam_carve(img, eimg, 'vertical', 90)
+resized = transform.resize(img, out.shape)
+plt.imshow(out)
+plt.show()
+"""
+.. image:: PLOT2RST.current_figure
+"""

skimage/data/__init__.py

@@ -26,6 +26,7 @@ __all__ = ['load',
            'chelsea',
            'coffee',
            'hubble_deep_field',
+           'rocket',
            'astronaut']
 
 
@@ -241,3 +242,21 @@ def hubble_deep_field():
 
     """
     return load("hubble_deep_field.jpg")
+
+def rocket():
+    """Launch photo of DSCOVR on Falcon 9 by SpaceX.
+
+    This is the launch photo of Falcon 9 carrying DSCOVR lifted off from
+    SpaceX's Launch Complex 40 at Cape Canaveral Air Force Station, Fla..
+
+    Notes
+    -----
+    This image was downloaded from
+    `SpaceX Photos
+    <https://www.flickr.com/photos/spacexphotos/16511594820/in/photostream/>`__.
+
+    The image was captured by SpaceX and `released in the public domain
+    <http://arstechnica.com/tech-policy/2015/03/elon-musk-puts-spacex-photos-into-the-public-domain/>`_.
+
+    """
+    return load("rocket.jpg")