mmsegmentation/tests/test_models/test_losses
MengzhangLI cc89c8d18d [Fix] Add avg_non_ignore in cross entropy loss (#1409)
* [Fix] Add avg_non_ignore in cross entropy loss

* [Fix] Add avg_non_ignore in cross entropy loss

* add docstring

* fix ut

* fix docstring and comments

* fix

* fix bce

* fix avg_factor in BCE and add more ut

* add avg_non_ignore

* add more ut

* fix part of ut

* fix part of ut

* test avg_non_ignore would not affect ce/bce when reduction none/sum

* test avg_non_ignore would not affect ce/bce when reduction none/sum/mean

* re-organize ut

* re-organize ut

* re-organize ut

* re-organize hardcode case

* fix parts of comments

* fix another parts of comments

* fix
2022-03-30 18:32:47 +08:00
..
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_ce_loss.py [Fix] Add avg_non_ignore in cross entropy loss (#1409) 2022-03-30 18:32:47 +08:00
test_dice_loss.py [Enhancement] Upgrade isort in pre-commit hook (#1270) 2022-02-08 19:06:33 +08:00
test_focal_loss.py [Feature] Add focal loss (#1024) 2021-12-03 00:18:40 -08:00
test_lovasz_loss.py [Enhancement] Upgrade isort in pre-commit hook (#1270) 2022-02-08 19:06:33 +08:00
test_utils.py [Fix] Fix the bug that when all pixels in an image is ignored, the ac… (#1336) 2022-03-09 13:20:46 +08:00