Conditional Random Fields as Recurrent Neural Networks

Code

• github.com/torrvision/crfasrnn
  OfficialIn papertf★ 0
• github.com/sadeepj/crfasrnn_keras
  tf★ 0
• github.com/sadeepj/crfasrnn_pytorch
  pytorch★ 0
• github.com/MiguelMonteiro/CRFasRNNLayer
  tf★ 0
• github.com/AsafBarZvi/Liver_project
  tf★ 0
• github.com/liyin2015/superpixel_crfasrnn
  tf★ 0

Abstract

Pixel-level labelling tasks, such as semantic segmentation, play a central role in image understanding. Recent approaches have attempted to harness the capabilities of deep learning techniques for image recognition to tackle pixel-level labelling tasks. One central issue in this methodology is the limited capacity of deep learning techniques to delineate visual objects. To solve this problem, we introduce a new form of convolutional neural network that combines the strengths of Convolutional Neural Networks (CNNs) and Conditional Random Fields (CRFs)-based probabilistic graphical modelling. To this end, we formulate mean-field approximate inference for the Conditional Random Fields with Gaussian pairwise potentials as Recurrent Neural Networks. This network, called CRF-RNN, is then plugged in as a part of a CNN to obtain a deep network that has desirable properties of both CNNs and CRFs. Importantly, our system fully integrates CRF modelling with CNNs, making it possible to train the whole deep network end-to-end with the usual back-propagation algorithm, avoiding offline post-processing methods for object delineation. We apply the proposed method to the problem of semantic image segmentation, obtaining top results on the challenging Pascal VOC 2012 segmentation benchmark.

Tasks

Benchmark Results

Reproductions