search

kmkurn / pytorch-crf

(Linear-chain) Conditional random field in PyTorch.
https://pytorch-crf.readthedocs.io
MIT License
935 stars 151 forks source link

Computer Vision usage #43

Closed yassouali closed 4 years ago

yassouali commented 4 years ago

Hi,

I was wandering if this implementation can work in a computer vision setting (semantic segmentation), I am using this implementation (https://github.com/lucasb-eyer/pydensecrf) but it is quite slow and I would love a GPU based implementation, so if there any chance this implementation is similar to the fully connected CRF using in https://arxiv.org/abs/1210.5644 or is it only for NLP.

Thank you.

kmkurn commented 4 years ago

Hi,

I've had a look at the paper, and at a glance, it seems very different from the linear-chain CRF this library is intended for. So I don't think this library is similar to that. You need to modify the code quite drastically, I think.

yassouali commented 4 years ago

Thank you very much for checking, much appreciated, I am reading the code to see if it is feasible.