谢昕辰
b3b7629d38
[Fix] Fix binary segmentation ( #2101 )
...
* add out_channels
* fix forward
* add decode_head ut
* add segmentor ut
* refine postprocess
* fix
2022-09-27 16:49:38 +08:00
谢昕辰
230246f557
[Refactor] Add pyupgrade pre-commit hook ( #2078 )
...
* add pyupgrade hook
* run pyupgrade precommit hook
2022-09-19 14:06:29 +08:00
MengzhangLI
8ce7055d1a
[Fix] Fix mmengine dump of tversky ut in dev-1.x ( #2001 )
2022-08-31 12:01:55 +08:00
MengzhangLI
ee7f340dff
[Feature] Support Tversky Loss in dev-1.x branch ( #2000 )
2022-08-31 10:58:21 +08:00
Miao Zheng
032ad84375
[Fix] Config name in ut ( #1978 )
...
* [Fix] Config name in ut
* lint
2022-08-26 21:20:26 +08:00
Miao Zheng
8de0050f25
[Refactor] data flow ( #1956 )
...
* [WIP] Refactor data flow
* model return
* [WIP] Refactor data flow
* support data_samples is optional
* fix benchmark
* fix base
* minors
* rebase
* fix api
* ut
* fix api inference
* comments
* docstring
* docstring
* docstring
* fix bug of slide inference
* add assert c > 1
2022-08-26 15:54:23 +08:00
Miao Zheng
50546da85c
[Fix]Remove modules from mmcv.runner and mmcv.utils ( #1966 )
...
* [WIP] mmcv-clean
* [WIP]Remove modules from mmcv.runner and mmcv.utils
* wip
* fix import mmengine
* remove ut
* loadcheckpoint in mae
2022-08-25 15:15:21 +08:00
Miao Zheng
e0499d5a77
[Fix] Fix repo based on refactoring standard ( #1869 )
...
* [Fix] Fix repo based on refactory standard
* fix ut
2022-08-19 20:50:02 +08:00
谢昕辰
167f94a70b
rename data to structures ( #1854 )
2022-08-19 20:50:02 +08:00
zhengmiao
4b76f277a6
[Refactory] MMSegmentation Content
2022-07-15 15:47:29 +00:00
zhengmiao
b23982d1b3
[Refactory] Refactory BaseSegmentor based BaseModel
2022-06-19 06:32:09 +00:00
linfangjian.vendor
dd9c411335
[Refactor] Refactor decode_head and segmentors and add preprocess_cfg
2022-06-10 14:02:40 +00:00
zhengmiao
35c695bba7
[Refactory] Clean UTs
2022-05-09 10:29:38 +00:00
Yuan Liu
43b8a1d8dc
[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
Rockey
23b7cfc59d
[Fix] Support single cahnnel `pred` for Binary Cross Entropy Loss ( #1454 )
...
* [Fix] Fix the bug that binary cross entropy loss doesn't support single channel input
* imcrease coverage
* modify implementation
* increase coverage
* add assert
* modify implementation
* enshollow condition judge
* fix
2022-04-14 11:26:02 +08:00
jiangyitong
7b6953fd5f
[Feature] Add with cp to mit and vit ( #1431 )
...
* add with cp to mit and vit
* add test unit
Co-authored-by: jiangyitong <jiangyitong1@sensetime.com>
2022-04-01 21:01:45 +08:00
MengzhangLI
cc89c8d18d
[Fix] Add avg_non_ignore in cross entropy loss ( #1409 )
...
* [Fix] Add avg_non_ignore in cross entropy loss
* [Fix] Add avg_non_ignore in cross entropy loss
* add docstring
* fix ut
* fix docstring and comments
* fix
* fix bce
* fix avg_factor in BCE and add more ut
* add avg_non_ignore
* add more ut
* fix part of ut
* fix part of ut
* test avg_non_ignore would not affect ce/bce when reduction none/sum
* test avg_non_ignore would not affect ce/bce when reduction none/sum/mean
* re-organize ut
* re-organize ut
* re-organize ut
* re-organize hardcode case
* fix parts of comments
* fix another parts of comments
* fix
2022-03-30 18:32:47 +08:00
FangjianLin
d1b8eae35e
[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
Rockey
42df28c7bd
[Feature] add nlc2nchw2nlc and nchw2nlc2nchw ( #1249 )
...
* [Feature] add nlc2nchw2nlc and nchw2nlc2nchw
* add example
* add test, add **kwargs to make it more universal
2022-03-10 20:27:28 +08:00
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
Rockey
6665b42159
[Fix] Fix the bug that when all pixels in an image is ignored, the ac… ( #1336 )
...
* [Fix] Fix the bug that when all pixels in an image is ignored, the accuracy calculation raises ZeroDivisionError
* use eps
* all close
* add ignore test
* add eps
2022-03-09 13:20:46 +08:00
Miao Zheng
b97cfa77d2
[Enhancement] Revise pre-commit-hooks ( #1315 )
2022-02-23 23:44:27 +08:00
MengzhangLI
304df56c78
[Fix] Add Pytorch HardSwish assertion in unit test ( #1294 )
...
* assert original HardSwish when PyTorch > 1.6 in unit test
* assert original HardSwish when PyTorch > 1.6 in unit test
* assert original HardSwish when PyTorch > 1.6 in unit test
* assert original HardSwish when PyTorch > 1.6 in unit test
* assert original HardSwish when PyTorch > 1.6 in unit test
* assert original HardSwish when PyTorch > 1.6 in unit test
2022-02-16 19:47:58 +08:00
Hyeokjoon Kwon
4d451a09dc
[Fix] Make accuracy take into account ignore_index ( #1259 )
...
* make accuracy take into account ignore_index
* add UT for accuracy
2022-02-15 00:39:59 +08:00
MengzhangLI
d4f5bb25fc
[Enhancement] Upgrade isort in pre-commit hook ( #1270 )
2022-02-08 19:06:33 +08:00
rstrudel
ee47c41740
[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
MengzhangLI
c1e72513c9
[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
Lxinyang
cd246e3635
[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
Rockey
a370777e3b
[Fix] Fix the bug that mit cannot process init_cfg ( #1102 )
...
* [Fix] Fix the bug that mit cannot process init_cfg
* fix error
2021-12-08 23:17:43 -08:00
Rockey
0ad0303ebc
[Fix] Remove the inplace operation in uper_head and fpn_neck ( #1103 )
...
* [Fix] Remove the inplace operation in uper_head
* remove the inplace operation in fpn neck
* fix conflict
* increase the coverage
2021-12-09 12:12:31 +08:00
Rockey
07cd6c98e0
[Feature] Add focal loss ( #1024 )
...
* [Feature] add focal loss
* fix the bug of 'non' reduction type
* refine the implementation
* add class_weight and ignore_index; support different alpha values for different classes
* fixed some bugs
* fix bugs
* add comments
* modify test
* Update mmseg/models/losses/focal_loss.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* update test_focal_loss.py
* modified the implementation
* Update mmseg/models/losses/focal_loss.py
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* update focal_loss.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-12-03 00:18:40 -08:00
MengzhangLI
6a2cfea73b
[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
Rockey
bc27f24109
[Fix] Fix the bug that vit cannot load pretrain properly when using i… ( #999 )
...
* [Fix] Fix the bug that vit cannot load pretrain properly when using init_cfg to specify the pretrain scheme
* [Fix] fix the coverage problem
* Update mmseg/models/backbones/vit.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* [Fix] make the predicate more concise and clearer
* [Fix] Modified the judgement logic
* Update tests/test_models/test_backbones/test_vit.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* add comments
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-11-03 10:36:09 -07:00
Junjun2016
0a06853bb6
[Feature] Support TIMMBackbone ( #998 )
...
* add TIMMBackbone and unittests
* add timm to tests requirements
* deprecate pt1.3.1
* reduce the unittests input of timm backbone
* fix ci
* fix ci
* fix ci
* fix ci
* fix ci
* fix ci
* fix ci
* fix ci
* fix ci
* remove unittests of large models of timm backbone
* generate coverage report for all unittests env
* reduce the unittests input of timm backbone
* reduce the unittests input of timm backbone
2021-11-02 12:51:11 +08:00
MengzhangLI
4ea92ebbe6
smaller input & channels of unittest ( #1004 )
2021-11-01 07:47:43 -07:00
MengzhangLI
992d577783
[Fix] Change `self.loss_decode` back to `dict` in Single Loss situation. ( #1002 )
...
* fix single loss type
* fix error in ohem & point_head
* fix coverage miss
* fix uncoverage error of PointHead loss
* fix coverage miss
* fix uncoverage error of PointHead loss
* nn.modules.container.ModuleList to nn.ModuleList
* more simple format
* merge unittest def
2021-11-01 15:28:37 +08:00
Junjun2016
67f1420472
[Enhancement] Add codespell pre-commit hook and fix typos ( #920 )
...
* add codespell pre-commit hook and fix typos
* Update mmseg/models/decode_heads/dpt_head.py
* Update mmseg/models/backbones/vit.py
* Update mmseg/models/backbones/vit.py
* fix typos
* skip formating typo
* deprecate formating
* skip ipynb
* unstage ipynb changes
* unstage ipynb changes
* fix typos in ipynb
* unstage ipynb changes
2021-10-13 06:21:17 -07:00
MengzhangLI
a9d1295bc3
[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
MengzhangLI
7db1cbb181
[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
Jerry Jiarui XU
7edc71d532
[Improvement] Refactor Swin-Transformer ( #800 )
...
* [Improvement] Refactor Swin-Transformer
* fixed swin test
* update patch emebd, add more tests
* fixed test
* remove pretrain_style
* fixed padding
* resolve coments
* use mmcv 2tuple
* refactor init_cfg
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-29 08:46:33 +08:00
MengzhangLI
e701497a36
[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
MengzhangLI
f82e4d6fc9
[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
Julius Zhang
96b369bd55
[Fix] Fix loss parse in val_step ( #906 )
...
* [Fix] Fix loss parse in val_step
* Add val_step unittest
* Add train_step unittest
2021-09-25 18:17:40 -07:00
MengzhangLI
0fd3972c41
[Feature] Support multiple losses during training ( #818 )
...
* multiple losses
* fix lint error
* fix typos
* fix typos
* Adding Attribute
* Fixing loss_ prefix
* Fixing loss_ prefix
* Fixing loss_ prefix
* Add Same
* loss_name must has 'loss_' prefix
* Fix unittest
* Fix unittest
* Fix unittest
* Update mmseg/models/decode_heads/decode_head.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-24 15:08:28 +08:00
sennnnn
9ac053f49a
[Fix] Convert SyncBN to BN when training on DP ( #772 )
...
* [Fix] Convert SyncBN to BN when training on DP.
* Modify SyncBN2BN.
* Add SyncBN2BN unit test.
* Resolve some comments.
* use mmcv official revert_sync_batchnorm
* Remove local syncbn2bn unit tests.
* Update mmcv version.
* Fix bugs of gather model tools.
* Modify warnings.
* Modify docker mmcv version.
* Update mmcv version table.
2021-09-15 09:39:37 -07:00
Layne
b0787b8be2
[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
谢昕辰
ef4b30038f
[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
Guangchen Lin
b5ad23e545
[Fix] The interface multiscale_output is defined but not used ( #830 )
...
* Add interface multiscale_output
* Add space between args and their types
* Fix default value
2021-08-27 18:53:55 +08:00
谢昕辰
119bbd838d
[Enhancement] Delete convert function and add instruction to ViT/Swin README.md ( #791 )
...
* delete convert function and add instruction to README.md
* unified model convert and README
* remove url
* fix import error
* fix unittest
* rename pretrain
* rename vit and deit pretrain
* Update upernet_deit-b16_512x512_160k_ade20k.py
* Update upernet_deit-b16_512x512_80k_ade20k.py
* Update upernet_deit-b16_ln_mln_512x512_160k_ade20k.py
* Update upernet_deit-b16_mln_512x512_160k_ade20k.py
* Update upernet_deit-s16_512x512_160k_ade20k.py
* Update upernet_deit-s16_512x512_80k_ade20k.py
* Update upernet_deit-s16_ln_mln_512x512_160k_ade20k.py
* Update upernet_deit-s16_mln_512x512_160k_ade20k.py
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-08-25 15:00:41 -07:00
Junjun2016
2fe0bddf5e
[Dcos] Add header for files ( #796 )
...
* Add header for files
* Delete header in config files
2021-08-16 23:16:55 -07:00