Active contour model

From HandWiki

Active contour model, also called snakes, is a framework in computer vision introduced by Michael Kass, Andrew Witkin, and Demetri Terzopoulos[1] for delineating an object outline from a possibly noisy 2D image. The snakes model is popular in computer vision, and snakes are widely used in applications like object tracking, shape recognition, segmentation, edge detection and stereo matching. A snake is an energy minimizing, deformable spline influenced by constraint and image forces that pull it towards object contours and internal forces that resist deformation. Snakes may be understood as a special case of the general technique of matching a deformable model to an image by means of energy minimization.[1] In two dimensions, the active shape model represents a discrete version of this approach, taking advantage of the point distribution model to restrict the shape range to an explicit domain learnt from a training set.

Snakes – active deformable models

Snakes do not solve the entire problem of finding contours in images, since the method requires knowledge of the desired contour shape beforehand. Rather, they depend on other mechanisms such as interaction with a user, interaction with some higher level image understanding process, or information from image data adjacent in time or space.

Motivation

In computer vision, contour models describe the boundaries of shapes in an image. Snakes in particular are designed to solve problems where the approximate shape of the boundary is known. By being a deformable model, snakes can adapt to differences and noise in stereo matching and motion tracking. Additionally, the method can find Illusory contours in the image by ignoring missing boundary information.

Compared to classical feature extraction techniques, snakes have multiple advantages:

  • They autonomously and adaptively search for the minimum state.
  • External image forces act upon the snake in an intuitive manner.
  • Incorporating Gaussian smoothing in the image energy function introduces scale sensitivity.
  • They can be used to track dynamic objects.

The key drawbacks of the traditional snakes are

  • They are sensitive to local minima states, which can be counteracted by simulated annealing techniques.
  • Minute features are often ignored during energy minimization over the entire contour.
  • Their accuracy depends on the convergence policy.[2]

Energy formulation

A simple elastic snake is defined by a set of n points [math]\displaystyle{ \mathbf v_i }[/math] for [math]\displaystyle{ i=0, \ldots, n-1 }[/math], the internal elastic energy term [math]\displaystyle{ E_\text{internal} }[/math], and the external edge-based energy term [math]\displaystyle{ E_\text{external} }[/math]. The purpose of the internal energy term is to control the deformations made to the snake, and the purpose of the external energy term is to control the fitting of the contour onto the image. The external energy is usually a combination of the forces due to the image itself [math]\displaystyle{ E_\text{image} }[/math] and the constraint forces introduced by the user [math]\displaystyle{ E_\text{con} }[/math]

The energy function of the snake is the sum of its external energy and internal energy, or

[math]\displaystyle{ E_\text{snake}^*=\int\limits_0^1E_\text{snake}(\mathbf{v}(s))\,ds=\int\limits_0^1 ( E_\text{internal}(\mathbf{v}(s))+E_\text{image}(\mathbf{v}(s))+E_\text{con}(\mathbf{v}(s)) )\,ds }[/math]

Internal energy

The internal energy of the snake is composed of the continuity of the contour [math]\displaystyle{ E_\text{cont} }[/math] and the smoothness of the contour [math]\displaystyle{ E_\text{curv} }[/math].

[math]\displaystyle{ E_\text{internal}=E_\text{cont}+E_\text{curv} }[/math] [3]

This can be expanded as

[math]\displaystyle{ E_\text{internal}=\frac{1}{2}(\alpha\,\!(s)\left | \mathbf{v}_s(s) \right \vert^2) + \frac{1}{2} (\beta\,\!(s)\left | \mathbf{v}_{ss}(s) \right \vert ^2) =\frac{1}{2} \bigg(\alpha\,\!(s) \left \| \frac{d\bar v}{ds}(s) \right \Vert^2 + \beta\,\!(s) \left \| \frac{d^2\bar v}{ds^2}(s) \right \Vert ^2 \bigg) }[/math]

where [math]\displaystyle{ \alpha (s) }[/math] and [math]\displaystyle{ \beta (s) }[/math] are user-defined weights; these control the internal energy function's sensitivity to the amount of stretch in the snake and the amount of curvature in the snake, respectively, and thereby control the number of constraints on the shape of the snake.

In practice, a large weight [math]\displaystyle{ \alpha (s) }[/math] for the continuity term penalizes changes in distances between points in the contour. A large weight [math]\displaystyle{ \beta (s) }[/math] for the smoothness term penalizes oscillations in the contour and will cause the contour to act as a thin plate.

Image energy

Energy in the image is some function of the features of the image. This is one of the most common points of modification in derivative methods. Features in images and images themselves can be processed in many and various ways.

