mmsegmentation/tests/test_models/test_heads
MengzhangLI 054dc66145 [Feature] Support K-Net (#1289)
* knet first commit

* fix import error in knet

* remove kernel update head from decoder head

* [Feature] Add kenerl updation for some decoder heads.

* [Feature] Add kenerl updation for some decoder heads.

* directly use forward_feature && modify other 3 decoder heads

* remover kernel_update attr

* delete unnecessary variables in forward function

* delete kernel update function

* delete kernel update function

* delete kernel_generate_head

* add unit test & comments in knet.py

* add copyright to fix lint error

* modify config names of knet

* rename swin-l 640

* upload models&logs and refactor knet_head.py

* modify docstrings and add some ut

* add url, modify docstring and add loss ut

* modify docstrings
2022-03-10 20:05:37 +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 [Feature] Support multiple losses during training (#818) 2021-09-24 15:08:28 +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