Class balanced loss code

richardaecn / class-balanced-loss

Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019

MIT License

597 stars 68 forks source link

Class balanced loss code #12

Open AmericaBG opened 4 years ago

AmericaBG commented 4 years ago

Hi! First of all, thank you ver much for your code, it's a great work!

I would like to know if you could tell me the part where the class-balanced-loss is implemented among all the files that you include.

I've been looking for it and I'm a little lost.

Thank you very much in advance!

rodrigob commented 3 years ago

code-by-jin commented 3 years ago

Loss weight calculation and assigning weights to loss are in different files. The author provided links to both parts in README under the section of "Key Implementation Details".