[P] CRF-as-RNN: Lasagne/Theano Implementation (GPU-only) • r/MachineLearning

As a side-project I've been working on a CRF-as-RNN layer for Lasagne, based on this paper. It allows you to add differentiable CRF inference as a layer to any fully-convolutional model that produces classification output whose resolution matches the input. It's particularly useful for semantic segmentation models, but can be used for any dense prediction task. It's still somewhat of a WIP but it does work. I was able to successfully train a network with the CRF layer as a component.