Skip to content
Snippets Groups Projects
  1. 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
  2. May 05, 2021
  3. 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
    • sennnnn's avatar
      Adjust vision transformer backbone architectures (#524) · c27ef919
      sennnnn 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;
      
      * * Add arg: out_indices to control model output;
      
      * Add unit test for DropPath;
      
      * Apply suggestions from code review
      
      Co-authored-by: default avatarJerry Jiarui XU <xvjiarui0826@gmail.com>
      c27ef919
  4. Apr 29, 2021
  5. Apr 25, 2021
  6. Apr 22, 2021
    • 谢昕辰's avatar
      support transformer backbone (#465) · b379b5a5
      谢昕辰 authored
      
      * vit backbone
      
      * fix lint
      
      * add docstrings and fix pretrained pos_embed dim not match prob
      
      * add unittest for vit
      
      * fix lint
      
      * add vit based fcn configs
      
      * fix import error
      
      * support multiple resolution input images
      
      * upsample pos_embed at init_weights
      
      * support resize pos_embed at evaluation
      
      * fix training errors
      
      * add more unitest code for vit backbone
      
      * unitest for uncovered code
      
      * add norm_eval unittest
      
      * refactor _pos_embeding
      
      * minor change
      
      * change var name
      
      * rafactor init_weight
      
      * load weights after resize
      
      * ignore 'module' in pretrain checkpoint
      
      * add with_cp
      
      * add with_cp
      
      Co-authored-by: default avatarJiarui XU <xvjiarui0826@gmail.com>
      b379b5a5
  7. Apr 14, 2021
  8. Mar 31, 2021
  9. Mar 30, 2021
  10. Mar 29, 2021
    • 谢昕辰's avatar
      [Bug fixed]Fix dice_loss errors (#417) · 71be1c27
      谢昕辰 authored
      * fix training bugs
      
      * fix unitest error
      
      * fix error in num_classes==2 case
      
      * delete comments
      71be1c27
    • 谢昕辰's avatar
      pytorch metrics implementation (#430) · d474cfde
      谢昕辰 authored
      * pytorch metrics impl and test
      
      * support list[str] input, delete unused test code and delete numpy version
      
      * modify input data type
      
      * add docstring and unitest of filename inputs
      
      * add indents in docstring and use tempfile lib to create dir
      
      * using with statement
      d474cfde
  11. Mar 11, 2021
    • 谢昕辰's avatar
      dice loss (#396) · a3b523e3
      谢昕辰 authored
      * dice loss
      
      * format code, add docstring and calculate denominator without valid_mask
      
      * minor change
      
      * restore
      a3b523e3
  12. Jan 20, 2021
  13. Jan 19, 2021
  14. Jan 07, 2021
  15. Dec 26, 2020
    • yamengxi's avatar
      [New model] Support MobileNetV3 (#268) · 25d8d77f
      yamengxi authored
      * delete markdownlint
      
      * Support MobileNetV3
      
      * fix import
      
      * add mobilenetv3 head and configs
      
      * Modify MobileNetV3 to semantic segmentation version
      
      * modify mobilenetv3 configs
      
      * add std configs
      
      * fix Conv2dAdaptivePadding bug
      
      * add configs
      
      * add unitest and fix bugs
      
      * fix lraspp unitest bugs
      
      * restore
      
      * fix unitest
      
      * add MobileNetV3 docstring
      
      * add mmcv
      
      * add mmcv
      
      * fix syntax bug
      
      * fix unitest bug
      
      * fix unitest bug
      
      * fix unitest bugs
      
      * fix docstring
      
      * add configs
      
      * restore
      
      * delete unnecessary assert
      
      * modify unitest
      
      * delete benchmark
      25d8d77f
  16. Dec 24, 2020
  17. Dec 18, 2020
    • Junjun2016's avatar
      Support APCNet (#299) · a5d15ae2
      Junjun2016 authored
      * Support APCNet
      
      * code optimization
      
      * add apcnet configs
      
      * add benchmark
      
      * add readme and model zoo
      
      * fix doc
      a5d15ae2
  18. Dec 15, 2020
  19. Dec 14, 2020
  20. 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
  21. 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
  22. Nov 17, 2020
  23. Nov 10, 2020
  24. Nov 09, 2020
  25. Nov 07, 2020
  26. Nov 03, 2020
    • yamengxi's avatar
      [New model] Support CGNet (#223) · 86d47300
      yamengxi authored
      
      * added cgnet
      
      * added testing for cgnet
      
      * git test
      
      * add cgnet
      
      * fix __init__
      
      * rename FGlo with GlobalContextExtractor
      
      * add readme.md and rename bn with norm
      
      * delete cg_head
      
      * fix a language mistake
      
      * rename cgnet_m3n21.py to cgnet.py
      
      * modify README.md
      
      * modify list to tuple
      
      * add fcn_head test
      
      * add assert to fcn_head
      
      * blank
      
      * fix fcn_head assert bug
      
      * add
      
      * add cgnet to README.md and model_zoo.md
      
      * modify cgnet README.md
      
      Co-authored-by: default avatarKID <wps_@mail.nankai.edu.cn>
      86d47300
  27. Oct 25, 2020
    • David de la Iglesia Castro's avatar
      Add runner type (#118) · 969c7fd8
      David de la Iglesia Castro authored
      * Add runner_type option
      
      * pre-commit
      
      * Fix max_iters
      
      * Add by_epoch to EvalHook
      
      * Add test_eval_hook for epoch runner
      
      * Remove runner-type arg from tools/train
      
      * Add missing every_n_iters check for epoch mode
      
      * Bump mmcv min version
      
      * Use build_runner
      
      * Use interval in tests
      
      * Update test_eval_hook.py
      
      * Use every_n_epochs instead of every_n_iters. Update DistEvalHook
      
      * Add test_dist_eval_hook_epoch
      
      * Fix tests
      
      * Add DeprecationWarning
      
      * Update docs
      
      * Replace DeprecationWarning with UserWarning
      969c7fd8
  28. Oct 21, 2020
    • Junjun2016's avatar
      add unet (#161) · 59564510
      Junjun2016 authored
      
      * add unet
      
      * add unet
      
      * add unet
      
      * update test_unet
      
      * update test_unet
      
      * update test_unet
      
      * update test_unet
      
      * fix bugs
      
      * add init method for unet
      
      * add test of UNet init_weights method
      
      * add registry
      
      * merge upsample
      
      * fix test
      
      * Update mmseg/models/backbones/unet.py
      
      Co-authored-by: default avatarJerry Jiarui XU <xvjiarui0826@gmail.com>
      
      * Update mmseg/models/backbones/unet.py
      
      Co-authored-by: default avatarJerry Jiarui XU <xvjiarui0826@gmail.com>
      
      * split UpConvBlock from UNet
      
      * use reversed
      
      * rename upsample module
      
      * rename upsample module
      
      * rename upsample module
      
      * rename upsample module
      
      Co-authored-by: default avatarJerry Jiarui XU <xvjiarui0826@gmail.com>
      59564510
  29. Sep 30, 2020
  30. Sep 27, 2020
  31. Sep 25, 2020
  32. Sep 21, 2020
    • sshuair's avatar
      fix acc and iou compute nan problem (#116) · cac4138f
      sshuair authored
      
      * fix acc and iou compute nan problem
      
      * fix acc and iou compute nan problem
      
      * add nan_to_num args for mean_iou
      
      * add nan_to_num args for mean_iou
      
      * add nan_to_num args for mean_iou
      
      * add nan_to_num args for mean_iou
      
      * add nan_to_num args for mean_iou
      
      * Update mmseg/core/evaluation/mean_iou.py
      
      * Update mean_iou.py
      
      * Update mean_iou.py
      
      Co-authored-by: default avatarJerry Jiarui XU <xvjiarui0826@gmail.com>
      cac4138f
  33. 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
  34. Sep 07, 2020
Loading