23fb3a9812
[Fix] Fix accepting an unexpected argument `local-rank` in PyTorch 2.0 ( #2812 )
2023-03-30 16:43:15 +08:00
ff95416c3b
[Features]Support dump segment predition ( #2712 )
...
## Motivation
1. It is used to save the segmentation predictions as files and upload
these files to a test server
## Modification
1. Add output_file and format only in `IoUMetric`
## BC-breaking (Optional)
No
## Use cases (Optional)
If this PR introduces a new feature, it is better to list some use cases
here, and update the documentation.
## Checklist
1. Pre-commit or other linting tools are used to fix the potential lint
issues.
3. The modification is covered by complete unit tests. If not, please
add more unit test to ensure the correctness.
4. If the modification has potential influence on downstream projects,
this PR should be tested with downstream projects, like MMDet or
MMDet3D.
5. The documentation has been modified accordingly, like docstring or
example tutorials.
2023-03-17 22:58:08 +08:00
45fae72de5
[Feature] Support calculating FLOPs of segmentors ( #2706 )
...
## Motivation
fix compute flops problems
## Modification
Please briefly describe what modification is made in this PR.
2023-03-10 19:25:47 +08:00
eca61d3cda
tools/analysis_tools browse_dataset.py ( #2649 )
...
## Motivation
browse_dataset before training
## Modification
create tools/analysis_tools/browse_dataset.py
2023-02-27 12:02:48 +08:00
039ba5d4ca
[Feature] Support auto import modules from registry. ( #2481 )
...
## Motivation
The registry now supports auto-import modules from the given location.
register_all_modules before running is no longer needed. The modules
will be lazy-imported during building.
- [x] This PR can be merged after
https://github.com/open-mmlab/mmengine/pull/643 . The MMEngine version
should be updated.
Ref: https://github.com/open-mmlab/mmdetection/pull/9143
2023-02-23 20:33:17 +08:00
49b062e365
CodeCamp #139 [Feature] Support REFUGE dataset. ( #2554 )
...
## Motivation
Add REFUGE datasets
Old PR: https://github.com/open-mmlab/mmsegmentation/pull/2420
---------
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
2023-02-03 16:02:19 +08:00
2d67e51db3
CodeCamp #140 [New] [Feature] Add synapse dataset and data augmentation in dev-1.x. ( #2432 )
...
## Motivation
Add Synapse dataset in MMSegmentation.
Old PR: https://github.com/open-mmlab/mmsegmentation/pull/2372 .
2023-01-06 16:14:54 +08:00
da4125587e
[Refactor] Support TTA ( #2184 )
...
* tta init
* use mmcv transform
* test city
* add multiscale
* fix merge
* add softmax to post process
* add ut
* add tta pipeline to other datasets
* remove softmax
* add encoder_decoder_tta ut
* add encoder_decoder_tta ut
* rename
* rename file
* rename config
* rm aug_test
* move flip to post process
* fix channel
2022-12-30 22:52:07 +08:00
230246f557
[Refactor] Add pyupgrade pre-commit hook ( #2078 )
...
* add pyupgrade hook
* run pyupgrade precommit hook
2022-09-19 14:06:29 +08:00
3388cfd475
[Refactor]Resume do not specify ckpt ( #2074 )
...
* [Refactor]Resume do not specify ckpt
* add default false value
2022-09-15 15:11:31 +08:00
ef111dff8b
[Fix] browse dataset ( #2005 )
2022-08-31 18:08:45 +08:00
2724d47f71
[Fix] Add out-file in demo and show_result_pyplot ( #2004 )
2022-08-31 17:44:41 +08:00
aed1886724
[Fix] benckmark script ( #1991 )
2022-08-29 18:51:24 +08:00
8de0050f25
[Refactor] data flow ( #1956 )
...
* [WIP] Refactor data flow
* model return
* [WIP] Refactor data flow
* support data_samples is optional
* fix benchmark
* fix base
* minors
* rebase
* fix api
* ut
* fix api inference
* comments
* docstring
* docstring
* docstring
* fix bug of slide inference
* add assert c > 1
2022-08-26 15:54:23 +08:00
50546da85c
[Fix]Remove modules from mmcv.runner and mmcv.utils ( #1966 )
...
* [WIP] mmcv-clean
* [WIP]Remove modules from mmcv.runner and mmcv.utils
* wip
* fix import mmengine
* remove ut
* loadcheckpoint in mae
2022-08-25 15:15:21 +08:00
13b108dbc8
[Enhancement] Support custom runner ( #1923 )
...
* support custom runner
* fix
2022-08-19 20:50:03 +08:00
671ed74332
fix logdict ( #1892 )
2022-08-19 20:50:02 +08:00
e0499d5a77
[Fix] Fix repo based on refactoring standard ( #1869 )
...
* [Fix] Fix repo based on refactory standard
* fix ut
2022-08-19 20:50:02 +08:00
1ded0a4278
[Enhancement] Replace MMCV with MMEngine in convert model scripts ( #1798 )
...
* replace mmcv with mmengine
* remove transforms
2022-07-27 17:54:37 +08:00
6873f9ece8
[Feature] Add SegVisualizer ( #1792 )
...
* [Feature] Add SegVisualizer
* change name to visualizer_example
* fix typo
* refactor folder structure
2022-07-27 16:28:09 +08:00
4de57b49c5
[Fix] Fix register_all_modules and imports ( #1793 )
...
* [Fix] Fix register_all_modules and imports
* fix transforms import
2022-07-21 22:44:42 +08:00
4b76f277a6
[Refactory] MMSegmentation Content
2022-07-15 15:47:29 +00:00
fba91957c0
Merge branch 'xiexinchen/amp_train' into 'refactor_dev'
...
[Enhancement] support '--amp' option
See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!68
2022-07-14 07:29:35 +00:00
2a138b47c4
[Refactor] Fix analyze log script
2022-07-14 07:28:44 +00:00
741190a864
support '--amp' option
2022-07-14 14:21:32 +08:00
0fc9071703
[Refactor] Fix benchmark script
2022-07-14 05:34:08 +00:00
c1c870af31
rename init/inference_segmentor to init/inference_model
2022-07-12 16:19:59 +08:00
f2bac79f03
[Refactor] Refactor DatasetWrapper
2022-06-27 14:36:18 +00:00
b23982d1b3
[Refactory] Refactory BaseSegmentor based BaseModel
2022-06-19 06:32:09 +00:00
3c54de06bb
[Refactor] Refactor train test interface.
2022-06-09 03:01:43 +00:00
0644b3cd30
[Enhancement] Delete seed in dist_train.sh ( #1519 )
2022-04-28 11:35:10 +08:00
f50bfe3e3d
[Enhance] Add extra dataloader settings in configs ( #1435 )
...
* [Enhance] Add extra dataloader settings in configs
* val default samples
* val default samples
* del unuse
* del unused
2022-04-13 13:30:10 +08:00
cba10b3f15
[fix] fix image save path bug in Windows ( #1423 )
...
* [fix] fix image save path bug in Windows
* fix lint error
2022-04-09 21:17:51 +08:00
549616888e
[Fix] Fix bug in `tools/analyse_logs.py` caused by wrong plot_iter in some cases. ( #1428 )
...
* fix_analyze_logs
* fix comments
* fix comments
2022-04-05 12:47:41 +08:00
d1b8eae35e
[Feature] Add BEiT backbone ( #1404 )
...
* [Feature] Add BEiT backbone
* fix
* fix
* fix
* fix
* add readme
* fix
* fix
* fix
* fix
* fix
* add link
* fix memory
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix test_beit.py
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
2022-03-30 15:25:10 +08:00
cfc080c9b1
[Fix] make arguments effective in tools/confusion_matrix.py ( #1401 )
...
* add an argument for customizing `title' of the output figure
* fix `color_theme' arguments not passing to plot function
Signed-off-by: code14 <mob5566@gmail.com>
2022-03-21 10:51:27 +08:00
415b20febd
[Feature] Add multi machine `dist_train`. ( #1383 )
...
* Add training startup documentation
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
* fix
2022-03-18 10:55:49 +08:00
912d109dbe
Update get_flops.py ( #1371 )
...
* Update get_flops.py
argparse.ArgumentParser(description='Get the FLOPs of a segmentor')
* Update get_flops.py
2022-03-14 21:02:43 +08:00
da6bb2c8c5
[Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn ( #1362 )
2022-03-10 09:41:16 +08:00
6d124aa091
[Feature] Generating and plotting confusion matrix ( #1301 )
...
* generate and plot confusion matrix
* fix typo
* add usage and examples for confusion matrix
* deal with nan values(pick pr#7147 mmdet)
* fix md format
2022-03-03 16:34:33 +08:00
4d00877b4c
Add deprecation message for deploy tool ( #1321 )
...
* Add deprecation message for deploy tool
* deploy test
2022-02-25 11:22:03 +08:00
b97cfa77d2
[Enhancement] Revise pre-commit-hooks ( #1315 )
2022-02-23 23:44:27 +08:00
a93423e1fd
[Feature] Support iSAID aerial dataset. ( #1115 )
...
* support iSAID aerial dataset
* Update and rename docs/dataset_prepare.md to 博士/dataset_prepare.md
* Update dataset_prepare.md
* fix typo
* fix typo
* fix typo
* remove imgviz
* fix wrong order in annotation name
* upload models&logs
* upload models&logs
* add load_annotations
* fix unittest coverage
* fix unittest coverage
* fix correct crop size in config
* fix iSAID unit test
* fix iSAID unit test
* fix typos
* fix wrong crop size in readme
* use smaller figure as test data
* add smaller dataset in test data
* add blank in docs
* use 0 bytes pseudo data
* add footnote and comments for crop size
* change iSAID to isaid and add default value in it
* change iSAID to isaid in _base_
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
2022-02-17 19:07:32 +08:00
85c5eebb17
Fix bug in conversion of potsdam ( #1279 )
2022-02-11 16:31:27 +08:00
d4f5bb25fc
[Enhancement] Upgrade isort in pre-commit hook ( #1270 )
2022-02-08 19:06:33 +08:00
0d978a8822
fix log repeat ( #1267 )
2022-02-08 13:40:19 +08:00
574b195be1
[Enhance] New-style CPU training and inference. ( #1251 )
...
* [Enhance] New-style CPU training and inference.
* assert mmcv version
* SyncBN to BN in training and testing
* SyncBN to BN in training and testing
* upload untracked files to this branch
* delete gpu_ids
* fix bugs
* assert args.gpu_id in train.py
* use cfg.gpu_ids = [args.gpu_id]
* use cfg.gpu_ids = [args.gpu_id]
* fix typo
* fix typo
* fix typos
2022-01-28 23:03:01 +08:00
83100bb023
[Fix] Fix bug in non-distributed multi-gpu training/testing ( #1247 )
...
* Fix bug in non-distributed training
* Fix bug in non-distributed testing
* delete uncomment lines
* add args.gpus
2022-01-28 10:38:16 +08:00
2729a6d4d1
add multi-processes script ( #1238 )
2022-01-27 21:18:55 +08:00
ee47c41740
[Feature] Support Segmenter ( #955 )
...
* segmenter: add model
* update
* readme: update
* config: update
* segmenter: update readme
* segmenter: update
* segmenter: update
* segmenter: update
* configs: set checkpoint path to pretrain folder
* segmenter: modify vit-s/lin, remove data config
* rreadme: update
* configs: transfer from _base_ to segmenter
* configs: add 8x1 suffix
* configs: remove redundant lines
* configs: cleanup
* first attempt
* swipe CI error
* Update mmseg/models/decode_heads/__init__.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* segmenter_linear: use fcn backbone
* segmenter_mask: update
* models: add segmenter vit
* decoders: yapf+remove unused imports
* apply precommit
* segmenter/linear_head: fix
* segmenter/linear_header: fix
* segmenter: fix mask transformer
* fix error
* segmenter/mask_head: use trunc_normal init
* refactor segmenter head
* Fetch upstream (#1 )
* [Feature] Change options to cfg-option (#1129 )
* [Feature] Change option to cfg-option
* add expire date and fix the docs
* modify docstring
* [Fix] Add <!-- [ABSTRACT] --> in metafile #1127
* [Fix] Fix correct num_classes of HRNet in LoveDA dataset #1136
* Bump to v0.20.1 (#1138 )
* bump version 0.20.1
* bump version 0.20.1
* [Fix] revise --option to --options #1140
Co-authored-by: Rockey <41846794+RockeyCoss@users.noreply.github.com>
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
* decode_head: switch from linear to fcn
* fix init list formatting
* configs: remove variants, keep only vit-s on ade
* align inference metric of vit-s-mask
* configs: add vit t/b/l
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/models/decode_heads/segmenter_mask_head.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* model_converters: use torch instead of einops
* setup: remove einops
* segmenter_mask: fix missing imports
* add necessary imported init funtion
* segmenter/seg-l: set resolution to 640
* segmenter/seg-l: fix test size
* fix vitjax2mmseg
* add README and unittest
* fix unittest
* add docstring
* refactor config and add pretrained link
* fix typo
* add paper name in readme
* change segmenter config names
* fix typo in readme
* fix typos in readme
* fix segmenter typo
* fix segmenter typo
* delete redundant comma in config files
* delete redundant comma in config files
* fix convert script
* update lateset master version
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: Rockey <41846794+RockeyCoss@users.noreply.github.com>
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-26 13:50:51 +08:00
谢昕辰
Miao Zheng
jinxianwei
Andrew Lau
王永韬
MengzhangLI
zhengmiao
xiexinchen.vendor
limengzhang.vendor
linfangjian.vendor
Pan Zhang
FangjianLin
Cody Wong
wangliujun
Rockey
Hyeokjoon Kwon
Waterman0524
Sebastian
rstrudel