R-CenterNet
R-CenterNet copied to clipboard
Problem of angle prediction
Thanks for you r code. I have trained this model on my dataset, but when i predict the targets, the angle prediction turns to be zero. Furthermore, the loss did not converge when epoch=100. Please give some suggestions.
I'm very sorry to hear from you so late. I guess there's something wrong with your label, and i have pushed the labelGenerator to here: https://github.com/ZeroE04/R-CenterNet/tree/master/labelGenerator
you can have a look, good luck for you!
Sorry, I have made the label as you suggested as retrained it, but the results are still with horizontal results. And the angle prediction is still zero.
@Fly-dream12 I also meet the same problem. I counted the angles of all my samples as below(ps:I changed the angle interval to (-180,180)):
Could it be caused by the unbalanced angular frequency of the sample? @ZeroE04