6563cb513e
[Feature]: Add MAE ( #1307 )
...
* [Fix]: Fix lint
* [WIP]: Add mae seg config
* [Feature]: Add MAE seg
* [Fix]: Fix mae dataset img scale bug
* [Fix]: Fix lint
* [Feature]: Change mae config to mae_segmentation's config
* [Feature]: Add interpolate pe when loading
* [Fix]: Fix pos_embed not used bug
* [Fix]: Fix lint
* [Fix]: Init rel pos embed with zeros
* [Fix]: Fix lint
* [Fix]: Change the type name of backbone to MAE
* [Fix]: Delete ade20k_512x512.py
* [Fix]: Use mmseg provided ade20k.py
* [Fix]: Change 1 sample per gpu to 2 samples per gpu
* [Fix]: Fix conflict
* [Refactor]: Use the TransformerEncoderLayer of BEiT
* [Feature]: Add UT
* [Fix]: Change the default value of qv bias to False
* [Fix]: Initialize relative pos table with zeros
* [Fix]: Delete redundant code in mae
* [Fix]: Fix lint
* [Fix]: Rename qkv_bias to qv_bias
* [Fix]: Add docstring to weight_init of MAEAttention
* [Refactor]: Delete qv_bias param
* [Fix]: Add reference to fix_init_weight
* [Fix]: Fix lint
* [Fix]: Delete extra crop size
* [Refactor]: Rename mae
* [Fix]: Set bias to True
* [Fix]: Delete redundant params
* [Fix]: Fix lint
* [Fix]: Fix UT
* [Fix]: Add resize abs pos embed
* [Fix]: Fix UT
* [Refactor]: Use build layer
* [Fix]: Add licsense and fix docstring
* [Fix]: Fix docstring
* [Feature]: Add README metafile
* [Fix]: Change 640 to 512
* [Fix]: Fix README
* fix readme of MAE
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
2022-04-28 00:54:20 +08:00
24f1563571
[Feature] Add BEiT backbone ( #1404 )
...
* [Feature] Add BEiT backbone
* fix
* fix
* fix
* fix
* add readme
* fix
* fix
* fix
* fix
* fix
* add link
* fix memory
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix test_beit.py
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
2022-03-30 15:25:10 +08:00
2e28db0dcb
[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
7ddd2fe2ec
[Feature] Support ConvNext ( #1216 )
...
* upload original backbone and configs
* ConvNext Refactor
* ConvNext Refactor
* convnext customization refactor with mmseg style
* convnext customization refactor with mmseg style
* add ade20k_640x640.py
* upload files for training
* delete dist_optimizer_hook and remove layer_decay_optimizer_constructor
* check max(out_indices) < num_stages
* add unittest
* fix lint error
* use MMClassification backbone
* fix bugs in base_1k
* add mmcls in requirements/mminstall.txt
* add mmcls in requirements/mminstall.txt
* fix drop_path_rate and layer_scale_init_value
* use logger.info instead of print
* add mmcls in runtime.txt
* fix f string && delete
* add doctring in LearningRateDecayOptimizerConstructor and fix mmcls version in requirements
* fix typo in LearningRateDecayOptimizerConstructor
* use ConvNext models in unit test for LearningRateDecayOptimizerConstructor
* add unit test
* fix typo
* fix typo
* add layer_wise and fix redundant backbone.downsample_norm in it
* fix unit test
* give a ground truth lr_scale and weight_decay
* upload models and readme
* delete 'backbone.stem_norm' and 'backbone.downsample_norm' in get_num_layer()
* fix unit test and use mmcls url
* update md2yml.py and metafile
* fix typo
2022-03-04 15:52:01 +08:00
cb1bf9f372
[Feature] Support Segmenter ( #955 )
...
* segmenter: add model
* update
* readme: update
* config: update
* segmenter: update readme
* segmenter: update
* segmenter: update
* segmenter: update
* configs: set checkpoint path to pretrain folder
* segmenter: modify vit-s/lin, remove data config
* rreadme: update
* configs: transfer from _base_ to segmenter
* configs: add 8x1 suffix
* configs: remove redundant lines
* configs: cleanup
* first attempt
* swipe CI error
* Update mmseg/models/decode_heads/__init__.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* segmenter_linear: use fcn backbone
* segmenter_mask: update
* models: add segmenter vit
* decoders: yapf+remove unused imports
* apply precommit
* segmenter/linear_head: fix
* segmenter/linear_header: fix
* segmenter: fix mask transformer
* fix error
* segmenter/mask_head: use trunc_normal init
* refactor segmenter head
* Fetch upstream (#1 )
* [Feature] Change options to cfg-option (#1129 )
* [Feature] Change option to cfg-option
* add expire date and fix the docs
* modify docstring
* [Fix] Add <!-- [ABSTRACT] --> in metafile #1127
* [Fix] Fix correct num_classes of HRNet in LoveDA dataset #1136
* Bump to v0.20.1 (#1138 )
* bump version 0.20.1
* bump version 0.20.1
* [Fix] revise --option to --options #1140
Co-authored-by: Rockey <41846794+RockeyCoss@users.noreply.github.com>
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
* decode_head: switch from linear to fcn
* fix init list formatting
* configs: remove variants, keep only vit-s on ade
* align inference metric of vit-s-mask
* configs: add vit t/b/l
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* model_converters: use torch instead of einops
* setup: remove einops
* segmenter_mask: fix missing imports
* add necessary imported init funtion
* segmenter/seg-l: set resolution to 640
* segmenter/seg-l: fix test size
* fix vitjax2mmseg
* add README and unittest
* fix unittest
* add docstring
* refactor config and add pretrained link
* fix typo
* add paper name in readme
* change segmenter config names
* fix typo in readme
* fix typos in readme
* fix segmenter typo
* fix segmenter typo
* delete redundant comma in config files
* delete redundant comma in config files
* fix convert script
* update lateset master version
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: Rockey <41846794+RockeyCoss@users.noreply.github.com>
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-26 13:50:51 +08:00
5783bc1d99
[Feature] Support STDC Network (new) ( #995 )
...
* refactor stdc code
* update key
* fix backbone inference
* remove comments
* fixing errors
* fixing version conflict
* fux typo
* use STDCHead
* upload models&logs
* adding model converters script and fix unittest
* fix error
* fix error
* fix error
* delete redundant keys in config
* fix errors in configs and unittest
* fix errors in configs and unittest
* fix errors in configs and unittest
* change Memory name
* refactor stdc2mmseg
* change name to STDC
* refactor stdc
* refactor stdc
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* refactor stdc
* stdc refactor
Co-authored-by: xiexinch <xinchen.xie@qq.com>
2021-12-10 23:09:32 +08:00
793e082eeb
[Feature] Support Twins (NeurIPS2021) ( #989 )
...
* debug
* debug
* debug
* this is a debug step, and needs to be recovered
* need recover
* git
* debug
* git
* git
* git
* git
* git
* git
* debug need recover
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debugf
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* git
* git
* use config small/base/large
* debug
* debug
* git
* debug
* git
* debug
* debug
* debug args
* debug
* debug
* git
* git
* debug
* git
* git
* git
* git
* git
* debug
* debug
* git
* debug
* git
* debug
* debug
* debug
* debug
* git
* debug
* git
* git
* debug
* debug
* git
* git
* git
* git
* debug
* debug
* debug
* debug
* git
* debug
* debug
* git
* git
* debug
* debug
* git
* debug
* debug
* debug
* git
* debug
* debug
* debug
* Please enter the commit message for your changes. Lines starting
* git
* git
* debug
* debug
* debug
* git
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* fix pre-commit error
* fix error
* git
* git
* git
* git
* git
* git
* debug
* debug
* debug
* debug
* debug
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* git
* git
* debug
* debug
* debug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* debug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* fix unittest error
* fix config errors
* fix twins2mmseg bug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* fix init_weights() in twins.py
* git
* git
* git
* git
* fix comment
* fix comment
* fix comment
* fix comment
* fix unit test coverage in TwinsPR
* Add Twins README
* Add Twins README
* twins refactor
* twins refactor
* delete init_cfg in FFN
* delete init_cfg in FFN
* Update mmseg/models/backbones/twins.py
* Update mmseg/models/backbones/twins.py
* Update mmseg/models/backbones/twins.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* Update mmseg/models/backbones/twins.py
* add conference name
Co-authored-by: linxinyang <linxinyang@meituan.com>
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-12-09 19:18:10 +08:00
ff1e2d8ef7
[Feature] Support Real-time model ERFNet ( #960 )
...
* first commit
* Fixing Unittest Error
* first refactory of ERFNet
* Refactorying NonBottleneck1d Module
* uploading models&logs
* uploading models&logs
* fix partial bugs & typos
* ERFNet
* add ERFNet with FCNHead
* fix typos of ERFNet
* add name on README.md cover
* chane name to T-ITS'2017
* fix lint error
2021-12-01 23:54:39 -08:00
6b3e5506c5
[Fix] Remove `fp16` folder in `configs`. ( #1031 )
...
* remove fp16 folder
* remove fp16 in docs
* fix some typos
* fix some typos
* fix fp16 in yml
2021-11-15 19:14:57 +08:00
f910caf229
[Feature] Support FastFCN ( #885 )
...
* FastFCN first commit
* FastFCN first commit
* Fixing lint error
* Fixing lint error
* use for loop on JPU
* Use For Loop
* Refactor FastFCN
* FastFCN
* FastFCN
* temp
* Uploading models & logs (4x4)
* Fixing typos
* fix typos
* rename config
* change README.md
* use _delete_=True
* change configs
* change start_level to 0
* change start_level to 0
* jpu
* add unittest for start_level!=0
2021-10-01 02:41:24 +08:00
d966f98f83
[Feature] Support ICNet ( #884 )
...
* add icnet backbone
* add icnet head
* add icnet configs
* nclass -> num_classes
* Support ICNet
* ICNet
* ICNet
* Add ICNeck
* Add ICNeck
* Add ICNeck
* Add ICNeck
* Adding unittest
* Uploading models & logs
* Uploading models & logs
* add comment
* smaller test_swin.py
* try to delete test_swin.py
* delete test_unet.py
* delete test_unet.py
* temp
* smaller test_unet.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-30 09:31:57 -07:00
ab12009414
[Feature] Support BiSeNetV1 ( #851 )
...
* First Commit
* fix typos
* fix typos
* Fix assertion bug
* Adding Assert
* Adding Unittest
* Fixing typo
* Uploading models & logs
* Fixing unittest error
* changing README.md
* changing README.md
2021-09-29 02:12:57 +08:00
4003b8f421
[Feature] Support BiSeNetV2 ( #804 )
...
* BiSeNetV2 first commit
* BiSeNetV2 unittest
* remove pytest
* add pytest module
* fix ConvModule input name
* fix pytest error
* fix unittest
* refactor
* BiSeNetV2 Refactory
* fix docstrings and add some small changes
* use_sigmoid=False
* fix potential bugs about upsampling
* Use ConvModule instead
* Use ConvModule instead
* fix typos
* fix typos
* fix typos
* discard nn.conv2d
* discard nn.conv2d
* discard nn.conv2d
* delete **kwargs
* uploading markdown and model
* final commit
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* Fix README conflict
* Fix unittest problem
* Fix unittest problem
* BiSeNetV2
* Fixing fps
* Fixing typpos
* bisenetv2
2021-09-26 18:52:16 +08:00
eb0baee414
[Feature] Support ISA module ( #70 )
...
* add isa module
* use more readable names, add more comments and exp results
* add unittests
* remove redundant docstring
* Apply suggestions from code review
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* fix unittest
* Update configs
* add results
* update yml
* Update README
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: xiexinch <xinchen.xie@qq.com>
2021-09-08 21:13:53 -07:00
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
f72727c563
[Tools] Add vit/swin/mit convert weight scripts ( #783 )
...
* init scripts
* update markdown
* update markdown
* add docs
* delete mit converter and use torch load function
* rename segformer readme
* update doc
* modify doc
* 更新中文文档
* Update useful_tools.md
* Update useful_tools.md
* modify doc
* update segformer.yml
2021-08-17 18:42:42 -07:00
52b4fa5b9a
[Enhancement] md2yml pre-commit hook ( #732 )
...
* init script
* update scripts and generate new yml
* fix lint: deeplabv3plus.yml
* modify resolution representation
* remove field
* format crop_size
2021-07-31 09:31:58 -07:00
36c81441c1
update metafiles ( #661 )
...
* update metafiles
* update metafiles
2021-07-01 22:31:00 +08:00
Yuan Liu
FangjianLin
MengzhangLI
rstrudel
Lxinyang
Layne
谢昕辰
Junjun2016