Where is the code of how to compute NWD loss?

jwwangchn / NWD

Official code for "A Normalized Gaussian Wasserstein Distance for Tiny Object Detection"

Apache License 2.0

204 stars 23 forks source link

Where is the code of how to compute NWD loss? #2

Closed wuqiangch closed 2 years ago

wuqiangch commented 2 years ago

@jwwangchn Where is the code of how to compute NWD loss? I want to put the losss to my own code. Thanks!

jwwangchn commented 2 years ago

Please refer wasserstein_loss()

wuqiangch commented 2 years ago

@jwwangchn Thanks.

zimenglan-sysu-512 commented 2 years ago

hi @jwwangchn for anchor-based or anchor-free one-stage detector, only use the wasserstein_loss to boost the performace? it seems that label assignment is used for rpn of two-stage detector.