mmsegmentation/tests/test_models
Rockey cd18b6d479
[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
..
test_backbones [Feature] Add with cp to mit and vit (#1431) 2022-04-01 21:01:45 +08:00
test_heads [Feature] Support K-Net (#1289) 2022-03-10 20:05:37 +08:00
test_losses [Fix] Support single cahnnel `pred` for Binary Cross Entropy Loss (#1454) 2022-04-14 11:26:02 +08:00
test_necks [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
test_segmentors [Fix] Fix loss parse in val_step (#906) 2021-09-25 18:17:40 -07:00
test_utils [Feature] add nlc2nchw2nlc and nchw2nlc2nchw (#1249) 2022-03-10 20:27:28 +08:00
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_forward.py [Fix] Convert SyncBN to BN when training on DP (#772) 2021-09-15 09:39:37 -07:00