mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/mmseg/models/decode_heads
History
谢昕辰 2825efe378
[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>
 		2021-08-30 16:53:05 +08:00
..
__init__.py [Feature] add DPT head (#605) 2021-08-30 16:53:05 +08:00
ann_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
apc_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
aspp_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
cascade_decode_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
cc_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
da_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
decode_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
dm_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
dnl_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
dpt_head.py [Feature] add DPT head (#605) 2021-08-30 16:53:05 +08:00
ema_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
enc_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
fcn_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
fpn_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
gc_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
lraspp_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
nl_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
ocr_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
point_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
psa_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
psp_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
segformer_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
sep_aspp_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
sep_fcn_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
setr_mla_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
setr_up_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00
uper_head.py [Dcos] Add header for files (#796) 2021-08-16 23:16:55 -07:00