Commit Graph

3 Commits (f6dca38283cde154d86074349bc8fe324d617d67)

Author SHA1 Message Date
Ziyi Wu cef8a4f611
[Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513)
* support reading class_weight from file in loss function

* add unit test of loss with class_weight from file

* minor fix

* move get_class_weight to utils
2021-04-29 01:04:15 -07:00
谢昕辰 1fc3e374e2
[Bug fixed]Fix dice_loss errors (#417)
* fix training bugs

* fix unitest error

* fix error in num_classes==2 case

* delete comments
2021-03-29 09:49:54 -07:00
谢昕辰 7e1b24dd32
dice loss (#396)
* dice loss

* format code, add docstring and calculate denominator without valid_mask

* minor change

* restore
2021-03-10 18:32:37 -08:00