-
- Downloads
[Feature] add DPT head (#605)
* add DPT head
* [fix] fix init error
* use mmcv function
* delete code
* remove transpose clas
* support NLC output shape
* Delete post_process_layer.py
* add unittest and docstring
* rename variables
* fix project error and add unittest
* match dpt weights
* add configs
* fix vit pos_embed bug and dpt feature fusion bug
* match vit output
* fix gelu
* minor change
* update unitest
* fix configs error
* inference test
* remove auxilary
* use local pretrain
* update training results
* update yml
* update fps and memory test
* update doc
* update readme
* add yml
* update doc
* remove with_cp
* update config
* update docstring
* remove dpt-l
* add init_cfg and modify readme.md
* Update dpt_vit-b16.py
* zh-n README
* use constructor instead of build function
* prevent tensor being modified by ConvModule
* fix unittest
Co-authored-by:
Junjun2016 <hejunjun@sjtu.edu.cn>
Showing
- configs/_base_/models/dpt_vit-b16.py 31 additions, 0 deletionsconfigs/_base_/models/dpt_vit-b16.py
- configs/dpt/README.md 47 additions, 0 deletionsconfigs/dpt/README.md
- configs/dpt/dpt.yml 28 additions, 0 deletionsconfigs/dpt/dpt.yml
- configs/dpt/dpt_vit-b16_512x512_160k_ade20k.py 32 additions, 0 deletionsconfigs/dpt/dpt_vit-b16_512x512_160k_ade20k.py
- mmseg/models/decode_heads/__init__.py 2 additions, 1 deletionmmseg/models/decode_heads/__init__.py
- mmseg/models/decode_heads/dpt_head.py 293 additions, 0 deletionsmmseg/models/decode_heads/dpt_head.py
- model-index.yml 1 addition, 0 deletionsmodel-index.yml
- tests/test_models/test_heads/test_dpt_head.py 48 additions, 0 deletionstests/test_models/test_heads/test_dpt_head.py
Loading
Please register or sign in to comment