Any plans for PyTorch version with MMDetection?

[LiJunnan1992]

Hi Yihui,

This method is hugely interesting and I would like to incorporate it in my research. Do you have any plans to release a PyTorch version of the code? It would be very useful to the community.

Thank you!

[LittleBoy7]

@LiJunnan1992 Do you incorporated successfully?

[LiJunnan1992]

@LittleBoy7 Not yet...

[apxlwl]

@LittleBoy7 @LiJunnan1992 Hi, I reimplemented kl-loss in with my own project Stronger-yolo-pytorch. You may find some help here.

[ethanhe42]

@wlguan This is awesome! I'll add a link to your repo in the readme

[lucs-C]

@wlguan I want to reimplemented KL loss , but I cannot find your project( Stronger-yolo-pytorch), could you please give me some help?

[ethanhe42]

@wlguan I want to reimplemented KL loss , but I cannot find your project( Stronger-yolo-pytorch), could you please give me some help?

here's a fork https://github.com/yihui-he/Stronger-yolo-pytorch

[cxliu0]

Hi, I have implemented a pytorch version of KL-Loss with MMDetection in my project KL-Loss-pytorch. The reproduced results are as follows:

KL Loss	Var Vote	soft-NMS	AP (Original Paper)	AP (Reproduced)
❌	❌	❌	37.9	38.4
✔️	❌	❌	38.5	39.2
✔️	✔️	❌	38.8	39.8
✔️	✔️	✔️	39.2	40.2