-
- Downloads
[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
Showing
- README.md 1 addition, 0 deletionsREADME.md
- README_zh-CN.md 1 addition, 0 deletionsREADME_zh-CN.md
- configs/knet/README.md 49 additions, 0 deletionsconfigs/knet/README.md
- configs/knet/knet.yml 169 additions, 0 deletionsconfigs/knet/knet.yml
- configs/knet/knet_s3_deeplabv3_r50-d8_8x2_512x512_adamw_80k_ade20k.py 93 additions, 0 deletions.../knet_s3_deeplabv3_r50-d8_8x2_512x512_adamw_80k_ade20k.py
- configs/knet/knet_s3_fcn_r50-d8_8x2_512x512_adamw_80k_ade20k.py 93 additions, 0 deletions...s/knet/knet_s3_fcn_r50-d8_8x2_512x512_adamw_80k_ade20k.py
- configs/knet/knet_s3_pspnet_r50-d8_8x2_512x512_adamw_80k_ade20k.py 92 additions, 0 deletions...net/knet_s3_pspnet_r50-d8_8x2_512x512_adamw_80k_ade20k.py
- configs/knet/knet_s3_upernet_r50-d8_8x2_512x512_adamw_80k_ade20k.py 93 additions, 0 deletions...et/knet_s3_upernet_r50-d8_8x2_512x512_adamw_80k_ade20k.py
- configs/knet/knet_s3_upernet_swin-l_8x2_512x512_adamw_80k_ade20k.py 19 additions, 0 deletions...et/knet_s3_upernet_swin-l_8x2_512x512_adamw_80k_ade20k.py
- configs/knet/knet_s3_upernet_swin-l_8x2_640x640_adamw_80k_ade20k.py 54 additions, 0 deletions...et/knet_s3_upernet_swin-l_8x2_640x640_adamw_80k_ade20k.py
- configs/knet/knet_s3_upernet_swin-t_8x2_512x512_adamw_80k_ade20k.py 57 additions, 0 deletions...et/knet_s3_upernet_swin-t_8x2_512x512_adamw_80k_ade20k.py
- mmseg/models/decode_heads/__init__.py 3 additions, 1 deletionmmseg/models/decode_heads/__init__.py
- mmseg/models/decode_heads/knet_head.py 453 additions, 0 deletionsmmseg/models/decode_heads/knet_head.py
- model-index.yml 1 addition, 0 deletionsmodel-index.yml
- tests/test_models/test_heads/test_knet_head.py 195 additions, 0 deletionstests/test_models/test_heads/test_knet_head.py
Loading
Please register or sign in to comment