mmsegmentation/tests/test_models/test_heads
谢昕辰 c1c942e8fc
[Fix] Fix binary segmentation when `num_classes==1` (#2016)
* fix binary

* add ut

* fix ut

* restore metric computation

* remove metric ut update

* set out_channels by num_classes

* replace num_classes in encoder_decoder

* update props setting and fix ut

* update ut

* minor change

* update warning
2022-09-08 14:43:21 +08:00
..
__init__.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_ann_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_apc_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_aspp_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_cc_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_da_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_decode_head.py [Fix] Fix binary segmentation when `num_classes==1` (#2016) 2022-09-08 14:43:21 +08:00
test_dm_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_dnl_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_dpt_head.py [Enhancement] Revise pre-commit-hooks (#1315) 2022-02-23 23:44:27 +08:00
test_ema_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_enc_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_fcn_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_gc_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_isa_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_knet_head.py [Feature] Support K-Net (#1289) 2022-03-10 20:05:37 +08:00
test_lraspp_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_nl_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_ocr_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_point_head.py [Fix] Change `self.loss_decode` back to `dict` in Single Loss situation. (#1002) 2021-11-01 15:28:37 +08:00
test_psa_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_psp_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_segformer_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
test_segmenter_mask_head.py [Feature] Support Segmenter (#955) 2022-01-26 13:50:51 +08:00
test_setr_mla_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_setr_up_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
test_stdc_head.py [Feature] Support STDC Network (new) (#995) 2021-12-10 23:09:32 +08:00
test_uper_head.py smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
utils.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00