Rockey 07cd6c98e0 [Feature] Add focal loss (#1024) ... * [Feature] add focal loss * fix the bug of 'non' reduction type * refine the implementation * add class_weight and ignore_index; support different alpha values for different classes * fixed some bugs * fix bugs * add comments * modify test * Update mmseg/models/losses/focal_loss.py Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn> * update test_focal_loss.py * modified the implementation * Update mmseg/models/losses/focal_loss.py Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com> * update focal_loss.py Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn> Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>		2021-12-03 00:18:40 -08:00
..
test_backbones	[Feature] Support Real-time model ERFNet (#960)	2021-12-01 23:54:39 -08:00
test_heads	smaller input & channels of unittest (#1004)	2021-11-01 07:47:43 -07:00
test_losses	[Feature] Add focal loss (#1024)	2021-12-03 00:18:40 -08:00
test_necks	smaller input & channels of unittest (#1004)	2021-11-01 07:47:43 -07:00
test_segmentors	[Fix] Fix loss parse in val_step (#906)	2021-09-25 18:17:40 -07:00
test_utils	[Enhancement] Add codespell pre-commit hook and fix typos (#920)	2021-10-13 06:21:17 -07: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