349fc2d812
[Fix] Change `self.loss_decode` back to `dict` in Single Loss situation. ( #1002 )
...
* fix single loss type
* fix error in ohem & point_head
* fix coverage miss
* fix uncoverage error of PointHead loss
* fix coverage miss
* fix uncoverage error of PointHead loss
* nn.modules.container.ModuleList to nn.ModuleList
* more simple format
* merge unittest def
2021-11-01 15:28:37 +08:00
845098b155
Update train.py ( #428 )
...
* Update train.py
Add user-defined hooks.
* Update train.py
* Update train.py
2021-10-29 12:13:40 -07:00
9feaa7c1d2
Fix typo in usage example ( #1003 )
2021-10-29 01:06:49 +08:00
93f8b142a3
vit contiguous ( #992 )
2021-10-25 18:11:48 +08:00
e27017efa7
[Docs] Fix the broken link ( #985 )
...
* [Fix] Fix the broken link
* [Fix] Fix the broken link
* 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>
* [Docs] Fix the broken link
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-10-25 15:07:40 +08:00
5fdb7695ca
[Fix] Fix bug in CI with py3.9 ( #994 )
...
* remove python-dev in CI
* remove pycocotools installation
* remove python 3.9-dev installation dependencies
2021-10-24 11:49:07 -07:00
6d35d76195
[Benchmark] Uploading FastFCN on ADE20K ( #972 )
...
* Uploading FastFCN on ADE20K
* fixing lint error
2021-10-19 20:27:33 -07:00
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
d4d64eb766
Change BiSeNetV2 to Method ( #945 )
2021-10-12 20:35:58 -07:00
267602705b
first commit ( #946 )
2021-10-12 20:32:41 -07:00
c403177bad
fix the wrong description in the Chinese documentation ( #954 )
2021-10-12 20:39:02 +08:00
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
796d5edebe
[Fix] Fix bug when loading class name form file in custom dataset ( #923 )
...
* [Fix] #916 expection string type classes
* add unittests for string path classes
* fix double quote string in test_dataset.py
* move the import to the top of the file
* fix isort lint error
fix isort lint error when move the import to the top of the file
2021-10-08 01:06:18 +08:00
8d49dd31e4
Bump to v0.18.0 ( #940 )
...
* bump to v0.18.0
* replace \ with /
2021-10-07 17:37:31 +08:00
d7924baab4
replace the unavailable pspnet model download address with the update… ( #934 )
...
* replace the unavailable pspnet model download address with the updated address
* replace the unavailable pspnet model download address with the updated address
2021-10-03 23:55:25 +08:00
f910caf229
[Feature] Support FastFCN ( #885 )
...
* FastFCN first commit
* FastFCN first commit
* Fixing lint error
* Fixing lint error
* use for loop on JPU
* Use For Loop
* Refactor FastFCN
* FastFCN
* FastFCN
* temp
* Uploading models & logs (4x4)
* Fixing typos
* fix typos
* rename config
* change README.md
* use _delete_=True
* change configs
* change start_level to 0
* change start_level to 0
* jpu
* add unittest for start_level!=0
2021-10-01 02:41:24 +08:00
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
10886b00f0
fix load ckpt bug in swin ( #928 )
2021-09-30 07:50:44 -07:00
c1dcf91c1a
change name to EncNet ( #929 )
2021-09-30 22:19:23 +08:00
85227b46c7
[Improvement] Refactor Swin-Transformer ( #800 )
...
* [Improvement] Refactor Swin-Transformer
* fixed swin test
* update patch emebd, add more tests
* fixed test
* remove pretrain_style
* fixed padding
* resolve coments
* use mmcv 2tuple
* refactor init_cfg
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-29 08:46:33 +08:00
ab12009414
[Feature] Support BiSeNetV1 ( #851 )
...
* First Commit
* fix typos
* fix typos
* Fix assertion bug
* Adding Assert
* Adding Unittest
* Fixing typo
* Uploading models & logs
* Fixing unittest error
* changing README.md
* changing README.md
2021-09-29 02:12:57 +08:00
2aa632ebe7
[Enhancement] Change readme style and Update metafiles. ( #895 )
...
* [Enhancement] Change readme style and prepare for metafiles update.
* Update apcnet github repo url.
* add code snippet.
* split code snippet & official repo.
* update md2yml hook.
* Update metafiles.
* Add converted from attribute.
* process conflict.
* Put defualt variable value.
* update bisenet v2 metafile.
* checkout to ubuntu environment.
* pop empty attribute & make task attribute list.
* update readme style
* update readme style
* update metafiles
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-28 16:25:37 +08:00
36eb2d8beb
[Enhancement] Using docker to skip CUDA installation in CI ( #882 )
...
* Use docker to skip CUDA installation in CI
* Update .github/workflows/build.yml
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* Remove Install system dependencies
* Add Install system dependencies for cuda usage
* Add Install system dependencies and requirements
* fix requirements error
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-09-26 22:25:21 -07:00
4003b8f421
[Feature] Support BiSeNetV2 ( #804 )
...
* BiSeNetV2 first commit
* BiSeNetV2 unittest
* remove pytest
* add pytest module
* fix ConvModule input name
* fix pytest error
* fix unittest
* refactor
* BiSeNetV2 Refactory
* fix docstrings and add some small changes
* use_sigmoid=False
* fix potential bugs about upsampling
* Use ConvModule instead
* Use ConvModule instead
* fix typos
* fix typos
* fix typos
* discard nn.conv2d
* discard nn.conv2d
* discard nn.conv2d
* delete **kwargs
* uploading markdown and model
* final commit
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* BiSeNetV2 adding Unittest for its modules
* Fix README conflict
* Fix unittest problem
* Fix unittest problem
* BiSeNetV2
* Fixing fps
* Fixing typpos
* bisenetv2
2021-09-26 18:52:16 +08:00
29c82eaf13
[Fix] Fix loss parse in val_step ( #906 )
...
* [Fix] Fix loss parse in val_step
* Add val_step unittest
* Add train_step unittest
2021-09-25 18:17:40 -07:00
e171e806de
The path of the annotation is wrong in the tutorial ( #907 )
...
* Update config.md
* Update config.md
2021-09-26 00:07:41 +08:00
186a1fc88e
[Feature] Support multiple losses during training ( #818 )
...
* multiple losses
* fix lint error
* fix typos
* fix typos
* Adding Attribute
* Fixing loss_ prefix
* Fixing loss_ prefix
* Fixing loss_ prefix
* Add Same
* loss_name must has 'loss_' prefix
* Fix unittest
* Fix unittest
* Fix unittest
* Update mmseg/models/decode_heads/decode_head.py
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-09-24 15:08:28 +08:00
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
cff01b36c9
Fixing README ( #896 )
2021-09-22 17:19:19 -07:00
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
4e6f2ebced
[Docs] Use Chinese menu ( #880 )
...
* Use Chinese menu
* replace \ with /
2021-09-18 18:36:43 +08:00
4d8229c141
[Docs] Improve docs style ( #879 )
...
* Improve docs style
* update lists
* update the size of image
* modify duplicate mmdet3d
2021-09-16 08:23:50 -07:00
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
5a7996db26
[Enhancement] Support loading GT for evaluation from multi-file backend ( #867 )
...
* support load gt for evaluation from multi-backend
* move some code from get_gt_seg_maps to get_one_gt_seg_map
* rename gt_seg_map_loader_conf to gt_seg_map_loader_cfg
* fix doc str
* rename get_one_gt_seg_map to get_gt_seg_map_by_idx
2021-09-14 19:16:01 -07:00
56e18ba9aa
[Fix] Update mmcv installation in dockerfile ( #860 )
...
* update dockerfile
* Update docker/Dockerfile
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* add ARG
* change dash to bash
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-09-11 14:34:35 +08:00
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
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
eb0baee414
[Feature] Support ISA module ( #70 )
...
* add isa module
* use more readable names, add more comments and exp results
* add unittests
* remove redundant docstring
* Apply suggestions from code review
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
* fix unittest
* Update configs
* add results
* update yml
* Update README
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: xiexinch <xinchen.xie@qq.com>
2021-09-08 21:13:53 -07:00
c5b0e8ac82
[Enhancement] Upgrade CI for PyTorch1.9 ( #712 )
...
* Upgrade CI for PyTorch1.9
* fix typo
* merge build_cu101 and build_cu102 to build_cuda
* fix substitution bug
* test
* test
* test
* test
* test
* test
* test
* test
* test
* test
* test
* test
* remove redudant items
* add python 3.9 to CI
* add python 3.9 to setup.py
* remove some versions of pytorch for python 3.9 test
* fix torch version error
* delete mim files
* update ubuntu version
* fix mmcv cuda version
2021-09-08 18:43:17 -07:00
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
4981ff68c2
[Fix] Fix docstring link problem in readthedocs ( #845 )
...
* fix docstring link
* fix docstring link
* fix docstring link
* fix docstring link
* fix docstring link
* fix docstring link
2021-09-02 17:06:43 +08:00
a7461d96b0
Fix sphinx version ( #827 )
2021-09-01 18:45:53 -07:00
ea4f7cb4fb
Show supported python versions in README ( #826 )
2021-09-01 18:39:44 -07:00
3ff9b5377b
Add MMSeg citation ( #825 )
...
* fix typo
* add citation
2021-09-01 18:38:58 -07:00
122448010b
Bump to v0.17.0 ( #841 )
2021-09-01 22:43:38 +08:00
2825efe378
[Feature] add DPT head ( #605 )
...
* add DPT head
* [fix] fix init error
* use mmcv function
* delete code
* remove transpose clas
* support NLC output shape
* Delete post_process_layer.py
* add unittest and docstring
* rename variables
* fix project error and add unittest
* match dpt weights
* add configs
* fix vit pos_embed bug and dpt feature fusion bug
* match vit output
* fix gelu
* minor change
* update unitest
* fix configs error
* inference test
* remove auxilary
* use local pretrain
* update training results
* update yml
* update fps and memory test
* update doc
* update readme
* add yml
* update doc
* remove with_cp
* update config
* update docstring
* remove dpt-l
* add init_cfg and modify readme.md
* Update dpt_vit-b16.py
* zh-n README
* use constructor instead of build function
* prevent tensor being modified by ConvModule
* fix unittest
Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-08-30 16:53:05 +08:00
5753f413ad
[Feature] Support dark dataset test ( #815 )
...
* 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: MengzhangLI <mcmong@pku.edu.cn>
2021-08-28 11:51:05 -07:00
aa438f5c95
[Fix] The interface multiscale_output is defined but not used ( #830 )
...
* Add interface multiscale_output
* Add space between args and their types
* Fix default value
2021-08-27 18:53:55 +08:00
4ca42a3ee4
Update inference_demo.ipynb ( #814 )
...
fixed the link to *.pth broken link
2021-08-27 14:56:13 +08:00
4686f1ec28
[Enhancement] Support inherit cityscapes dataset ( #750 )
...
* rewrite init function
* init function parameters
2021-08-27 14:53:36 +08:00
MengzhangLI
gszh
lkm2835
Yuan Haobo
Rockey
Junjun2016
VVsssssk
Shouping Shan
VoyagerXVoyagerXisavailable
Jerry Jiarui XU
sennnnn
Julius Zhang
kira
DerrickWang005
uni19
FreyWang
Layne
谢昕辰
Guangchen Lin
Pedro Machado