About the loss function

[backseason]

It's a great piece of work.
As one of the most important contributions in your paper, will you release the code for the segmentation-aware affinity loss?
Thx ahead.