The present invention relates to image processing systems, and, more particularly to a technique for interactive segmentation of images using graph cuts.
Various types of software products for editing digital images exist in the marketplace. Although many of these products perform basic editing tasks such as cutting, cropping, and touching-up reasonably well, it can be difficult to segment images using these products. Segmentation involves isolating a desired portion of an image and separating that portion from the rest of the image.
Conventionally, some photo-editors provide segmentation tools based on user-based seeds that must be placed on desired segmentation boundaries. However, this is often a tedious task since the seeds must be very carefully placed. For instance, it is usually necessary to place numerous seeds directly on the boundary. Although some conventional photo-editors do provide for more relaxed seed placement, these segmentation tools tend to produce inaccurate results.
Accordingly, it would be desirable and highly advantageous to provide improved segmentation techniques for photo-editing that overcome the problems of conventional approaches.
In various embodiments of the present invention, an image editing system comprises an input device for inputting an image, a graphical user interface for selecting background and object seeds for the image, and an image processor for editing the image. The image processor has various editing routines, including a segmentation routine that builds a graph associated with the image and uses a graph cut algorithm to cut the graph into segments. The user marks certain pixels as “object” or “background” to provide hard constraints for segmentation. Additional soft constraints incorporate both boundary and regional information. Graph cuts are used to find the globally optimal segementation of the image. The obtained solution gives the best balance of boundary and region properties satisfying the constraints.
According to various embodiments of the present invention, the nodes of the graph represent portions of the image, such as pixels (voxels). The edges of the graph represent neighborhood relationships among the nodes. The graph also includes a background terminal node and an object terminal node. Each of the non-terminal nodes of the graph is connected to both the background terminal node and the object terminal node.
The edges of the graph are each assigned a cost value. The cost values are preferably determined according to a cost function that is defined in terms of boundary and region properties of segments. In addition, the cost function uses the seed positions to assign cost values. A graph cut is performed using a suitable graph cut algorithm. Inexpensive edges are attractive choices for a minimum graph cut.
The image editing system can further be configured with an output device (e.g, computer monitor, printer) for outputting the segmented image. The graphical user interface allows additional background and object seeds to be input. By reviewing the results of initial segmentation, the user will see which areas of the image were incorrectly segmented. Then the user can place additional seeds to correct the problem. This interactive process of adding seeds may be continued until the user obtains satisfactory results.
These and other aspects, features and advantages of the present invention will become apparent from the following detailed description of preferred embodiments, which is to be read in connection with the accompanying drawings.
Referring to
In an exemplary embodiment of the present invention, as shown in simplified block diagram form in
Referring to FIGS. 3(a) and 3(b) the process of using the segmentation routine 246 is illustrated.
The present invention provides a general-purpose interactive segmentation technique that divides an image into two segments: “object” and “background”. A user imposes certain hard constraints for segmentation by indicating certain pixels (seeds) that have to be part of the object and certain pixels that have to be part of the background. Intuitively, these hard constraints provide clues on what the user intends to segment. The rest of the image is segmented automatically by computing a global optimum among all segmentations satisfying the hard constraints. A cost function is defined in terms of boundary and region properties of the segments. These properties can be viewed as soft constraints for segmentation. A globally optimal segmentation can be very efficiently recomputed when the user adds or removes any hard constraints (seeds). This allows the user to get any desired segmentation results quickly via very intuitive interactions. This method applies to all N-D images (volumes).
One of the main advantages of the interactive segmentation method employed herein is that it provides a globally optimal solution for the N-dimensional segmentation when the cost function is clearly defined. Some earlier techniques can do that only in 2D applications when a segmentation boundary is a 1D curve. Other techniques either don't have a clear cost function at all (e.g., region growing, split and merger) or compute only an approximate solution (e.g., a local minimum) that can be arbitrarily far from the global optimum (region competition, level set methods, normalized cuts). Global properties of such segmentation may be difficult to analyze or predict. Imperfections in the result might come from deficiencies at the minimization stage. In contrast, imperfections of a globally optimal solution are directly related to the definition of the cost function. Thus, the segmentation can be controlled more reliably.
It is also important that the cost function used as a soft constraint for segmentation is general enough to include both region and boundary properties of segments. Consider an arbitrary set of data elements P and some neighborhood system represented by a set N of all unordered pairs {p,q} of neighboring elements in P. For example, P can contain pixels (or voxels) in a 2D (or 3D) grid and N can contain all unordered pairs of neighboring pixels (voxels) under a standard 8- (or 26-) neighborhood system. Let A=(A1, . . . , Ap, . . . , A|p|) be a binary vector whose components Ap specify assignments to pixels p in P. Each Ap can be either “obj” or “bkg” (abbreviations of “object” and “background”). Vector A defines a segmentation. Then, the soft constraints that we impose on boundary and region properties of A are described by the cost function E(A):
The coefficient λ≧0 in (1) specifies the relative importance of the region properties term R(A) versus the boundary properties term B(A). The regional term R(A) assumes that the individual penalties for assigning pixel p to “object” and “background”, correspondingly Rp(·) may reflect on how the intensity of pixel p fits into a known intensity model (e.g., histogram) of the object and background.
The term B(A) comprises the “boundary” properties of segmentation A. Coefficient B{p,q}≧0 should be interpreted as a penalty for discontinuity between p and q. Normally, B{q,p} is large when pixels p and q are similar (e.g., in their intensity) and B{q,p} is close to zero when the two are very different. The penalty B{q,p} can also decrease as a function of distance between p and q. Costs B{q,p} may be based on local intensity gradient, Laplacian zero-crossing, gradient direction, and other criteria.
Hard constraints that indicate segmentation regions rather than the boundary are considered. It is assumed that some pixels were marked as internal and some as external for the given object of interest. The subsets marked pixels will be referred to as “object” and “background” seeds. The segmentation boundary can be anywhere but it has to separate the object seeds from the background seeds. Note that the seeds can be loosely positioned inside the object and background regions. The segmentation technique described herein is quite stable and normally produces the same results regardless of particular seed positioning within the same image object.
Obviously, the hard constraints by themselves are not enough to obtain a good segmentation. A segmentation method decides how to segment unmarked pixels. Some conventional techniques use the same type of hard constraints as the present invention but they do not employ a clear cost function and segment unmarked pixels based on variations of “region growing”. Since the properties of segmentation boundary are not optimized, the results are prone to “leaking” where the boundary between objects is blurry. In contrast, the present invention combines the hard constraints as above with energy (1) that incorporates region and boundary properties of segments.
The segmentation technique described herein is based on powerful graph cut algorithms from the field of combinational optimization. The implementation uses a new version of the “max-flow” algorithm. Next we provide some terminology for graph cuts and provide some background information.
Graph Cuts
First, we describe the basic terminology that pertains to graph cuts in the context of our segmentation method. An undirected graph G=<V,E> is defined as a set of nodes (vertices V) and a set of undirected edges (E) that connect these nodes. An example of a graph that we use in this paper is shown in
Graph cut formalism is well suited for segmentation of images. In fact, it is completely appropriate for N-dimensional volumes. The nodes of the graph can represent pixels (or voxels) and the edges can represent any neighborhood relationship between the pixels. A cut partitions the node in the graph. As illustrated in FIGS. 4 (c) and (d), this partitioning corresponds to a segmentation of the underlying image or volume. A minimum cost cut generates a segmentation that is optimal in terms of properties that are built into the edge weights.
Segmentation Technique
In this section we provide algorithmic details about the segmentation technique. Assume that O and B denote the subsets of pixels marked as “object” and “background” seeds. Naturally, the subsets O⊂P and B⊂P are such that O∩B=φ. Remember that our goal is to compute global minimum of (1) among all segmentations A satisfying hard constraints
∀pεO,Ap=“obj” (4)
∀pεB,Ap=“bkg” (5)
The general work flow is described in conjunction with
The next step is to compute the globally optimal minimum cut. This cut gives a segmentation 430 of the original image 410. In the simplistic examples of
Below we describe the details of the graph and prove that the obtained segmentation is optimal. To segment a given image we create a graph G—<V,E> with nodes corresponding to pixels pεP of the image. There are two additional nodes: an “object” terminal (a source S) and a “background” terminal (a sink T). Therefore,
V=P∪{S,T}.
The set of edges E consists of two types of undirected edges: n-links (neighborhood links) and t-links (terminal links). Each pixel p has two t-links {p,S} and {p,T} connecting it to each terminal. Each pair of neighboring pixels {p,q} in N is connected by an n-link. Without introducing any ambiguity, an n-link connecting a pair of neighbors p and q will be denoted {p,q}. Therefore,
The following table gives weights of edges in E
The graph G is now completely defined. We draw the segmentation boundary between the object and the background by finding the minimum cost cut on the graph G. The minimum cost cut Ĉ on G can be computed exactly in polynomial time via algorithms for two terminal graph cuts assuming that the edge weights specified in the table above are non-negative.
Below we state exactly how the minimum cut Ĉ defines a segmentation  and prove this segmentation is optimal. We need one technical lemma. Assume that F denotes a set of all feasible cuts C on graph G such that
For any feasible cut CεF we can define a unique corresponding segmentation A(C) such that
The definition above is coherent since any feasible cut severs exactly one of the two t-links at each pixel p. The lemma showed that a minimum cut Ĉ is feasible. Thus, we can define a corresponding segmentation Â=A(Ĉ). The next theorem completes the description of our algorithm.
Theorem 1 The segmentation ÂA=A(Ĉ) defined by the minimum cut Ĉ as in (6) minimizes (1) among all segmentations satisfying constraints (4, 5).
Proof: Using the table of edge weights, definition of feasible cuts F, and equation (6) one can show that a cost of any CεF is
Therefore, |C|=E(A(C))−const(C). Note that for any CεF assignment A(C) satisfies constraints (4, 6). In fact, equation (6) gives one-to-one correspondence between the set of all feasible cuts in F and the set H of all assignments A that satisfy hard constraints (4, 5). Then,
and the theorem is proved.
To conclude this section we would like to show that the algorithm can efficiently adjust the segmentation to incorporate any additional seeds that the user might interactively add. To be specific, assume that a max-flow algorithm is used to determine the minimum cut G. The max-flow algorithm gradually increases the flow sent from the source S to the sink T along the edges in G given their capacities (weights). Upon termination the maximum flow saturates the graph. The saturated edges correspond to the minimum cost cut on G giving us an optimal segmentation.
Assume now that an optimal segmentation is already computed for some initial set of seeds. A user adds a new “object” seed to pixel p that was not previously assigned any seed. We need to change the costs for two t-links at p
and then compute the maximum flow (minimum cut) on the new graph. In fact, we can start from the flow found at the end of initial computation. The only problem is that reassignment of edge weights as above reduces capacities of some edges. If there is a flow through such an edge then we may break the flow consistency. Increasing an edge capacity, on the other hand, is never a problem. Then, we can solve the problem as follows.
To accommodate the new “object” seed at pixel p we increase the t-links weights according to the table
These new costs are consistent with the edge weight table for pixels in O since the extra constant cp at both t-links of a pixel does not change the optimal cut. Then, a maximum flow (minimum cut) on a new graph can be efficiently obtained starting from the previous flow without re-computing the whole solution from scratch.
Note that the same trick can be done to adjust the segmentation when a new “background” seed is added or when a seed is deleted. One has to figure the right amounts that have to be added to the costs of two t-links at the corresponding pixel. The new costs should be consistent with the edge weight table plus or minus the same constant.
We demonstrate the general-purpose segmentation method in several examples including photo/video editing. We show original data and segments generated by our technique for a given set of seeds. Our actual interface allows a user to enter seeds via mouse operated brush of red (for object) or blue (for background) color. Due to limitations of the B&W publication we show seeds as strokes of white (object) or black (background) brush. In addition, these strokes are marked by the letters “O” and “B”. For the purpose of clarity, we employ different methods for the presentation of segmentation results in our examples below.
Our current implementation actually makes a double use of the seeds entered by a user. First of all, they provide the hard constraints for the segmentation process as discussed above. In addition, we use intensities of pixels (voxels) marked as seeds to get histograms for “object” and “background” intensity distributions: Pr(I|O) and Pr(I|B). Then, we use these histograms to set the regional penalties Rp(·) as negative log-liklihoods:
Rp(“obj”)=−ln Pr(Ip|O)
Rp(“bkg”)=−ln Pr(Ip|B).
To set the boundary penalties we use an ad-hoc function
This function penalizes a lot for discontinuities between pixels of similar intensities when |Ip−Iq|<σ. However, if pixels are very different, |Ip−Iq>σ, then the penalty is small. Intuitively, this function corresponds to the distribution of noise among neighboring pixels of an image. Thus, σ can be estimated as “camera noise”.
Note that we use an 8-neighborhood system in 2D examples and 26-neighborhood system in 3D examples. All running times are given for 333 MHz Pentium III. Our implementation uses a new “max-flow” algorithm from [2].
Photo and Video Editing
In FIGS. 3(a)-(b) we illustrated the segmentation a bell with a group of people from a photograph. The user can start with a few “object” and “background” seeds loosely positioned inside and, correspondingly, outside the object(s) of interest. By reviewing the results of initial segmentation the user will see what areas are segmented incorrectly. Then (s)he can put additional seeds into the troubled places and efficiently recomputed the optimal segmentation. This process of adding seeds gives more clues to the algorithm and may be continued until the user likes the results.
Naturally, the hope is that the method can quickly identify the right object. The user would not like to keep adding new seeds until the whole image is covered in seeds. This is no better than manual segmentation. The performance of the algorithm can be judged by the efforts required from the user. Thus, the results of our segmentation are shown with seeds that we entered to get this segmentation.
Our segmentation algorithm runs in less than a second for most 2D images (up to 512×512) with a fixed set of seeds. When additional seeds are entered the solution is recomputed in the blink of an eye. Thus, the speed evaluation of our method in 2D is mainly concerned with the user efforts. The detailed segmentation in
In many cases the regional term of energy (1) helps to get the right results faster. In FIGS. 5(a) and 5(b) we show some details of segmentation with and without the regional term (λ=0) given the same sets of seeds. In
The globally minimum two-terminal cut can be computed on any graph. Thus, our technique is valid for segmentation of N-D data. In
The computation of the minimum cut is slower in 3D cases. The initial segmentation might take from 2-3 seconds on smaller volumes (200×200×10) to a few minutes on bigger ones (512×512×50). Thus, efficient re-computing of an optimal solution when new seeds are added is crucial. Most of the time the new seeds are incorporated in a few seconds even for bigger volumes. Therefore, we can still consider our method as “interactive”. The results in
The process of placing seeds can be automated for certain applications. The seeds should be positioned with a low probability of “false alarm” while the probability of “right detect” is not required to be high. Even very simple recognition techniques based on filters might be good enough if the corresponding threshold is set high. Such filters would have difficulties near the boundaries of objects but they can confidently place many seeds anywhere else. These seeds give hard constraints. Based on additional soft constraints (1) the minimum cut can complete the segmentation where the recognition method failed.
Although illustrative embodiments of the present invention have been described herein with reference to the accompanying drawings, it is to be understood that the invention is not limited to those precise embodiments, and that various other changes and modifications may be affected therein by one skilled in the art without departing from the scope or spirit of the invention.
This application is a continuation of U.S. patent application Ser. No. 10/413,974, filed on Apr. 15, 2003, which claims the benefit of U.S. Provisional Application Ser. No. 60/393,163, filed on Jul. 2, 2002, which is incorporated by reference herein in its entirety.
Number | Date | Country | |
---|---|---|---|
60393163 | Jul 2002 | US |
Number | Date | Country | |
---|---|---|---|
Parent | 10413974 | Apr 2003 | US |
Child | 11772915 | Jul 2007 | US |