SamuelJoutard / Permutohedral_attention_module

Apache License 2.0
29 stars 12 forks source link

Fully-connected 3D CRF #3

Closed LucasFidon closed 4 years ago

LucasFidon commented 4 years ago

I have added an implementation of a fully-connected 3D CRF for segmentation in PyTorch /CRF/crf.py using the efficient permutohedral lattice implementation of this repo.