mmsegmentation/tests/test_models/test_losses
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
..
__init__.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00
test_ce_loss.py [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513) 2021-04-29 01:04:15 -07:00
test_dice_loss.py [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513) 2021-04-29 01:04:15 -07:00
test_lovasz_loss.py [Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513) 2021-04-29 01:04:15 -07:00
test_utils.py refactor test organization (#440) 2021-03-30 17:55:09 -07:00