For an image [math]\displaystyle{ I(x,y) }[/math], lines, edges, and terminations present in the image, the general formulation of energy due to the image is

[math]\displaystyle{ E_\text{image}=w_\text{line}E_\text{line}+w_\text{edge}E_\text{edge}+w_\text{term} E_\text{term}, }[/math]

where [math]\displaystyle{ w_\text{line} }[/math], [math]\displaystyle{ w_\text{edge} }[/math], [math]\displaystyle{ w_\text{term} }[/math] are weights of these salient features. Higher weights indicate that the salient feature will have a larger contribution to the image force.

Line functional

The line functional is the intensity of the image, which can be represented as

[math]\displaystyle{ E_\text{line}= I(x,y) }[/math]

The sign of [math]\displaystyle{ w_\text{line} }[/math] will determine whether the line will be attracted to either dark lines or light lines.

Some smoothing or noise reduction may be used on the image, which then the line functional appears as

[math]\displaystyle{ E_\text{line}= \operatorname{filter}(I(x,y)) }[/math]

Edge functional

The edge functional is based on the image gradient. One implementation of this is

[math]\displaystyle{ E_\text{edge}=-\left | \nabla I(x,y)\right \vert ^2. }[/math]

A snake originating far from the desired object contour may erroneously converge to some local minimum. Scale space continuation can be used in order to avoid these local minima. This is achieved by using a blur filter on the image and reducing the amount of blur as the calculation progresses to refine the fit of the snake. The energy functional using scale space continuation is

[math]\displaystyle{ E_\text{edge}=-\left | G_\sigma \cdot \nabla^2 I \right \vert ^2 }[/math]

where [math]\displaystyle{ G_\sigma }[/math] is a Gaussian with standard deviation [math]\displaystyle{ \sigma }[/math]. Minima of this function fall on the zero-crossings of [math]\displaystyle{ G_\sigma \, \nabla^2 I }[/math] which define edges as per Marr–Hildreth theory.

Termination functional

Curvature of level lines in a slightly smoothed image can be used to detect corners and terminations in an image. Using this method, let [math]\displaystyle{ C(x,y) }[/math] be the image smoothed by

[math]\displaystyle{ C(x,y)= G_\sigma \cdot I(x,y) }[/math]

with gradient angle

[math]\displaystyle{ \theta = \arctan \left( \frac {C_y}{C_x} \right), }[/math]

unit vectors along the gradient direction

[math]\displaystyle{ \mathbf n = (\cos \theta,\sin \theta), }[/math]

and unit vectors perpendicular to the gradient direction

[math]\displaystyle{ \mathbf n_\perp = (-\sin \theta,\cos \theta). }[/math]

The termination functional of energy can be represented as

[math]\displaystyle{ E_\text{term}={\partial \theta\over\partial n_\perp} = {\partial^2 C / \partial n_\perp^2 \over \partial C / \partial n} = {{C_{yy} C_x^2-2C_{xy} C_xC_y+C_{xx} C_y^2}\over(C_x^2+C_y^2)^{3/2}} }[/math]

Constraint energy

Some systems, including the original snakes implementation, allowed for user interaction to guide the snakes, not only in initial placement but also in their energy terms. Such constraint energy [math]\displaystyle{ E_{con} }[/math] can be used to interactively guide the snakes towards or away from particular features.

Optimization through gradient descent

Given an initial guess for a snake, the energy function of the snake is iteratively minimized. Gradient descent minimization is one of the simplest optimizations which can be used to minimize snake energy.[4] Each iteration takes one step in the negative gradient of the point with controlled step size [math]\displaystyle{ \gamma }[/math] to find local minima. This gradient-descent minimization can be implemented as

[math]\displaystyle{ \bar v_i \leftarrow \bar v_i + F_\text{snake}(\bar v_i) }[/math]

Where [math]\displaystyle{ F_\text{snake}(\bar v_i) }[/math] is the force on the snake, which is defined by the negative of the gradient of the energy field.

[math]\displaystyle{ F_\text{snake}(\bar v_i) = - \nabla E_\text{snake} (\bar v_i) = - \Bigg( w_\text{internal} \, \nabla E_\text{internal} (\bar v_i)+w_\text{external} \, \nabla E_\text{external} (\bar v_i) \Bigg) }[/math]

Assuming the weights [math]\displaystyle{ \alpha(s) }[/math] and [math]\displaystyle{ \beta(s) }[/math] are constant with respect to [math]\displaystyle{ s }[/math], this iterative method can be simplified to

