89 Commits

Author	SHA1	Message	Date
charlie880624	7716a0060f	feat: add golf dataset, kneron configs, and tools ... - Add golf1/2/4/7/8 dataset classes for semantic segmentation - Add kneron-specific configs (meconfig series, kn_stdc1_golf4class) - Organize scripts into tools/check/ and tools/kneron/ - Add kneron_preprocessing module - Update README with quick-start guide - Update .gitignore to exclude data dirs, onnx, nef outputs Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>	2026-03-18 13:14:30 +08:00
FangjianLin	288a6911ee	[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-05-05 22:09:25 +08:00
Cody Wong	93dd54136d	[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-05-05 22:09:25 +08:00
FangjianLin	319b897ea3	[Feature] Add multi machine dist_train. (#1383) ... * Add training startup documentation * fix * fix * fix * fix * fix * fix * fix * fix * fix	2022-05-05 22:09:25 +08:00
wangliujun	08ee236106	Update get_flops.py (#1371) ... * Update get_flops.py argparse.ArgumentParser(description='Get the FLOPs of a segmentor') * Update get_flops.py	2022-05-05 22:09:25 +08:00
Rockey	2313c64fb3	[Feature]: Add diff seeds to diff ranks and set torch seed in worker_init_fn (#1362)	2022-05-05 22:09:25 +08:00
Hyeokjoon Kwon	23738751bb	[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-05-05 22:09:25 +08:00
Miao Zheng	f2ecbe64be	Add deprecation message for deploy tool (#1321) ... * Add deprecation message for deploy tool * deploy test	2022-05-05 22:09:25 +08:00
Miao Zheng	54dea9ff70	[Enhancement] Revise pre-commit-hooks (#1315)	2022-05-05 22:09:25 +08:00
Waterman0524	2cada12660	[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-05-05 22:09:25 +08:00
Sebastian	bae9f23d57	Fix bug in conversion of potsdam (#1279)	2022-05-05 22:09:25 +08:00
chingning.chen	0136a5b2bd	style: fix format so pep8 is satisfied	2022-04-12 14:26:54 +08:00
chingning.chen	b324a669c7	style: pytorch2onnx_kneron.py line too long	2022-04-11 15:53:36 +08:00
EricChunYi	f02cdeae3d	use shorter kneron license line	2022-04-11 11:48:31 +08:00
EricChunYi	577de6d968	add kneron license for kneron modified code	2022-04-11 11:15:29 +08:00
chingning.chen	dcac233a60	fix: shape assignment for deploy_test_kneron and pytorch2onnx_kneron	2022-03-29 17:43:55 +08:00
chingning.chen	4d7356fda7	chore: print for clarified information in pytorch2onnx_kneron	2022-03-28 16:45:55 +08:00
chingning.chen	caaa31c231	chore: normalization-in-onnx as an argument in pytorch2onnx_kneron	2022-03-28 16:43:00 +08:00
chingning.chen	b6a868c3e2	feat: onnx auto bn for kneron quantization	2022-03-25 17:03:14 +08:00
chingning.chen	0ec0b33ed7	fix: pytorch 1.10 generates ONNX with ir_version==7, which should be 6	2022-03-25 16:03:14 +08:00
chingning.chen	5b99260c9b	feat: integrate kneron optimizer script to pytorch2onnx_kneron.py	2022-03-24 15:12:06 +08:00
chingning.chen	b2246495f5	refactor: not using MMDataParallel for onnx inferencing	2022-03-23 17:02:03 +08:00
chingning.chen	5846e789be	feat: deploy_test_kneron.py using ONNXRuntimeSegmentorKN	2022-03-21 17:04:29 +08:00
chingning.chen	6b27600a1a	feat: pytorch2onnx_kneron.py	2022-03-17 15:42:01 +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 b877e121bb2935ceefc503c09675019489829feb. * 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