Karthik Yearning Deep Learning

Digital Matting

Paper: A Late Fusion CNN for Digital Matting

This paper studies the structure of a deep convolutional neural network to predict the foreground alpha matte by taking a single RGB image as input. Our network is fully convolutional with two decoder branches for the foreground and background classification respectively. Then a fusion branch is used to integrate the two classification results which gives rise to alpha values as the soft segmentation result

Digital matting is to accurately extract the foreground object in an image for object-level image composition.

We assume that the observed image I is generated from three underlying images: the foreground image F, the background image B, and the alpha matte α, through the following model:

A common approach to digital matting proceeds in three steps, namely,

We propose a fully convolutional network (FCN) for automatic image matting by taking a single RGB image as input. We achieve this goal by designing two decoder branches in the network for the foreground and background classification, and then use a fusion branch to integrate the two classification results, which gives rise to the soft segmentation result

It is based on the observation that the classification branches can well predict the hard segmentation result, but have difficulties in predicting precise probabilities as the alpha values at the pixel level. The two-decoder branch structure allows us to design a fusion branch to correct the residuals left in the classification branches. Moreover, our training loss encourages the two decoder branches to agree with each other at the hard segmentation part and leave the soft segmentation part to be corrected by the fusion branch.

The network is trained over three consecutive steps: segmentation network pretraining step, fusion network pretraining step and finally end-to-end joint training step whose training loss is imposed on the output alpha matte.

we try to predict the alpha values with the following fusion formula:

where F_p and B_p represent the predicted foreground and background probability at pixel p. βp is the blending weight predicted by the fusion network.

From the optimization perspective, the derivative of αp with respect to βp vanishes when

First, the fusion network will focus on learning the transition region from the foreground to the background if the predictions of the foreground/background probability maps are accurate which is the bottleneck for solving the matting problem.

Second, we can carefully design the loss function to encourage that the F_p + B_p ≠ 1 within the transition region, which can provide useful gradients to train the fusion network

Segmentation Network

The segmentation network consists of one encoder and two decoders. The encoder extracts semantic features from the input image. The two decoders share the same encoded bottleneck and predict the foreground and background probability maps, respectively. Specifically, we use DenseNet-201 without the fully-connected layer head as our encoder. Each branch consists of five decoder blocks which correspond to the five encoder blocks, and the decoder block follows the design of feature pyramid network structure

Training Loss

The training loss combines the L1 loss, theL2 loss, and the cross-entropy loss. In particular, we control the behaviour of the training process of our network by setting different weights for different pixels according to the alpha matte.

We first measure the difference between the predicted probability values and the ground truth alpha values:

The difference is chosen to be L1 inside transition regions so as to recover the details of the alpha matte there, while the L2 loss is used in the rest of the regions to penalize the possible segmentation error. We find this setting can well balance between the soft segmentation and the hard segmentation

We also introduce the L1 loss on the gradients of the predicted alpha matte since it is beneficial to remove the over-blurred alpha matte after classification:

The cross-entropy (CE) loss for the foreground classification branch at a pixel p is given by:

The final loss function of the foreground classification branch with respect to an image is:

For the background classification branch, its loss L_B can be simply computed by setting α_p = 1 − α_p.

We also impose the L_F and L_B loss at each decoder block of two branches to further regulate the behaviour of the network.

Fusion Network

The goal of the fusion network is to output β_p at pixels to fuse the foreground and background classification results.

It is a fully convolution network with five convolution layers and one sigmoid layer to compute the blending weights β_p. The input of the network consists of

Training Loss


comments powered by Disqus