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. Aug 12, 2021
  4. Jul 15, 2021
  5. Jul 04, 2021
  6. Jun 28, 2021
  7. 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
  8. May 17, 2021
  9. Apr 30, 2021
    • sshuair's avatar
      add metric mFscore (#509) · 7fbdd6f1
      sshuair authored
      * add mFscore and refactor the metrics return value
      
      * fix linting
      
      * some docstring and name fix
      7fbdd6f1
  10. Apr 20, 2021
    • sennnnn's avatar
      Add support for Pascal Context 59 classes (#459) · 38d2e969
      sennnnn authored
      
      * Add support for Pascal Context 59 classes (#459)
      
      * Create PascalContextDataset59 class in mmseg/datasets/pascal_context.py;
      
      * Set reduce_zero_label=True for train_pipeline and PascalContextDataset59;
      
      * Add some configs for Pascal-Context 59 classes training and testing;
      
      * Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";
      
      * Continue(1): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";
      
      * ignore files and folders named tempxxx;
      
      * Continue(2): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";
      
      * Modify the calculation of IoU;
      
      * Modify the CLASSES order of PascalContextDataset;
      
      * Add "fcn", "deeplabv3", "deeplabv3+", "pspnet" config file for model training based on PascalContextDataset59;
      
      Add some ignore items in ".gitignore";
      
      * fix the bug "test_cfg specified in both outer field and model field " of pspnet config file;
      
      * * Clean unnecessary codes;
      
      * Add weighs link, config link, log link and evaluation results about PascalContextDataset59 in README.md
      
      * Add command line argument: "-p | --port", this arg can change the transmit port when you transmit data to distributed machine.
      
      * * Remove rebundant config files;
      
      * Remove "-p|--port" command argument;
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      38d2e969
  11. Mar 30, 2021
  12. Mar 29, 2021
  13. Feb 25, 2021
  14. Jan 10, 2021
    • yamengxi's avatar
      memory efficient test (#330) · 7e156454
      yamengxi authored
      * memory efficient test
      
      * implement efficient test
      
      * merge
      
      * Add document and docstring
      
      * fix unit test
      
      * add memory usage report
      7e156454
  15. Jan 07, 2021
  16. Dec 15, 2020
  17. Dec 13, 2020
  18. 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
  19. Nov 24, 2020
    • Junjun2016's avatar
      add dice evaluation metric (#225) · 1530af65
      Junjun2016 authored
      * add dice evaluation metric
      
      * add dice evaluation metric
      
      * add dice evaluation metric
      
      * support 2 metrics
      
      * support 2 metrics
      
      * support 2 metrics
      
      * support 2 metrics
      
      * fix docstring
      
      * use np.round once for all
      1530af65
  20. Nov 18, 2020
  21. Nov 16, 2020
  22. Nov 10, 2020
  23. Nov 09, 2020
  24. Nov 07, 2020
  25. Oct 28, 2020
  26. Oct 23, 2020
  27. Sep 30, 2020
  28. Sep 24, 2020
  29. Sep 22, 2020
  30. 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
  31. Aug 25, 2020
  32. Jul 12, 2020
    • Jerry Jiarui XU's avatar
      Add test tutorial (#9) · 2fc821ec
      Jerry Jiarui XU authored
      * add test tutorial
      
      * remote torch/torchvision from requirements
      
      * update getting started
      
      * rename drop_out_ratio -> dropout_ratio
      2fc821ec
  33. 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
  34. Jul 09, 2020
Loading