[math]\displaystyle{ \bar v_i \leftarrow \bar v_i - \gamma \Bigg\{ w_\text{internal} \bigg[ \alpha \frac{\partial ^2 \bar v}{\partial s^2} (\bar v_i)+\beta \frac{\partial ^4 \bar v}{\partial s^4} (\bar v_i) \bigg] + \nabla E_\text{ext} (\bar v_i) \Bigg\} }[/math]

Discrete approximation

In practice, images have finite resolution and can only be integrated over finite time steps [math]\displaystyle{ \tau }[/math]. As such, discrete approximations must be made for practical implementation of snakes.

The energy function of the snake can be approximated by using the discrete points on the snake.

[math]\displaystyle{ E_\text{snake}^*\approx \sum_1^n E_\text{snake}(\bar v_i) }[/math]

Consequentially, the forces of the snake can be approximated as

[math]\displaystyle{ F_\text{snake}^*\approx - \sum_{i=1}^n \nabla E_\text{snake} (\bar v_i). }[/math]

Gradient approximation can be done through any finite approximation method with respect to s, such as Finite difference.

Numerical instability due to discrete time

The introduction of discrete time into the algorithm can introduce updates which the snake is moved past the minima it is attracted to; this further can cause oscillations around the minima or lead to a different minima being found.

This can be avoided through tuning the time step such that the step size is never greater than a pixel due to the image forces. However, in regions of low energy, the internal energies will dominate the update.

Alternatively, the image forces can be normalized for each step such that the image forces only update the snake by one pixel. This can be formulated as

[math]\displaystyle{ F_\text{image} = - k \frac{\nabla E_\text{image}}{\|\nabla E_\text{image}\|} }[/math]

where [math]\displaystyle{ \tau k }[/math] is near the value of the pixel size. This avoids the problem of dominating internal energies that arise from tuning the time step.[5]

Numerical instability due to discrete space

The energies in a continuous image may have zero-crossing that do not exist as a pixel in the image. In this case, a point in the snake would oscillate between the two pixels that neighbor this zero-crossing. This oscillation can be avoided by using interpolation between pixels instead of nearest neighbor.[5]

Some variants of snakes

The default method of snakes has various limitation and corner cases where the convergence performs poorly. Several alternatives exist which addresses issues of the default method, though with their own trade-offs. A few are listed here.

GVF snake model

The gradient vector flow (GVF) snake model[6] addresses two issues with snakes:

  • poor convergence performance for concave boundaries
  • poor convergence performance when snake is initialized far from minimum

In 2D, the GVF vector field [math]\displaystyle{ F_\text{GVF} }[/math] minimizes the energy functional

[math]\displaystyle{ E_\text{GVF} = \iint \mu(u_x^2+u_y^2+v_x^2+v_y^2)+|\nabla f|^2|\mathbf v-\nabla f|^2 \, dx \, dy }[/math]

where [math]\displaystyle{ \mu }[/math] is a controllable smoothing term. This can be solved by solving the Euler equations

[math]\displaystyle{ \mu \, \nabla^2 u - \Bigg (u - \frac{\partial}{\partial x}F_\text{ext}\Bigg )\Bigg (\frac{\partial}{\partial x} F_\text{ext}(x,y)^2 + \frac{\partial}{\partial y}F_\text{ext}(x,y)^2\Bigg ) = 0 }[/math]
[math]\displaystyle{ \mu \, \nabla^2 v - \Bigg (v - \frac{\partial}{\partial y}F_\text{ext}\Bigg )\Bigg (\frac{\partial}{\partial x} F_\text{ext}(x,y)^2 + \frac{\partial}{\partial y} F_\text{ext}(x,y)^2\Bigg ) = 0 }[/math]

This can be solved through iteration towards a steady-state value.

[math]\displaystyle{ u_{i+1} = u_i + \mu \, \nabla^2 u_i - \Bigg (u_i - \frac{\partial}{\partial x} F_\text{ext}\Bigg )\Bigg (\frac{\partial}{\partial x} F_\text{ext}(x,y)^2 + \frac{\partial}{\partial y} F_\text{ext}(x,y)^2\Bigg ) }[/math]
[math]\displaystyle{ v_{i+1} = v_i + \mu \, \nabla^2 v_i - \Bigg (v_i - \frac{\partial}{\partial y}F_\text{ext}\Bigg )\Bigg (\frac{\partial}{\partial x}F_\text{ext}(x,y)^2 + \frac{\partial}{\partial y} F_\text{ext}(x,y)^2\Bigg ) }[/math]

This result replaces the default external force.

[math]\displaystyle{ F_\text{ext}^* = F_\text{GVF} }[/math]

