Megvii-BaseDetection / OTA

Official implementation of our CVPR2021 paper "OTA: Optimal Transport Assignment for Object Detection" in Pytorch.
Apache License 2.0
242 stars 24 forks source link

Sinkhorn distance implementation #19

Open utkutpcgl opened 1 year ago

utkutpcgl commented 1 year ago

Thanks for the great contribution!

As mentioned in https://github.com/Megvii-BaseDetection/OTA/issues/17#issue-1484922452, the original sinkhorn implementation is different (page 11 of your paper): image

Why is this the case?

utkutpcgl commented 1 year ago

By the way here is the original implementation u is updated first. u and v updates seem to be switched (confused?) in this paper:

https://github.com/gpeyre/SinkhornAutoDiff/blob/1e6c8c38974a325bb59cbc5b8cb6af5787b4f58d/sinkhorn_pointcloud.py#L63