mmsegmentation/tests/test_models/test_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 [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_ce_loss.py [Fix] Support single cahnnel `pred` for Binary Cross Entropy Loss (#1454) 2022-04-14 11:26:02 +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