Skip to content
Snippets Groups Projects
  1. Aug 17, 2021
  2. 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
  3. 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
  4. Mar 30, 2021
  5. Feb 25, 2021
  6. Jan 07, 2021
  7. Dec 15, 2020
  8. Dec 13, 2020
  9. Dec 02, 2020
    • yamengxi's avatar
      [Feature]add CLAHE transform (#229) · 26f4bed2
      yamengxi authored
      * add CLAHE transform
      
      * fix syntax error
      
      * fix syntax error
      
      * restore
      
      * add a test
      
      * modify cv2 to mmcv
      
      * add docstring
      
      * modify
      
      * restore
      
      * fix mmcv.clahe error
      
      * change mmcv version to 1.3.0
      
      * fix bugs
      
      * add all data transformers to __init__
      
      * fix __init__
      
      * fix test_transform
      26f4bed2
    • yamengxi's avatar
      [Feature] add AdjustGamma transform (#232) · e8d643fe
      yamengxi authored
      * add AdjustGamma transform
      
      * restore
      
      * change cv2 to mmcv
      
      * simplify AdjustGamma
      
      * fix syntax error
      
      * modify
      
      * fix syntax error
      
      * change mmcv version to 1.3.0
      
      * fix lut function name error
      
      * fix syntax error
      
      * fix range
      e8d643fe
  10. Nov 18, 2020
  11. Nov 16, 2020
  12. Nov 10, 2020
  13. Nov 09, 2020
  14. Nov 07, 2020
  15. Sep 16, 2020
    • Iago González's avatar
      Add support for custom classes (#71) · 7c6fa484
      Iago González authored
      
      * Support for custom classes
      
      * Fix test
      
      * Fix pre-commit
      
      * Add pipeline logic for custom classes
      
      * Fix minor issues, fix test
      
      * Fix issues from PR review
      
      * Fix tests
      
      * Remove palette as str
      
      * Rename old_to_new_ids to label_map
      
      * Test for load_anns
      
      * Remove get_palette function
      
      * fixed temp
      
      * Add subset of palette, remove palette as arg
      
      * minor update
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      7c6fa484
  16. Jul 10, 2020
    • Jerry Jiarui XU's avatar
      Add Colab Tutorial (#7) · 6c6330fe
      Jerry Jiarui XU authored
      * add badge
      
      * Created using Colaboratory
      
      * add read docs
      
      * Fixed readthedocs
      
      * fixed colab ref
      
      * add readthedocs.txt
      
      * add link
      
      * fixed modelzoo link
      
      * add missing reference
      
      * fixed docs
      
      * remove relative path in docs
      
      * add colab in README.md
      
      * update docker image
      
      * add newline
      
      * fixed br
      6c6330fe
  17. Jul 09, 2020
Loading