Background Subtraction Using Deep Learning – Part III

Published:

If you’ve seen the DL Background Subtraction project on my Github, you may find that there are five different models. The summary reports finished during my internship didn’t include model IV and V. Since I’ve been contacted by several people asking about these two models, I decide to briefly describe them.

Forget about model I~III?
summary report I
summary report II

Contents

Part I Inspiration

I was inspired by an ICCV paper: Learning deconvolution network for semantic segmentation, ICCV 2015. The original model is shown in the following figure.

figure_1_1
Figure 1.1 H. Noh, S. Hong, and B. Han,
“Learning deconvolution network for semantic segmentation”, in ICCV, 2015.

Part II Model IV

The encoder-decoder part follows the ICCV paper:

  • Encoder: VGG-16 (convolutional and pooling layers)
  • Decoder: deconvolutional layers and unpooling layers
  • All the convolutional and deconvolutional layers use same padding, down-sampling and up-sampling are performed by pooling and unpooling respectively.

The differences are:

  • The input data has 6 channels, so a convolutional layer before VGG-net is required to map the input to a 3-channel “image”.
  • After the decoder, the feature map goes through two extra convolutional layes and is transformed to a 1-channel feature. A sigmoid function is then used to get the final probability map.

Part III Model V

Until model IV, the output is a 1-channel probability map created by using sigmoid function. This means that ReLU cannot be used.

Why not just treat this task as segmentation? Here we have only two categories of pixels: foreground and background. This idea leads to model V.

  • Add ReLU after convolutional and deconvolutional layers.
  • The output is a 2-channel feature map (rather than 1-channel), which is then fed to a softmax function.
figure_3_1
Figure 3.1 results of model V

Reference

[1] H. Noh, S. Hong, and B. Han, Learning deconvolution network for semantic segmentation, in ICCV, 2015