segloss function

[Clever-Huskie]

Hallo, I have a question about whether the seg function in your pytorch implementation is somewhat different from the original paper

[JianghaoWu]

The provided code is nearly identical to what you need. You can utilize this code as a basis, and kindly acknowledge the implementation from this GitHub repository in your footnotes.