mmsegmentation/mmseg/models
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
..
backbones [Feature]: Add MAE (#1307) 2022-04-28 00:54:20 +08:00
decode_heads [Fix] inconsistent dtype ofseg_label in stdc decode (#1463) 2022-04-10 14:18:16 +08:00
losses [Fix] Fix the bug in binary_cross_entropy (#1527) 2022-04-29 19:32:12 +08:00
necks [Feature] Add BEiT backbone (#1404) 2022-03-30 15:25:10 +08:00
segmentors Fix validation loss logging (#1494) 2022-04-22 15:25:32 +08:00
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
builder.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00