mmsegmentation/mmseg/models/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 [Feature] Add focal loss (#1024) 2021-12-03 00:18:40 -08:00
accuracy.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
cross_entropy_loss.py [Fix] Add avg_non_ignore in cross entropy loss (#1409) 2022-03-30 18:32:47 +08:00
dice_loss.py [Fix] Remove unexist arg in docstring #1106 2021-12-06 20:03:25 +08:00
focal_loss.py [Feature] Add focal loss (#1024) 2021-12-03 00:18:40 -08:00
lovasz_loss.py [Feature] Support multiple losses during training (#818) 2021-09-24 15:08:28 +08:00
utils.py [Fix] Add avg_non_ignore in cross entropy loss (#1409) 2022-03-30 18:32:47 +08:00