The primary issue with using GVF is the smoothing term [math]\displaystyle{ \mu }[/math] causes rounding of the edges of the contour. Reducing the value of [math]\displaystyle{ \mu }[/math] reduces the rounding but weakens the amount of smoothing.

The balloon model

The balloon model[5] addresses these problems with the default active contour model:

  • The snake is not attracted to distant edges.
  • The snake will shrink inwards if no substantial images forces are acting upon it.
  • a snake larger than the minima contour will eventually shrink into it, but a snake smaller than the minima contour will not find the minima and instead continue to shrink.

The balloon model introduces an inflation term into the forces acting on the snake

[math]\displaystyle{ F_\text{inflation} = k_1 \vec n (s) }[/math]

where [math]\displaystyle{ \vec n (s) }[/math] is the normal unitary vector of the curve at [math]\displaystyle{ v(s) }[/math] and [math]\displaystyle{ k_1 }[/math] is the magnitude of the force. [math]\displaystyle{ k_1 }[/math] should have the same magnitude as the image normalization factor [math]\displaystyle{ k }[/math] and be smaller in value than [math]\displaystyle{ k }[/math] to allow forces at image edges to overcome the inflation force.

Three issues arise from using the balloon model:

  • Instead of shrinking, the snake expands into the minima and will not find minima contours smaller than it.
  • The outward force causes the contour to be slightly larger than the actual minima. This can be solved by decreasing the balloon force after a stable solution has been found.
  • The inflation force can overpower forces from weak edges, amplifying the issue with snakes ignoring weaker features in an image.

Diffusion snakes model

The diffusion snake model[7] addresses the sensitivity of snakes to noise, clutter, and occlusion. It implements a modification of the Mumford–Shah functional and its cartoon limit and incorporates statistical shape knowledge. The default image energy functional [math]\displaystyle{ E_\text{image} }[/math] is replaced with

[math]\displaystyle{ E_\text{image}^* = E_i + \alpha E_c }[/math]

where [math]\displaystyle{ E_i }[/math] is based on a modified Mumford–Shah functional

[math]\displaystyle{ E[J,B] = \frac{1}{2} \int_D (I(\vec x) - J(\vec x))^2 \, d \vec x + \lambda \frac{1}{2} \int_{D/B} \vec \nabla J(\vec x) \cdot \vec \nabla J(\vec x) \, d \vec x + \nu \int_0^1 \Bigg ( \frac{d}{ds} B(s) \Bigg )^2 \, ds }[/math]

where [math]\displaystyle{ J(\vec x) }[/math] is the piecewise smooth model of the image [math]\displaystyle{ I(\vec x) }[/math] of domain [math]\displaystyle{ D }[/math]. Boundaries [math]\displaystyle{ B(s) }[/math] are defined as

[math]\displaystyle{ B(s) = \sum_{n=1}^N \vec p_n B_n (s) }[/math]

where [math]\displaystyle{ B_n (s) }[/math] are quadratic B-spline basis functions and [math]\displaystyle{ \vec p_n }[/math] are the control points of the splines. The modified cartoon limit is obtained as [math]\displaystyle{ \lambda \to \infty }[/math] and is a valid configuration of [math]\displaystyle{ E_i }[/math].

The functional [math]\displaystyle{ E_c }[/math] is based on training from binary images of various contours and is controlled in strength by the parameter [math]\displaystyle{ \alpha }[/math]. For a Gaussian distribution of control point vectors [math]\displaystyle{ \vec z }[/math] with mean control point vector [math]\displaystyle{ \vec z_0 }[/math] and covariance matrix [math]\displaystyle{ \Sigma }[/math] , the quadratic energy that corresponds to the Gaussian probability is

[math]\displaystyle{ E_c(\vec z) = \frac{1}{2} (\vec z - \vec z_0)^t \Sigma^*(\vec z - \vec z_0) }[/math]

The strength of this method relies on the strength of the training data as well as the tuning of the modified Mumford–Shah functional. Different snakes will require different training data sets and tunings.

Geometric active contours

Geometric active contour, or geodesic active contour (GAC)[8] or conformal active contours[9] employs ideas from Euclidean curve shortening evolution. Contours split and merge depending on the detection of objects in the image. These models are largely inspired by level sets, and have been extensively employed in medical image computing.

For example, the gradient descent curve evolution equation of GAC is [8]

[math]\displaystyle{ \frac{\partial C}{\partial t} = g(I)(c+\kappa)\vec{N}-\langle \, \nabla g,\vec{N}\rangle\vec{N} }[/math]

