mmsegmentation/tests
Rockey 1b41989934
[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
..
data [Feature] Support LoveDA dataset (#1028) 2021-11-24 19:41:19 +08:00
test_apis [Refactor] Support progressive test with fewer memory cost (#709) 2021-08-19 20:44:58 -07:00
test_data [Feature] Add Cutout transform (#1022) 2021-11-30 20:37:06 +08:00
test_models [Feature] Add focal loss (#1024) 2021-12-03 00:18:40 -08:00
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_config.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_digit_version.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_eval_hook.py [Refactor] Support progressive test with fewer memory cost (#709) 2021-08-19 20:44:58 -07:00
test_inference.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_metrics.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_sampler.py [Fix] Change `self.loss_decode` back to `dict` in Single Loss situation. (#1002) 2021-11-01 15:28:37 +08:00