Question about angle field’s loss function

[KunningZheng]

Hello, thank you for your work. I tried drawing the plot of the angle field’s loss function with a fixed groundtruth A and found the following problem. I am wondering if I have misunderstood something in the paper.

[rpautrat]

Hi, I think that you are correct. The prediction in your example will converge to 0 and stay stuck there, so the loss is not optimal indeed.

I think that the second part of the loss was added at a time when the predicted angle was in [0, 2pi], but this is not the case anymore. So changing the loss directly to ||\hat{A} - A|| should work better. Thank you very much for reporting this!