mmsegmentation/mmseg/models/losses
wangbin 578d4d0c42 [Fix] Fix the bug in binary_cross_entropy (#1527)
* [Fix] Fix the bug in binary_cross_entropy

 Fix the bug in binary_cross_entropy
'label.max() <= 1' should mask out ignore_index, since the ignore_index often set as 255.

* [Fix] Fix the bug in binary_cross_entropy, add comments

As the ignore_index often set as 255, so the binary class label check should mask out ignore_index.

Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>

* [Fix] Fix the bug in binary_cross_entropy

As the ignore_index often set as 255, so the binary class label check should mask out ignore_index.

Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: MeowZheng <meowzheng@outlook.com>
2022-04-29 19:32:12 +08:00
..
__init__.py [Feature] Add focal loss (#1024) 2021-12-03 00:18:40 -08:00
accuracy.py Fix potential bugs in accuracy.py (#1496) 2022-04-21 10:18:37 +08:00
cross_entropy_loss.py [Fix] Fix the bug in binary_cross_entropy (#1527) 2022-04-29 19:32:12 +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