Skip to content
Snippets Groups Projects
  1. Oct 07, 2021
  2. Sep 22, 2021
    • DerrickWang005's avatar
      support coco stuff-10k/164k (#625) · bf1bdba6
      DerrickWang005 authored
      
      * support coco stuff-10k/164k
      
      * update docs
      
      * fix docs
      
      * update docs
      
      * fix import lints
      
      * Update docs/dataset_prepare.md
      
      * Update docs/dataset_prepare.md
      
      * Update tools/convert_datasets/coco_stuff164k.py
      
      * Update tools/convert_datasets/coco_stuff10k.py
      
      * Update tools/convert_datasets/coco_stuff10k.py
      
      * Update tools/convert_datasets/coco_stuff10k.py
      
      * Update tools/convert_datasets/coco_stuff10k.py
      
      * Update coco_stuff.py
      
      fix the description of the dataset
      
      * Update dataset_prepare.md
      
      fix the doc tree of coco stuff 10k
      
      * Update coco_stuff10k.py
      
      fix img_dir
      
      * Update coco_stuff.py
      
      fix descriptions
      
      * Update coco_stuff164k.py
      
      fix out_dir
      
      * Update coco_stuff10k.py
      
      fix save file name
      
      * Update coco_stuff.py
      
      fix seg_map_suffix
      
      * Update dataset_prepare.md
      
      fix -p
      
      * Update dataset_prepare.md
      
      fix doc tree
      
      * modify coco stuff convertor
      
      * Remove redundant code
      
      * fix 164k convert bug
      
      * remove redundant comment
      
      * add deeplabv3 configs and more iterations
      
      * replace shutil.move with shtil.copyfile
      
      * Update deeplabv3_r50-d8_512x512_4x4_80k_coco_stuff10k.py
      
      fix wrong config
      
      * Update deeplabv3_r101-d8_512x512_4x4_80k_coco_stuff164k.py
      
      fix wrong config
      
      * fix wrong configs
      
      * fix wrong configs
      
      * fix wrong path for coco stuff 10k
      
      * fix convert bugs
      
      * fix seg_filename bug
      
      * when nproc=0, use track progress
      
      * rename configs: coco_stuff --> coco-stuff
      
      * add coco-stuff 10k and 164k to README.md
      
      * update configs
      
      * add deeplabv3 benchmark
      
      * add pspnet benchmark
      
      * remove redundant comma
      
      Co-authored-by: default avatarJunjun2016 <hejunjun@sjtu.edu.cn>
      bf1bdba6
  3. Sep 15, 2021
  4. Sep 09, 2021
  5. Sep 02, 2021
  6. Aug 28, 2021
    • 谢昕辰's avatar
      [Feature] Support dark dataset test (#815) · 0cf838f2
      谢昕辰 authored
      
      * rewrite init function
      
      * support dark_zurich test
      
      * reset image size
      
      * add night
      
      * add train_pipeline
      
      * init function parameters
      
      * remove base dataset config
      
      * remove fcn config
      
      * update doc
      
      * add datasets to README
      
      * update doc
      
      * fix table of PSPNet config
      
      * fix table of PSPNet config
      
      * change 'model' tp 'evaluation checkpoint'
      
      * fix typos in README_zh-CN
      
      Co-authored-by: default avatarMengzhangLI <mcmong@pku.edu.cn>
      0cf838f2
  7. Aug 27, 2021
  8. 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
  9. Aug 19, 2021
  10. Aug 17, 2021
  11. 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
  12. Aug 12, 2021
  13. Jul 15, 2021
  14. Jul 04, 2021
  15. Jun 28, 2021
  16. 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
  17. May 17, 2021
  18. 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
  19. 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
  20. Mar 30, 2021
  21. Mar 29, 2021
  22. Feb 25, 2021
  23. 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
  24. Jan 07, 2021
  25. Dec 15, 2020
  26. Dec 13, 2020
  27. 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
  28. 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
  29. Nov 18, 2020
  30. Nov 16, 2020
  31. Nov 10, 2020
  32. Nov 09, 2020
  33. Nov 07, 2020
  34. Oct 28, 2020
  35. Oct 23, 2020
  36. Sep 30, 2020
  37. Sep 24, 2020
  38. Sep 22, 2020
  39. 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
Loading