where [math]\displaystyle{ g(I) }[/math] is a halting function, c is a Lagrange multiplier, [math]\displaystyle{ \kappa }[/math] is the curvature, and [math]\displaystyle{ \vec{N} }[/math] is the unit inward normal. This particular form of curve evolution equation is only dependent on the velocity in the normal direction. It therefore can be rewritten equivalently in an Eulerian form by inserting the level set function [math]\displaystyle{ \Phi }[/math] into it as follows

[math]\displaystyle{ \frac{\partial \Phi}{\partial t} = |\nabla \Phi| \operatorname{div} \Bigg ( g(I)\frac{ \nabla \Phi}{|\nabla \Phi|} \Bigg ) + c g(I) |\nabla \Phi| }[/math]

This simple yet powerful level-set reformation enables active contours to handle topology changes during the gradient descent curve evolution. It has inspired tremendous progress in the related fields, and using numerical methods to solve the level-set reformulation is now commonly known as the level-set method. Although the level set method has become quite a popular tool for implementing active contours, Wang and Chan argued that not all curve evolution equations should be directly solved by it.[10]

More recent developments in active contours address modeling of regional properties, incorporation of flexible shape priors and fully automatic segmentation, etc.

Statistical models combining local and global features have been formulated by Lankton and Allen Tannenbaum.[11]

Relations to graph cuts

Graph cuts, or max-flow/min-cut, is a generic method for minimizing a particular form of energy called Markov random field (MRF) energy. The Graph cuts method has been applied to image segmentation as well, and it sometimes outperforms the level set method when the model is MRF or can be approximated by MRF.

See also

References

  1. 1.0 1.1 Kass, M.; Witkin, A.; Terzopoulos, D. (1988). "Snakes: Active contour models". International Journal of Computer Vision 1 (4): 321. doi:10.1007/BF00133570. http://ww.vavlab.ee.boun.edu.tr/courses/574/material/Variational%20Image%20Segmentation/kaas_snakes.pdf. Retrieved 2015-08-29. 
  2. Snakes: an active model, Ramani Pichumani, http://homepages.inf.ed.ac.uk/rbf/CVonline/LOCAL_COPIES/RAMANI1/node31.html
  3. Dr. George Bebis, University of Nevada, http://www.cse.unr.edu/~bebis/CS791E/Notes/DeformableContours.pdf
  4. Image Understanding, Bryan S. Morse, Brigham Young University, 1998–2000 http://homepages.inf.ed.ac.uk/rbf/CVonline/LOCAL_COPIES/MORSE/iu.pdf
  5. 5.0 5.1 5.2 Cohen, Laurent D. (1991). "On active contour models and balloons". CVGIP: Image Understanding 53 (2): 211–218. doi:10.1016/1049-9660(91)90028-N. 
  6. Chenyang Xu; Prince, J.L. (1997). "Gradient vector flow: A new external force for snakes". Proceedings of IEEE Computer Society Conference on Computer Vision and Pattern Recognition. pp. 66–71. doi:10.1109/CVPR.1997.609299. ISBN 0-8186-7822-4. http://iacl.ece.jhu.edu/pubs/p087c.pdf. 
  7. Cremers, D.; Schnorr, C.; Weickert, J. (2001). "Diffusion-snakes: Combining statistical shape knowledge and image information in a variational framework". Proceedings IEEE Workshop on Variational and Level Set Methods in Computer Vision. 50. pp. 137–144. doi:10.1109/VLSM.2001.938892. ISBN 978-0-7695-1278-5. 
  8. 8.0 8.1 Geodesic Active Contours, V. Caselles, R. Kimmel, G. Sapiro http://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.21.2196
  9. Kichenassamy, Satyanad; Kumar, Arun; Olver, Peter; Tannenbaum, Allen; Yezzi, Anthony (1996). "Conformal curvature flows: From phase transitions to active vision". Archive for Rational Mechanics and Analysis 134 (3): 275–301. doi:10.1007/BF00379537. Bibcode1996ArRMA.134..275K. 
  10. Wang, Junyan; Chan, Kap Luk (2014-07-08). "Active Contour with a Tangential Component". Journal of Mathematical Imaging and Vision 51 (2): 229–247. doi:10.1007/s10851-014-0519-y. ISSN 0924-9907. 
  11. Lankton, S.; Tannenbaum, A. (2008). "Localizing Region-Based Active Contours". IEEE Transactions on Image Processing 17 (11): 2029–2039. doi:10.1109/TIP.2008.2004611. PMID 18854247. Bibcode2008ITIP...17.2029L. 

External links

Sample code