zhijiejia
8dbbdd8c17
[Feature] Add model ensemble tools ( #2218 )
...
* [Feature] Add model ensemble tool
* [Enhance] Add en and zh_cn instructions for model_ensemble
* [Enhance] Add default-value for --out and modify instruction
* [Enhance] Add arg-type for --out
* [Enhance] Delete redundant code
2022-10-24 19:03:48 +08:00
Timo Kaiser
319ab5f2f3
Fixed Bug as explained in Issue #1969 ( #1992 )
...
For explanations, have a look at
https://github.com/open-mmlab/mmsegmentation/issues/1969
2022-08-29 18:20:30 +08:00
Yi Zhao
5fc7c5cd12
fix beit model converter ( #1616 )
...
* fix beit model converter
* Update beit2mmseg.py
I have modified the code according to your suggestion. Thanks.
2022-06-21 11:08:15 +08:00
Aston.He
7628a61f92
feat(mlu): Support PyTorch backend on MLU. ( #1515 )
...
* feat(mlu): Support PyTorch backend on MLU.
* fix redundant device variable.
* Update mmseg/apis/train.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update comments.
* Update mmseg/apis/train.py
* Update is_mlu_available flag.
* align util_distribution.py to mmdet.
* align util_distribution.py to mmdet.
* add build_dp, build_ddp testcase.
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmseg/utils/util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update tests/test_utils/test_util_distribution.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* add mmcv version check for mlu device.
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-25 18:11:42 +08:00
Miao Zheng
b96e0d1641
[Enhancement] Delete seed in dist_train.sh ( #1519 )
2022-04-28 11:35:10 +08:00
Miao Zheng
3f797072d8
[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
Pan Zhang
da8b79b4da
[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
MengzhangLI
dfcc0b68df
[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
FangjianLin
24f1563571
[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
Cody Wong
18859e4de0
[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
FangjianLin
1b24ad656f
[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
wangliujun
a3aab38bb4
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
Rockey
7dba7a2620
[Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn ( #1362 )
2022-03-10 09:41:16 +08:00
Hyeokjoon Kwon
369a2ee9bb
[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
Miao Zheng
c0442f1b4a
Add deprecation message for deploy tool ( #1321 )
...
* Add deprecation message for deploy tool
* deploy test
2022-02-25 11:22:03 +08:00
Miao Zheng
e5cd755481
[Enhancement] Revise pre-commit-hooks ( #1315 )
2022-02-23 23:44:27 +08:00
Waterman0524
4f4e7728b0
[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
Sebastian
0934a57f4f
Fix bug in conversion of potsdam ( #1279 )
2022-02-11 16:31:27 +08:00
MengzhangLI
b4314f98c1
[Enhancement] Upgrade isort in pre-commit hook ( #1270 )
2022-02-08 19:06:33 +08:00
MengzhangLI
af88b868a6
fix log repeat ( #1267 )
2022-02-08 13:40:19 +08:00
MengzhangLI
ecda117ea5
[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
MengzhangLI
622f28eea3
[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
MengzhangLI
8ea3d64c65
add multi-processes script ( #1238 )
2022-01-27 21:18:55 +08:00
rstrudel
cb1bf9f372
[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
MengzhangLI
4b905cbe2f
[Feature] Support ISPRS Vaihingen Dataset. ( #1171 )
...
* Add Vaihingen
* upload models&logs of vaihingen
* fix unit test
* fix dataset pipeline
* fix unit test coverage
* fix vaihingen docstring
2022-01-22 20:27:51 +08:00
Rockey
7512f05990
[Feature] benchmark can add work_dir and repeat times, test.py now has default work-dir ( #1126 )
...
* [Feature] benchmark can add work_dir and repeat times
* change the parameter's name
* change the name of the log file
* add skp road
* add default work dir
* make it optional
* Update tools/benchmark.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update tools/benchmark.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* fix typo
* modify json name
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-18 16:38:31 +08:00
Kingdrone
b997a13e28
[Feature] Support ISPRS Potsdam Dataset. ( #1097 )
...
* add isprs potsdam dataset
* add isprs dataset configs
* fix lint error
* fix potsdam conversion bug
* fix error in potsdam class
* fix error in potsdam class
* add vaihingen dataset
* add vaihingen dataset
* add vaihingen dataset
* fix some description errors.
* fix some description errors.
* fix some description errors.
* upload models & logs of Potsdam
* remove vaihingen and add unit test
* add chinese readme
* add pseudodataset
* use mmcv and add class_names
* use f-string
* add new dataset unittest
* add docstring and remove global variables args
* fix metafile error in PSPNet
* fix pretrained value
* Add dataset info
* fix typo
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
2022-01-18 14:15:15 +08:00
Kyungmin Lee
6c3e63e48b
[Feature] Add MultiImageMixDataset ( #1105 )
...
* Fix typo in usage example
* original MultiImageMixDataset code in mmdet
* Add MultiImageMixDataset unittests in test_dataset_wrapper
* fix lint error
* fix value name ann_file to ann_dir
* modify retrieve_data_cfg (#1 )
* remove dynamic_scale & add palette
* modify retrieve_data_cfg method
* modify retrieve_data_cfg func
* fix error
* improve the unittests coverage
* fix unittests error
* Dataset (#2 )
* add cfg-options
* Add unittest in test_build_dataset
* add blank line
* add blank line
* add a blank line
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Co-authored-by: Younghoon-Lee <72462227+Younghoon-Lee@users.noreply.github.com>
Co-authored-by: MeowZheng <meowzheng@outlook.com>
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-11 16:18:24 +08:00
Rockey
78fdf65382
[Feature] add auto resume ( #1172 )
...
* [Feature] add auto resume
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* modify docstring
* Update mmseg/utils/find_latest_checkpoint.py
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
* add copyright
Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
2022-01-11 12:27:24 +08:00
Rockey
4e9a95d4f2
[Fix] revise --option to --options #1140
2021-12-15 15:24:08 +08:00
Rockey
6f844429ea
[Feature] Change options to cfg-option ( #1129 )
...
* [Feature] Change option to cfg-option
* add expire date and fix the docs
* modify docstring
2021-12-14 19:11:52 +08:00
MengzhangLI
5783bc1d99
[Feature] Support STDC Network (new) ( #995 )
...
* refactor stdc code
* update key
* fix backbone inference
* remove comments
* fixing errors
* fixing version conflict
* fux typo
* use STDCHead
* upload models&logs
* adding model converters script and fix unittest
* fix error
* fix error
* fix error
* delete redundant keys in config
* fix errors in configs and unittest
* fix errors in configs and unittest
* fix errors in configs and unittest
* change Memory name
* refactor stdc2mmseg
* change name to STDC
* refactor stdc
* refactor stdc
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* stdc refactor
* refactor stdc
* stdc refactor
Co-authored-by: xiexinch <xinchen.xie@qq.com>
2021-12-10 23:09:32 +08:00
Lxinyang
793e082eeb
[Feature] Support Twins (NeurIPS2021) ( #989 )
...
* debug
* debug
* debug
* this is a debug step, and needs to be recovered
* need recover
* git
* debug
* git
* git
* git
* git
* git
* git
* debug need recover
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debugf
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* git
* git
* use config small/base/large
* debug
* debug
* git
* debug
* git
* debug
* debug
* debug args
* debug
* debug
* git
* git
* debug
* git
* git
* git
* git
* git
* debug
* debug
* git
* debug
* git
* debug
* debug
* debug
* debug
* git
* debug
* git
* git
* debug
* debug
* git
* git
* git
* git
* debug
* debug
* debug
* debug
* git
* debug
* debug
* git
* git
* debug
* debug
* git
* debug
* debug
* debug
* git
* debug
* debug
* debug
* Please enter the commit message for your changes. Lines starting
* git
* git
* debug
* debug
* debug
* git
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* fix pre-commit error
* fix error
* git
* git
* git
* git
* git
* git
* debug
* debug
* debug
* debug
* debug
* debug
* git
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* debug
* git
* git
* git
* debug
* debug
* debug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* debug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* fix unittest error
* fix config errors
* fix twins2mmseg bug
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* git
* fix init_weights() in twins.py
* git
* git
* git
* git
* fix comment
* fix comment
* fix comment
* fix comment
* fix unit test coverage in TwinsPR
* Add Twins README
* Add Twins README
* twins refactor
* twins refactor
* delete init_cfg in FFN
* delete init_cfg in FFN
* Update mmseg/models/backbones/twins.py
* Update mmseg/models/backbones/twins.py
* Update mmseg/models/backbones/twins.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* Update mmseg/models/backbones/twins.py
* add conference name
Co-authored-by: linxinyang <linxinyang@meituan.com>
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-12-09 19:18:10 +08:00
Junjun2016
f0426c97d7
[Enchance] Set a random seed when the user does not set a seed. ( #1039 )
2021-11-26 10:55:15 +08:00
Kingdrone
2bd7f60785
[Feature] Support LoveDA dataset ( #1028 )
...
* update LoveDA dataset api
* revised lint errors in dataset_prepare.md
* revised lint errors in loveda.py
* revised lint errors in loveda.py
* revised lint errors in dataset_prepare.md
* revised lint errors in dataset_prepare.md
* checked with isort and yapf
* checked with isort and yapf
* checked with isort and yapf
* Revert "checked with isort and yapf"
This reverts commit 686a51d9
* Revert "checked with isort and yapf"
This reverts commit b877e121bb
.
* Revert "revised lint errors in dataset_prepare.md"
This reverts commit 2289e27c
* Revert "checked with isort and yapf"
This reverts commit 159db2f8
* Revert "checked with isort and yapf"
This reverts commit 159db2f8
* add configs & fix bugs
* update new branch
* upload models&logs and add format-only
* change pretraied model path of HRNet
* fix the errors in dataset_prepare.md
* fix the errors in dataset_prepare.md and configs in loveda.py
* change the description in docs_zh-CN/dataset_prepare.md
* use init_cfg
* fix test converage
* adding pseudo loveda dataset
* adding pseudo loveda dataset
* adding pseudo loveda dataset
* adding pseudo loveda dataset
* adding pseudo loveda dataset
* adding pseudo loveda dataset
* Update docs/dataset_prepare.md
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* Update docs_zh-CN/dataset_prepare.md
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* Update docs_zh-CN/dataset_prepare.md
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* Delete unused lines of unittest and Add docs
* add convert .py file
* add downloading links from zenodo
* move place of LoveDA and Cityscapes in doc
* move place of LoveDA and Cityscapes in doc
Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-11-24 19:41:19 +08:00
MengzhangLI
14dc00af62
delete benchmark_new.py ( #1012 )
2021-11-03 23:27:48 +08:00
Junjun2016
54bd4bdd82
[Enhancement] Add codespell pre-commit hook and fix typos ( #920 )
...
* add codespell pre-commit hook and fix typos
* Update mmseg/models/decode_heads/dpt_head.py
* Update mmseg/models/backbones/vit.py
* Update mmseg/models/backbones/vit.py
* fix typos
* skip formating typo
* deprecate formating
* skip ipynb
* unstage ipynb changes
* unstage ipynb changes
* fix typos in ipynb
* unstage ipynb changes
2021-10-13 06:21:17 -07:00
VVsssssk
47052f7e75
[Update] Update torchserve about ( #951 )
...
* test_torchserver1.1
* test_torchserver1.2
* update
* update mmseg_handler.py
* update docs
* update torchserver
* tranfer torchserver to torchserve
* update docs
* updata torchserve support
2021-10-12 19:56:02 +08:00
MengzhangLI
d966f98f83
[Feature] Support ICNet ( #884 )
...
* add icnet backbone
* add icnet head
* add icnet configs
* nclass -> num_classes
* Support ICNet
* ICNet
* ICNet
* Add ICNeck
* Add ICNeck
* Add ICNeck
* Add ICNeck
* Adding unittest
* Uploading models & logs
* Uploading models & logs
* add comment
* smaller test_swin.py
* try to delete test_swin.py
* delete test_unet.py
* delete test_unet.py
* temp
* smaller test_unet.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-30 09:31:57 -07:00
Junjun2016
10886b00f0
fix load ckpt bug in swin ( #928 )
2021-09-30 07:50:44 -07:00
VVsssssk
0b11d58773
[Fix] fix_torchserver1.1 ( #844 )
...
* test_torchserver1.1
* test_torchserver1.2
* update
* update mmseg_handler.py
* update docs
* update torchserver
* tranfer torchserver to torchserve
* update docs
2021-09-23 19:14:52 +08:00
DerrickWang005
62f70ebedd
support coco stuff-10k/164k ( #625 )
...
* 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: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-22 20:48:08 +08:00
sennnnn
cae715a4b6
[Fix] Convert SyncBN to BN when training on DP ( #772 )
...
* [Fix] Convert SyncBN to BN when training on DP.
* Modify SyncBN2BN.
* Add SyncBN2BN unit test.
* Resolve some comments.
* use mmcv official revert_sync_batchnorm
* Remove local syncbn2bn unit tests.
* Update mmcv version.
* Fix bugs of gather model tools.
* Modify warnings.
* Modify docker mmcv version.
* Update mmcv version table.
2021-09-15 09:39:37 -07:00
FreyWang
c4c2fdcff3
[Fix] Fix iter bug when resuming checkpoint in distributed train ( #866 )
...
* [Fix] Fix iter bug when resuming checkpoint in distributed train
* fix lint error
Signed-off-by: FreyWang <wangwxyz@qq.com>
2021-09-11 14:33:01 +08:00
FreyWang
872e54497e
[Feature] Support eval concate dataset and add tool to show dataset ( #833 )
...
* [Feature] Add tool to show origin or augmented train data
* [Feature] Support eval concate dataset
* Add docstring and modify evaluate of concate dataset
Signed-off-by: FreyWang <wangwxyz@qq.com>
* format concat dataset in subfolder of imgfile_prefix
Signed-off-by: FreyWang <wangwxyz@qq.com>
* add unittest of concate dataset
Signed-off-by: FreyWang <wangwxyz@qq.com>
* update unittest for eval dataset with CLASSES is None
Signed-off-by: FreyWang <wangwxyz@qq.com>
* [FIX] bug of generator, which lead metric to nan when pre_eval=False
Signed-off-by: FreyWang <wangwxyz@qq.com>
* format code
Signed-off-by: FreyWang <wangwxyz@qq.com>
* add more unittest
* add more unittest
* optim concat dataset builder
2021-09-09 13:00:23 +08:00
sennnnn
d35fbbdb47
[Enhancement] Add Dev tools to boost develop ( #798 )
...
* Modify default work dir when training.
* Refactor gather_models.py.
* Add train and test matching list.
* Regression benchmark list.
* lower readme name to upper readme name.
* Add url check tool and model inference test tool.
* Modify tool name.
* Support duplicate mode of log json url check.
* Add regression benchmark evaluation automatic tool.
* Add train script generator.
* Only Support script running.
* Add evaluation results gather.
* Add exec Authority.
* Automatically make checkpoint root folder.
* Modify gather results save path.
* Coarse-grained train results gather tool.
* Complete benchmark train script.
* Make some little modifications.
* Fix checkpoint urls.
* Fix unet checkpoint urls.
* Fix fast scnn & fcn checkpoint url.
* Fix fast scnn checkpoint urls.
* Fix fast scnn url.
* Add differential results calculation.
* Add differential results of regression benchmark train results.
* Add an extra argument to select model.
* Update nonlocal_net & hrnet checkpoint url.
* Fix checkpoint url of hrnet and Fix some tta evaluation results and modify gather models tool.
* Modify fast scnn checkpoint url.
* Resolve new comments.
* Fix url check status code bug.
* Resolve some comments.
* Modify train scripts generator.
* Modify work_dir of regression benchmark results.
* model gather tool modification.
2021-09-02 09:44:51 -07:00
谢昕辰
c11da07e18
[Enhancement] Delete convert function and add instruction to ViT/Swin README.md ( #791 )
...
* delete convert function and add instruction to README.md
* unified model convert and README
* remove url
* fix import error
* fix unittest
* rename pretrain
* rename vit and deit pretrain
* Update upernet_deit-b16_512x512_160k_ade20k.py
* Update upernet_deit-b16_512x512_80k_ade20k.py
* Update upernet_deit-b16_ln_mln_512x512_160k_ade20k.py
* Update upernet_deit-b16_mln_512x512_160k_ade20k.py
* Update upernet_deit-s16_512x512_160k_ade20k.py
* Update upernet_deit-s16_512x512_80k_ade20k.py
* Update upernet_deit-s16_ln_mln_512x512_160k_ade20k.py
* Update upernet_deit-s16_mln_512x512_160k_ade20k.py
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-08-25 15:00:41 -07:00
sennnnn
e235c1a824
[Refactor] Support progressive test with fewer memory cost ( #709 )
...
* 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: Jiarui XU <xvjiarui0826@gmail.com>
2021-08-19 20:44:58 -07:00
谢昕辰
f72727c563
[Tools] Add vit/swin/mit convert weight scripts ( #783 )
...
* init scripts
* update markdown
* update markdown
* add docs
* delete mit converter and use torch load function
* rename segformer readme
* update doc
* modify doc
* 更新中文文档
* Update useful_tools.md
* Update useful_tools.md
* modify doc
* update segformer.yml
2021-08-17 18:42:42 -07:00
Junjun2016
441be4e435
[Dcos] Add header for files ( #796 )
...
* Add header for files
* Delete header in config files
2021-08-16 23:16:55 -07:00