NVlabs / pacnet

Pixel-Adaptive Convolutional Neural Networks (CVPR '19)
https://suhangpro.github.io/pac/
Other
514 stars 79 forks source link

CRF usage #10

Closed yassouali closed 5 years ago

yassouali commented 5 years ago

HI, first of all thank you for this amazing work,

I am working on a semantic segmentation problem, and using CRF as a post processing step, currently I am using pydensecrf, which is a CPU implementation, it would really speed things up if I can replace it with your version of CRF, the only problem is that I can't find all the parameters corresponding to this kernel defined in the fully connected CRF paper:

The kernel above is the one used in dense CRF, and for seeing the provided example, when calling create_YXRGB I've seen that we only provide theta_alpha and theta_beta for the appearance kernel, but I can seem to find the parameters for the smoothness kernel and the weighting factors between the two kernels, I think i am missing something, I wonder how can I specify them.

Thanks.

suhangpro commented 5 years ago

Hi @yassouali,

Really sorry that I just see this now that notified it's closed. The create_YXRGB function along won't sufficiently give you the exact kernel you want. You will need to have two kernels (as two calls to add_pairwise_kernel), one with create_YXRGB and another with create_postion_feats. The pairwise_weight argument can be used to model the weighting factors.

yassouali commented 5 years ago

No worries, I figured the issue got lost, Thank you very much for the answer!