Warping images
Contents
from __future__ import division, print_function
import numpy as np
%matplotlib inline
Warping images¶
Overview¶
http://scikit-image.org/docs/stable/api/skimage.transform.html
http://scikit-image.org/docs/stable/api/skimage.transform.html#skimage.transform.warp
http://scikit-image.org/docs/stable/api/skimage.transform.html#skimage.transform.AffineTransform (and other similar classes)
Image rotation from scratch¶
The following code shows how to rotate an image using the skimage (scikit-image) library.
import matplotlib.pyplot as plt
from skimage import transform, data
camera = data.camera()
rotated = transform.rotate(camera, 30)
f, (ax0, ax1) = plt.subplots(1, 2, figsize=(10, 5))
ax0.imshow(camera, cmap='gray')
ax1.imshow(rotated, cmap='gray');
Exercise: Write an algorithm from scratch that will do the same (i.e., take an input image as an ndarray, and rotate it).
If you feel creative, you can also write code to magnify (zoom) the image.
You may need: http://en.wikipedia.org/wiki/Polar_coordinate_system A (bad) solution is given below--but try it yourself before looking!A problematic approach¶
from skimage import color
def rotate(image, theta):
theta = np.deg2rad(theta)
height, width = image.shape[:2]
out = np.zeros_like(image)
centre_x, centre_y = width / 2., height / 2.
for x in range(width):
for y in range(height):
x_c = x - centre_x
y_c = y - centre_y
# Determine polar coordinate of pixel
radius = np.sqrt(x_c**2 + y_c**2)
angle = np.arctan2(y_c, x_c)
new_angle = angle + theta
new_x = radius * np.cos(new_angle)
new_y = radius * np.sin(new_angle)
new_x = new_x + centre_x
new_y = new_y + centre_y
if (new_x >= width) or (new_x < 0) or\
(new_y >= height) or (new_y < 0):
continue
else:
out[int(new_y), int(new_x)] = image[y, x]
return out
rotated = rotate(camera, 40)
plt.imshow(rotated, cmap='gray', interpolation='nearest');
And while we can attempt to fix the problem…¶
…this is not an optimal approach
# Attempt at fixing the holes using a median filter
# -- it works, sort of, but it's not the best approach.
height, width = rotated.shape[:2]
out = rotated.copy()
for x in range(1, width - 1):
for y in range(1, height - 1):
if out[y, x] == 0:
out[y, x] = np.median([out[y, x-1],
out[y, x+1],
out[y+1, x],
out[y-1, x]])
plt.imshow(out, cmap='gray', interpolation='nearest');
A = np.array([[4, 2], [1, 6]])
print(A)
[[4 2]
[1 6]]
plt.imshow(A, cmap='gray', interpolation='nearest');
For later discussion: interpolation¶
Some warping experiments!¶
Fish-eye¶
from skimage import transform, data, io
import numpy as np
import matplotlib.pyplot as plt
# Load face
face = io.imread('../images/stefan.jpg')
# Get the eye nicely in the middle
face = face[:185, 15:]
plt.imshow(face)
plt.plot([face.shape[1]/2.], [face.shape[0]/2.], 'or', markersize=14, alpha=0.4)
plt.axis('image');
# Define a transformation on the x-y coordinates
def fisheye(xy):
center = np.mean(xy, axis=0)
xc, yc = (xy - center).T
# Polar coordinates
r = np.sqrt(xc**2 + yc**2)
theta = np.arctan2(yc, xc)
r = 0.8 * np.exp(r**(1/2.1) / 1.8)
return np.column_stack((r * np.cos(theta), r * np.sin(theta))) + center
# Warp and display
out = transform.warp(face, fisheye)
f, (ax0, ax1) = plt.subplots(1, 2, figsize=(10, 5))
ax0.imshow(face)
ax0.set_axis_off()
ax1.imshow(out)
ax1.set_axis_off()
plt.title('Knock! Knock!')
plt.show()
Run the following scripts for fun:¶
(Open up the terminal in the “scripts” directory first)
deswirl.py (run using:
python deswirl.py)In the UK, a criminal tried to hide his identity by posting swirled pictures of his face online. Here, we use the Mona Lisa to illustrate what he did. Can you restore her face back to normal? (Note that you can adjust the position of the red dot, as well as move the sliders.)
clock_deblur.py
I took a picture of a wall clock while moving the camera. Or perhaps the clock moved. Either way, now I cannot read the time! I’ve implemented a deblurring algorithm–can you adjust its parameters to help me pin-point the time?
Here’s code for a swirl transform:¶
from skimage import transform
def swirl(xy, center=[0, 0], strength=1, radius=100, rotation=0):
"""Compute the coordinate mapping for a swirl transformation.
"""
x, y = xy.T
x0, y0 = center
rho = np.sqrt((x - x0)**2 + (y - y0)**2)
# Ensure that the transformation decays to approximately 1/1000-th
# within the specified radius.
radius = radius / 5 * np.log(2)
theta = rotation + strength * \
np.exp(-rho / radius) + \
np.arctan2(y - y0, x - x0)
xy[..., 0] = x0 + rho * np.cos(theta)
xy[..., 1] = y0 + rho * np.sin(theta)
return xy
h, w = face.shape[:2]
parameters = {'center': [w/2., h/2.],
'strength': 8,
'radius': 90,
'rotation': 0}
out = transform.warp(face, swirl, parameters)
f, (ax0, ax1) = plt.subplots(1, 2, figsize=(8, 4))
ax0.imshow(face)
ax1.imshow(out);
Can you come up with an even better distortion?¶
Start with this template:¶
def my_warp(xy):
x = xy[:, 0]
y = xy[:, 1]
x = x + 1.5 * np.sin(y / 3)
return np.hstack((x, y))
image = plt.imread('../images/stefan.jpg')
out = transform.warp(image, my_warp)
f, (ax0, ax1) = plt.subplots(1, 2, figsize=(8, 4))
ax0.imshow(image)
ax1.imshow(out);
Composing Transformations¶
scikit-image allows you to compose several transformations. For example:
from skimage import data
cat = data.chelsea()
horizontal_shift = transform.SimilarityTransform(translation=[20, 0])
multiple_shifts = horizontal_shift + horizontal_shift + horizontal_shift
f, (ax0, ax1, ax2) = plt.subplots(1, 3, figsize=(15, 5))
ax0.imshow(cat)
ax1.imshow(transform.warp(cat, horizontal_shift.inverse)) # Note the inverse!
ax2.imshow(transform.warp(cat, multiple_shifts.inverse));
The transform module allows us to rotate images. The inner workings is something like this:
def my_rotate(image, angle):
rotation_tf = transform.SimilarityTransform(rotation=np.deg2rad(angle))
return transform.warp(image, rotation_tf.inverse)
plt.imshow(my_rotate(cat, 30))
<matplotlib.image.AxesImage at 0x7f4b35d86b90>
Note that this rotates the cat around the origin (top-left).
Can you modify my_rotate to rotate the image around the center?
Hint:
Shift the image (see above) so that the center of the image lies at (0, 0)
Rotate the image
Shift the image back—the opposite of what you did in step 1
All of this can be achieved by composing transformations and calling warp once.
Advanced challenge: rectifying an image¶
We know the above tiles are laid out in a square–can you transform the image so that the tiles are displayed as if you were viewing them from above?
The centre-points of the corner circles are, given as (row, column) coordinates:
(72, 129) -- top left
(76, 302) -- top right
(185, 90) -- bottom left
(193, 326) -- bottom right
Hint: there is a linear transformation matrix, \(H\), such that
\(H \mathbf{x} = \mathbf{x}'\)
where \(\mathbf{x}\) is the homogeneous coordinate in the original image and \(\mathbf{x}'\) is the homogeneous coordinate in the rectified image (with homogeneous we simply mean that we add an extra 1 at the end, e.g. (72, 129) becomes (72, 129, 1). The values for \(\mathbf{x}\) and their new values, \(\mathbf{x}'\), are therefore:
x = (72, 129, 1), x' = (0, 0, 1)
x = (76, 302, 1), x' = (0, 400, 1)
x = (185, 90, 1), x' = (400, 0, 1)
x = (193, 326, 1) x' = (400, 400, 1)
(You can choose any output size you like–I chose \(400 \times 400\))
Why do we need homogeneous coordinates? It allows us to have translation as part of H:
\( \left[\begin{array}{ccc} H_{00} & H_{01} & H_{02}\\ H_{10} & H_{11} & H_{12}\\ H_{20} & H_{21} & 1 \end{array}\right]\left[\begin{array}{c} x\\ y\\ 1 \end{array}\right]=\left[\begin{array}{c} H_{00}x+H_{01}y+H_{02}\\ H_{10}x+H_{11}y+H_{12}\\ H_{20}x+H_{21}y+H_{22} \end{array}\right] \)
Note that each element of the output coordinate is of the form \(ax + by + c\). Without the 1 in the last position of the coordinate, there would have been no \(+ c\) and therefore no translation!
The question on how to determine \(H\) is left for another day. If you are curious,
the answer can be found here.
In the meantime, I provide some code to calculate \(H\):
from skimage.transform import estimate_transform
source = np.array([(129, 72),
(302, 76),
(90, 185),
(326, 193)])
target = np.array([[0, 0],
[400, 0],
[0, 400],
[400, 400]])
tf = estimate_transform('projective', source, target)
H = tf.params
print(H)
[[ 3.04026872e+00 1.04929628e+00 -4.67743998e+02]
[-1.44134582e-01 6.23382067e+00 -4.30241727e+02]
[ 2.63620672e-05 4.17694527e-03 1.00000000e+00]]
Using the code in the cell above, you can compute the target coordinate of any position in the original image.
# Verify that the top left corner maps to (0, 0)
x = np.array([[129, 72, 1]])
z = np.dot(H, x.T)
z /= z[2]
print(z)
[[4.35868737e-14]
[1.30760621e-13]
[1.00000000e+00]]
Here’s a template solution:¶
def rectify(xy):
x = xy[:, 0]
y = xy[:, 1]
# We need to provide the backward mapping, from the target
# image to the source image.
HH = np.linalg.inv(H)
# You must fill in your code here to take
# the matrix HH (given above) and to transform
# each coordinate to its new position.
#
# Hint: handy functions are
#
# - np.dot (matrix multiplication)
# - np.ones_like (make an array of ones the same shape as another array)
# - np.column_stack
# - A.T -- type .T after a matrix to transpose it
# - x.reshape -- reshapes the array x
# ... your code
# ... your code
# ... your code
# ... your code
return ...
image = plt.imread('images/chapel_floor.png')
out = transform.warp(image, rectify, output_shape=(400, 400))
f, (ax0, ax1) = plt.subplots(1, 2, figsize=(8, 4))
ax0.imshow(image)
ax1.imshow(out)
---------------------------------------------------------------------------
FileNotFoundError Traceback (most recent call last)
Input In [19], in <module>
9 # You must fill in your code here to take
10 # the matrix HH (given above) and to transform
11 # each coordinate to its new position.
(...)
23 # ... your code
24 # ... your code
26 return ...
---> 29 image = plt.imread('images/chapel_floor.png')
30 out = transform.warp(image, rectify, output_shape=(400, 400))
32 f, (ax0, ax1) = plt.subplots(1, 2, figsize=(8, 4))
File ~/.virtualenvs/skimage-tutorials/lib/python3.10/site-packages/matplotlib/pyplot.py:2139, in imread(fname, format)
2137 @_copy_docstring_and_deprecators(matplotlib.image.imread)
2138 def imread(fname, format=None):
-> 2139 return matplotlib.image.imread(fname, format)
File ~/.virtualenvs/skimage-tutorials/lib/python3.10/site-packages/matplotlib/image.py:1560, in imread(fname, format)
1558 response = io.BytesIO(response.read())
1559 return imread(response, format=ext)
-> 1560 with img_open(fname) as image:
1561 return (_pil_png_to_float_array(image)
1562 if isinstance(image, PIL.PngImagePlugin.PngImageFile) else
1563 pil_to_array(image))
File ~/.virtualenvs/skimage-tutorials/lib/python3.10/site-packages/PIL/ImageFile.py:100, in ImageFile.__init__(self, fp, filename)
96 self.decodermaxblock = MAXBLOCK
98 if isPath(fp):
99 # filename
--> 100 self.fp = open(fp, "rb")
101 self.filename = fp
102 self._exclusive_fp = True
FileNotFoundError: [Errno 2] No such file or directory: 'images/chapel_floor.png'