Skip to content
Snippets Groups Projects
  1. Dec 09, 2021
  2. Dec 03, 2021
  3. Dec 02, 2021
    • MengzhangLI's avatar
      [Feature] Support Real-time model ERFNet (#960) · 6a2cfea7
      MengzhangLI authored
      * 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
      6a2cfea7
  4. Nov 30, 2021
    • Kyungmin Lee's avatar
      [Feature] Add Cutout transform (#1022) · 40b9ebb5
      Kyungmin Lee authored
      * Fix typo in usage example
      
      * [Feature] Add CutOut transform
      
      * CutOut repr covered by unittests
      
      * Cutout ignore index, test
      
      * ignore_index -> seg_fill_in, defualt is None
      
      * seg_fill_in is added to repr
      
      * test is modified for seg_fill_in is None
      
      * seg_fill_in (int), 0-255
      
      * add seg_fill_in test
      
      * doc string for seg_fill_in
      
      * rename CutOut to RandomCutOut, add prob
      
      * Add unittest when cutout is False
      40b9ebb5
  5. Nov 24, 2021
    • Kingdrone's avatar
      [Feature] Support LoveDA dataset (#1028) · 48d42224
      Kingdrone authored
      
      * update LoveDA dataset api
      
      * revised lint errors in dataset_prepare.md
      
      * revised lint errors in loveda.py
      
      * revised lint errors in loveda.py
      
      * revised lint errors in dataset_prepare.md
      
      * revised lint errors in dataset_prepare.md
      
      * checked with isort and yapf
      
      * checked with isort and yapf
      
      * checked with isort and yapf
      
      * Revert "checked with isort and yapf"
      
      This reverts commit 686a51d9
      
      * Revert "checked with isort and yapf"
      
      This reverts commit b877e121bb2935ceefc503c09675019489829feb.
      
      * Revert "revised lint errors in dataset_prepare.md"
      
      This reverts commit 2289e27c
      
      * Revert "checked with isort and yapf"
      
      This reverts commit 159db2f8
      
      * Revert "checked with isort and yapf"
      
      This reverts commit 159db2f8
      
      * add configs & fix bugs
      
      * update new branch
      
      * upload models&logs and add format-only
      
      * change pretraied model path of HRNet
      
      * fix the errors in dataset_prepare.md
      
      * fix the errors in dataset_prepare.md and configs in loveda.py
      
      * change the description in docs_zh-CN/dataset_prepare.md
      
      * use init_cfg
      
      * fix test converage
      
      * adding pseudo loveda dataset
      
      * adding pseudo loveda dataset
      
      * adding pseudo loveda dataset
      
      * adding pseudo loveda dataset
      
      * adding pseudo loveda dataset
      
      * adding pseudo loveda dataset
      
      * Update docs/dataset_prepare.md
      
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      
      * Update docs_zh-CN/dataset_prepare.md
      
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      
      * Update docs_zh-CN/dataset_prepare.md
      
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      
      * Delete unused lines of unittest and Add docs
      
      * add convert .py file
      
      * add downloading links from zenodo
      
      * move place of LoveDA and Cityscapes in doc
      
      * move place of LoveDA and Cityscapes in doc
      
      Co-authored-by: default avatarMengzhangLI <mcmong@pku.edu.cn>
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      48d42224
  6. Nov 03, 2021
  7. Nov 02, 2021
    • Junjun2016's avatar
      [Feature] Support TIMMBackbone (#998) · 0a06853b
      Junjun2016 authored
      * 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
      0a06853b
  8. Nov 01, 2021
  9. Oct 13, 2021
    • Junjun2016's avatar
      [Enhancement] Add codespell pre-commit hook and fix typos (#920) · 67f14204
      Junjun2016 authored
      * 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
      67f14204
  10. Oct 07, 2021
  11. Sep 30, 2021
    • MengzhangLI's avatar
      [Feature] Support FastFCN (#885) · a9d1295b
      MengzhangLI authored
      * 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
      a9d1295b
    • MengzhangLI's avatar
      [Feature] Support ICNet (#884) · 7db1cbb1
      MengzhangLI authored
      
      * 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: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      7db1cbb1
  12. Sep 29, 2021
  13. Sep 28, 2021
    • MengzhangLI's avatar
      [Feature] Support BiSeNetV1 (#851) · e701497a
      MengzhangLI authored
      * 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
      e701497a
  14. Sep 26, 2021
    • MengzhangLI's avatar
      [Feature] Support BiSeNetV2 (#804) · f82e4d6f
      MengzhangLI authored
      * 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
      f82e4d6f
    • Julius Zhang's avatar
      [Fix] Fix loss parse in val_step (#906) · 96b369bd
      Julius Zhang authored
      * [Fix] Fix loss parse in val_step
      
      * Add val_step unittest
      
      * Add train_step unittest
      96b369bd
  15. Sep 24, 2021
  16. Sep 15, 2021
    • sennnnn's avatar
      [Fix] Convert SyncBN to BN when training on DP (#772) · 9ac053f4
      sennnnn authored
      * [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.
      9ac053f4
  17. Sep 09, 2021
  18. Aug 30, 2021
    • 谢昕辰's avatar
      [Feature] add DPT head (#605) · ef4b3003
      谢昕辰 authored
      
      * 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: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      ef4b3003
  19. Aug 27, 2021
  20. Aug 25, 2021
    • 谢昕辰's avatar
      [Enhancement] Delete convert function and add instruction to ViT/Swin README.md (#791) · 119bbd83
      谢昕辰 authored
      
      * 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: default avatarJiarui XU <xvjiarui0826@gmail.com>
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      119bbd83
  21. Aug 20, 2021
    • sennnnn's avatar
      [Refactor] Support progressive test with fewer memory cost (#709) · 4e9c26bb
      sennnnn authored
      
      * Support progressive test with fewer memory cost.
      
      * Temp code
      
      * Using processor to refactor evaluation workflow.
      
      * refactor eval hook.
      
      * Fix process bar.
      
      * Fix middle save argument.
      
      * Modify some variable name of dataset evaluate api.
      
      * Modify some viriable name of eval hook.
      
      * Fix some priority bugs of eval hook.
      
      * Depreciated efficient_test.
      
      * Fix training progress blocked by eval hook.
      
      * Depreciated old test api.
      
      * Fix test api error.
      
      * Modify outer api.
      
      * Build a sampler test api.
      
      * TODO: Refactor format_results.
      
      * Modify variable names.
      
      * Fix num_classes bug.
      
      * Fix sampler index bug.
      
      * Fix grammaly bug.
      
      * Support batch sampler.
      
      * More readable test api.
      
      * Remove some command arg and fix eval hook bug.
      
      * Support format-only arg.
      
      * Modify format_results of datasets.
      
      * Modify tool which use test apis.
      
      * support cityscapes eval
      
      * fixed cityscapes
      
      * 1. Add comments for batch_sampler;
      
      2. Keep eval hook api same and add deprecated warning;
      
      3. Add doc string for dataset.pre_eval;
      
      * Add efficient_test doc string.
      
      * Modify test tool to compat old version.
      
      * Modify eval hook to compat with old version.
      
      * Modify test api to compat old version api.
      
      * Sampler explanation.
      
      * update warning
      
      * Modify deploy_test.py
      
      * compatible with old output, add efficient test back
      
      * clear logic of exclusive
      
      * Warning about efficient_test.
      
      * Modify format_results save folder.
      
      * Fix bugs of format_results.
      
      * Modify deploy_test.py.
      
      * Update doc
      
      * Fix deploy test bugs.
      
      * Fix custom dataset unit tests.
      
      * Fix dataset unit tests.
      
      * Fix eval hook unit tests.
      
      * Fix some imcompatible.
      
      * Add pre_eval argument for eval hooks.
      
      * Update eval hook doc string.
      
      * Make pre_eval false in default.
      
      * Add unit tests for dataset format_results.
      
      * Fix some comments and bc-breaking bug.
      
      * Fix pre_eval set cfg field.
      
      * Remove redundant codes.
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      4e9c26bb
  22. Aug 17, 2021
  23. Aug 13, 2021
    • sennnnn's avatar
      [Feature] Add segformer decode head and related train config (#599) · b4fd32d0
      sennnnn authored
      * [Feature]Segformer re-implementation
      
      * Using act_cfg and norm_cfg to control activation and normalization
      
      * Split this PR into several little PRs
      
      * Fix lint error
      
      * Remove SegFormerHead
      
      * [Feature] Add segformer decode head and related train config
      
      * Add ade20K trainval support for segformer
      
      1. Add related train and val configs;
      
      2. Add AlignedResize;
      
      * Set arg: find_unused_parameters = True
      
      * parameters init refactor
      
      * 1. Refactor segformer backbone parameters init;
      
      2. Remove rebundant functions and unit tests;
      
      * Remove rebundant codes
      
      * Replace Linear Layer to 1X1 Conv
      
      * Use nn.ModuleList to refactor segformer head.
      
      * Remove local to_xtuple
      
      * 1. Remove rebundant codes;
      
      2. Modify module name;
      
      * Refactor the backbone of segformer using mmcv.cnn.bricks.transformer.py
      
      * Fix some code logic bugs.
      
      * Add mit_convert.py to match pretrain keys of segformer.
      
      * Resolve some comments.
      
      * 1. Add some assert to ensure right params;
      
      2. Support flexible peconv position;
      
      * Add pe_index assert and fix unit test.
      
      * 1. Add doc string for MixVisionTransformer;
      
      2. Add some unit tests for MixVisionTransformer;
      
      * Use hw_shape to pass shape of feature map.
      
      * 1. Fix doc string of MixVisionTransformer;
      
      2. Simplify MixFFN;
      
      3. Modify H, W to hw_shape;
      
      * Add more unit tests.
      
      * Add doc string for shape convertion functions.
      
      * Add some unit tests to improve code coverage.
      
      * Fix Segformer backbone pretrain weights match bug.
      
      * Modify configs of segformer.
      
      * resolve the shape convertion functions doc string.
      
      * Add pad_to_patch_size arg.
      
      * Support progressive test with fewer memory cost.
      
      * Modify default value of pad_to_patch_size arg.
      
      * Temp code
      
      * Using processor to refactor evaluation workflow.
      
      * refactor eval hook.
      
      * Fix process bar.
      
      * Fix middle save argument.
      
      * Modify some variable name of dataset evaluate api.
      
      * Modify some viriable name of eval hook.
      
      * Fix some priority bugs of eval hook.
      
      * Fix some bugs about model loading and eval hook.
      
      * Add ade20k 640x640 dataset.
      
      * Fix related segformer configs.
      
      * Depreciated efficient_test.
      
      * Fix training progress blocked by eval hook.
      
      * Depreciated old test api.
      
      * Modify error patch size.
      
      * Fix pretrain of mit_b0
      
      * Fix the test api error.
      
      * Modify dataset base config.
      
      * Fix test api error.
      
      * Modify outer api.
      
      * Build a sampler test api.
      
      * TODO: Refactor format_results.
      
      * Modify variable names.
      
      * Fix num_classes bug.
      
      * Fix sampler index bug.
      
      * Fix grammaly bug.
      
      * Add part of benchmark results.
      
      * Support batch sampler.
      
      * More readable test api.
      
      * Remove some command arg and fix eval hook bug.
      
      * Support format-only arg.
      
      * Modify format_results of datasets.
      
      * Modify tool which use test apis.
      
      * Update readme.
      
      * Update readme of segformer.
      
      * Updata readme of segformer.
      
      * Update segformer readme and fix segformer mit_b4.
      
      * Update readme of segformer.
      
      * Clean AlignedResize related config.
      
      * Clean code from pr #709
      
      * Clean code from pr #709
      
      * Add 512x512 segformer_mit-b5.
      
      * Fix lint.
      
      * Fix some segformer head bugs.
      
      * Add segformer unit tests.
      
      * Replace AlignedResize to ResizeToMultiple.
      
      * Modify readme of segformer.
      
      * Fix bug of ResizeToMultiple.
      
      * Add ResizeToMultiple unit tests.
      
      * Resolve conflict.
      
      * Simplify the implementation of ResizeToMultiple.
      
      * Update test results.
      
      * Fix multi-scale test error when resize_ratio=1.75 and input size=640x640.
      
      * Update segformer results.
      
      * Update Segformer results.
      
      * Fix some url bugs and pipelines bug.
      
      * Move ckpt convertion to tools.
      
      * Add segformer official pretrain weights usage.
      
      * Clean redundant codes.
      
      * Remove redundant codes.
      
      * Unfied format.
      
      * Add description for segformer converter.
      
      * Update workers.
      b4fd32d0
  24. Aug 12, 2021
  25. Aug 03, 2021
  26. Jul 28, 2021
  27. Jul 19, 2021
    • sennnnn's avatar
      [Feature] Segformer backbone re-implementation (#594) · 095ed243
      sennnnn authored
      * [Feature]Segformer re-implementation
      
      * Using act_cfg and norm_cfg to control activation and normalization
      
      * Split this PR into several little PRs
      
      * Fix lint error
      
      * Remove SegFormerHead
      
      * parameters init refactor
      
      * 1. Refactor segformer backbone parameters init;
      
      2. Remove rebundant functions and unit tests;
      
      * Remove rebundant codes
      
      * 1. Remove rebundant codes;
      
      2. Modify module name;
      
      * Refactor the backbone of segformer using mmcv.cnn.bricks.transformer.py
      
      * Fix some code logic bugs.
      
      * Add mit_convert.py to match pretrain keys of segformer.
      
      * Resolve some comments.
      
      * 1. Add some assert to ensure right params;
      
      2. Support flexible peconv position;
      
      * Add pe_index assert and fix unit test.
      
      * 1. Add doc string for MixVisionTransformer;
      
      2. Add some unit tests for MixVisionTransformer;
      
      * Use hw_shape to pass shape of feature map.
      
      * 1. Fix doc string of MixVisionTransformer;
      
      2. Simplify MixFFN;
      
      3. Modify H, W to hw_shape;
      
      * Add more unit tests.
      
      * Add doc string for shape convertion functions.
      
      * Add some unit tests to improve code coverage.
      
      * Fix Segformer backbone pretrain weights match bug.
      
      * resolve the shape convertion functions doc string.
      
      * Add pad_to_patch_size arg.
      
      * Modify default value of pad_to_patch_size arg.
      095ed243
    • 谢昕辰's avatar
      [Fix] fix patch_embed and pos_embed mismatch error (#685) · f6246d6e
      谢昕辰 authored
      
      * fix patch_embed and pos_embed mismatch error
      
      * add docstring
      
      * update unittest
      
      * use downsampled image shape
      
      * use tuple
      
      * remove unused parameters and add doc
      
      * fix init weights function
      
      * revise docstring
      
      * Update vit.py
      
      If -> Whether
      
      * fix lint
      
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      f6246d6e
  28. Jul 01, 2021
    • Ze Liu's avatar
      [WIP] Add Swin Transformer (#511) · 214d083c
      Ze Liu authored
      
      * add Swin Transformer
      
      * add Swin Transformer
      
      * fixed import
      
      * Add some swin training settings.
      
      * Fix some filename error.
      
      * Fix attribute name: pretrain -> pretrained
      
      * Upload mmcls implementation of swin transformer.
      
      * Refactor Swin Transformer to follow mmcls style.
      
      * Refactor init_weigths of swin_transformer.py
      
      * Fix lint
      
      * Match inference precision
      
      * Add some comments
      
      * Add swin_convert to load official style ckpt
      
      * Remove arg: auto_pad
      
      * 1. Complete comments for each block;
      
      2. Correct weight convert function;
      
      3. Fix the pad of Patch Merging;
      
      * Clean function args.
      
      * Fix vit unit test.
      
      * 1. Add swin transformer unit tests;
      
      2. Fix some pad bug;
      
      3. Modify config to adapt new swin implementation;
      
      * Modify config arg
      
      * Update readme.md of swin
      
      * Fix config arg error and Add some swin benchmark msg.
      
      * Add MeM and ms test content for readme.md of swin transformer.
      
      * Fix doc string of swin module
      
      * 1. Register swin transformer to model list;
      
      2. Modify pth url which keep meta attribute;
      
      * Update swin.py
      
      * Merge config settings.
      
      * Modify config style.
      
      * Update README.md
      
      Add ViT link
      
      * Modify main readme.md
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      Co-authored-by: default avatarsennnnn <201730271412@mail.scut.edu.cn>
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      214d083c
    • 谢昕辰's avatar
      add configs for vit backbone plus decode_heads (#520) · 5245edb0
      谢昕辰 authored
      * add config
      
      * add cityscapes config
      
      * add default value to docstring
      
      * fix lint
      
      * add deit-s and deit-b
      
      * add readme
      
      * add eps at norm_cfg
      
      * add drop_path_rate experiment
      
      * add deit case at init_weight
      
      * add upernet result
      
      * update result and add upernet 160k config
      
      * update upernet result and fix settings
      
      * Update iters number
      
      * update result and delete some configs
      
      * fix import error
      
      * fix drop_path_rate
      
      * update result and restore config
      
      * update benchmark result
      
      * remove cityscapes exp
      
      * remove neck
      
      * neck exp
      
      * add more configs
      
      * fix init error
      
      * fix ffn setting
      
      * update result
      
      * update results
      
      * update result
      
      * update results and fill table
      
      * delete or rename configs
      
      * fix link delimiter
      
      * rename configs and fix link
      
      * rename neck to mln
      5245edb0
  29. Jun 24, 2021
  30. Jun 23, 2021
    • Sixiao Zheng's avatar
      [Feature] Official implementation of SETR (#531) · 5876868a
      Sixiao Zheng authored
      
      * Adjust vision transformer backbone architectures;
      
      * Add DropPath, trunc_normal_ for VisionTransformer implementation;
      
      * Add class token buring intermediate period and remove it during final period;
      
      * Fix some parameters loss bug;
      
      * * Store intermediate token features and impose no processes on them;
      
      * Remove class token and reshape entire token feature from NLC to NCHW;
      
      * Fix some doc error
      
      * Add a arg for VisionTransformer backbone to control if input class token into transformer;
      
      * Add stochastic depth decay rule for DropPath;
      
      * * Fix output bug when input_cls_token=False;
      
      * Add related unit test;
      
      * Re-implement of SETR
      
      * Add two head -- SETRUPHead (Naive, PUP) & SETRMLAHead (MLA);
      
      * * Modify some docs of heads of SETR;
      
      * Add MLA auxiliary head of SETR;
      
      * * Modify some arg of setr heads;
      
      * Add unit test for setr heads;
      
      * * Add 768x768 cityscapes dataset config;
      
      * Add Backbone: SETR -- Backbone: MLA, PUP, Naive;
      
      * Add SETR cityscapes training & testing config;
      
      * * Fix the low code coverage of unit test about heads of setr;
      
      * Remove some rebundant error capture;
      
      * * Add pascal context dataset & ade20k dataset config;
      
      * Modify auxiliary head relative config;
      
      * Modify folder structure.
      
      * add setr
      
      * modify vit
      
      * Fix the test_cfg arg position;
      
      * Fix some learning schedule bug;
      
      * optimize setr code
      
      * Add arg: final_reshape to control if converting output feature information from NLC to NCHW;
      
      * Fix the default value of final_reshape;
      
      * Modify arg: final_reshape to arg: out_shape;
      
      * Fix some unit test bug;
      
      * Add MLA neck;
      
      * Modify setr configs to add MLA neck;
      
      * Modify MLA decode head to remove rebundant structure;
      
      * Remove some rebundant files.
      
      * * Fix the code style bug;
      
      * Remove some rebundant files;
      
      * Modify some unit tests of SETR;
      
      * Ignoring CityscapesCoarseDataset and MapillaryDataset.
      
      * Fix the activation function loss bug;
      
      * Fix the img_size bug of SETR_PUP_ADE20K
      
      * * Fix the lint bug of transformers.py;
      
      * Add mla neck unit test;
      
      * Convert vit of setr out shape from NLC to NCHW.
      
      * * Modify Resize action of data pipeline;
      
      * Fix deit related bug;
      
      * Set find_unused_parameters=False for pascal context dataset;
      
      * Remove arg: find_unused_parameters which is False by default.
      
      * Error auxiliary head of PUP deit
      
      * Remove the minimal restrict of slide inference.
      
      * Modify doc string of Resize
      
      * Seperate this part of code to a new PR #544
      
      * * Remove some rebundant codes;
      
      * Modify unit tests of SETR heads;
      
      * Fix the tuple in_channels of mla_deit.
      
      * Modify code style
      
      * Move detailed definition of auxiliary head into model config dict;
      
      * Add some setr config for default cityscapes.py;
      
      * Fix the doc string of SETR head;
      
      * Modify implementation of SETR Heads
      
      * Remove setr aux head and use fcn head to replace it;
      
      * Remove arg: img_size and remove last interpolate op of heads;
      
      * Rename arg: conv3x3_conv1x1 to kernel_size of SETRUPHead;
      
      * non-square input support for setr heads
      
      * Modify config argument for above commits
      
      * Remove norm_layer argument of SETRMLAHead
      
      * Add mla_align_corners for MLAModule interpolate
      
      * [Refactor]Refactor of SETRMLAHead
      
      * Modify Head implementation;
      
      * Modify Head unit test;
      
      * Modify related config file;
      
      * [Refactor]MLA Neck
      
      * Fix config bug
      
      * [Refactor]SETR Naive Head and SETR PUP Head
      
      * [Fix]Fix the lack of arg: act_cfg and arg: norm_cfg
      
      * Fix config error
      
      * Refactor of SETR MLA, Naive, PUP heads.
      
      * Modify some attribute name of SETR Heads.
      
      * Modify setr configs to adapt new vit code.
      
      * Fix trunc_normal_ bug
      
      * Parameters init adjustment.
      
      * Remove redundant doc string of SETRUPHead
      
      * Fix pretrained bug
      
      * [Fix] Fix vit init bug
      
      * Add some vit unit tests
      
      * Modify module import
      
      * Remove norm from PatchEmbed
      
      * Fix pretrain weights bug
      
      * Modify pretrained judge
      
      * Fix some gradient backward bugs.
      
      * Add some unit tests to improve code cov
      
      * Fix init_weights of setr up head
      
      * Add DropPath in FFN
      
      * Finish benchmark of SETR
      
      1. Add benchmark information into README.MD of SETR;
      
      2. Fix some name bugs of vit;
      
      * Remove DropPath implementation and use DropPath from mmcv.
      
      * Modify out_indices arg
      
      * Fix out_indices bug.
      
      * Remove cityscapes base dataset config.
      
      Co-authored-by: default avatarsennnnn <201730271412@mail.scut.edu.cn>
      Co-authored-by: default avatarCuttlefishXuan <zhaoxinxuan1997@gmail.com>
      5876868a
  31. Jun 19, 2021
    • sennnnn's avatar
      [Fix] Fix some vit init bugs (#609) · 2a9bf2d2
      sennnnn authored
      * [Fix] Fix vit init bug
      
      * Add some vit unit tests
      
      * Modify module import
      
      * Fix pretrain weights bug
      
      * Modify pretrained judge
      
      * Add some unit tests to improve code cov
      
      * Optimize code
      
      * Fix vit unit test
      2a9bf2d2
  32. Jun 17, 2021
    • sennnnn's avatar
      [Refactor] Using mmcv transformer bricks to refactor vit. (#571) · c01abb4f
      sennnnn authored
      * [Refactor] Using mmcv bricks to refactor vit
      
      * Follow the vit code structure from mmclassification
      
      * Add MMCV install into CI system.
      
      * Add  to 'Install MMCV' CI item
      
      * Add 'Install MMCV_CPU' and 'Install MMCV_GPU CI' items
      
      * Fix & Add
      
      1. Fix low code coverage of vit.py;
      
      2. Remove HybirdEmbed;
      
      3. Fix doc string of VisionTransformer;
      
      * Add helpers unit test.
      
      * Add converter to convert vit pretrain weights from timm style to mmcls style.
      
      * Clean some rebundant code and refactor init
      
      1. Use timm style init_weights;
      
      2. Remove to_xtuple and trunc_norm_;
      
      * Add comments for VisionTransformer.init_weights()
      
      * Add arg: pretrain_style to choose timm or mmcls vit pretrain weights.
      c01abb4f
    • Jerry Jiarui XU's avatar
      [Refactor]: Unified parameter initialization (#567) · 9849a8dc
      Jerry Jiarui XU authored
      * [Refactor]: Unified parameter initialization
      
      * fixed pretrained
      9849a8dc
  33. May 06, 2021
    • sennnnn's avatar
      Add option for output shape of ViT (#530) · 5c195db1
      sennnnn authored
      * Add arg: final_reshape to control if converting output feature information from NLC to NCHW;
      
      * Fix the default value of final_reshape;
      
      * Modify arg: final_reshape to arg: out_shape;
      
      * Fix some unit test bug;
      5c195db1
Loading