mmsegmentation/mmseg/models/losses
Rockey 23b7cfc59d [Fix] Support single cahnnel `pred` for Binary Cross Entropy Loss (#1454)
* [Fix] Fix the bug that binary cross entropy loss doesn't support single channel input

* imcrease coverage

* modify implementation

* increase coverage

* add assert

* modify implementation

* enshollow condition judge

* fix
2022-04-14 11:26:02 +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] Support single cahnnel `pred` for Binary Cross Entropy Loss (#1454) 2022-04-14 11:26:02 +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