Commit Graph

300 Commits (07cd6c98e047a1b66e8b5ed002d68467d4fcde0a)
 

Author SHA1 Message Date
Rockey 07cd6c98e0 [Feature] Add focal loss (#1024)
* [Feature] add focal loss

* fix the bug of 'non' reduction type

* refine the implementation

* add class_weight and ignore_index; support different alpha values for different classes

* fixed some bugs

* fix bugs

* add comments

* modify test

* Update mmseg/models/losses/focal_loss.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>

* update test_focal_loss.py

* modified the implementation

* Update mmseg/models/losses/focal_loss.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>

* update focal_loss.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-12-03 00:18:40 -08:00
MengzhangLI 504a5c6bd2 [Doc] add MMHuman3D link #1101 2021-12-02 21:35:49 -08:00
MengzhangLI 6a2cfea73b [Feature] Support Real-time model ERFNet (#960)
* first commit

* Fixing Unittest Error

* first refactory of ERFNet

* Refactorying NonBottleneck1d Module

* uploading models&logs

* uploading models&logs

* fix partial bugs & typos

* ERFNet

* add ERFNet with FCNHead

* fix typos of ERFNet

* add name on README.md cover

* chane name to T-ITS'2017

* fix lint error
2021-12-01 23:54:39 -08:00
MengzhangLI defb21bf8a [Fix] Change the upper version of mmcv to 1.5.0 #1096 2021-12-01 21:16:23 +08:00
MengzhangLI 5f99f7d0f1 change LoveDA codalab server address (#1095) 2021-12-01 13:29:05 +08:00
MengzhangLI 78fbaaf179 0~ (#1090) 2021-11-30 23:03:50 +08:00
MengzhangLI 1cf6049758 change (#1083) 2021-11-30 20:54:25 +08:00
Kyungmin Lee 40b9ebb565 [Feature] Add Cutout transform (#1022)
* Fix typo in usage example

* [Feature] Add CutOut transform

* CutOut repr covered by unittests

* Cutout ignore index, test

* ignore_index -> seg_fill_in, defualt is None

* seg_fill_in is added to repr

* test is modified for seg_fill_in is None

* seg_fill_in (int), 0-255

* add seg_fill_in test

* doc string for seg_fill_in

* rename CutOut to RandomCutOut, add prob

* Add unittest when cutout is False
2021-11-30 20:37:06 +08:00
MengzhangLI c4a52a523c [Docs] Add abstract and image for every paper (#1060)
* add abstract and main figure

* add abstract and main figure
2021-11-30 20:34:45 +08:00
MengzhangLI ae8d5ff0d4 [Enhancement] CircleCI Setup (#1086)
* uploading circle-ci

* uploading circle-ci

* uploading circle-ci

* correct circleci folder

* change mmsegmentation to mmseg in config.yml

* adding timm package in circleci

* adding timm package in circleci

* fix original lint error
2021-11-30 00:25:35 -08:00
del-zhenwu 621114fa39 Skip CI on ignoring given paths (#1078)
* Skip CI on ignoring given paths

Signed-off-by: del-zhenwu <dele.zhenwu@gmail.com>

* Fix ignoring path with using

Signed-off-by: del-zhenwu <dele.zhenwu@gmail.com>
2021-11-29 15:48:57 +08:00
Ebby 3525d35beb Update get_started.md (#1066)
Fix error in the chinese get_started file

完成的安装脚本 ==> 完整的安装脚本
2021-11-27 03:51:01 +08:00
Junjun2016 341ecc7633 [Enchance] Set a random seed when the user does not set a seed. (#1039) 2021-11-26 10:55:15 +08:00
MengzhangLI fdc054614c [Fix] Fix Dataset Display on README.md (#1072)
* fix dataset display on readme

* delete pytorch1.3.1

* change PyTorch 1.5.1 to 1.5or1.5.0

* change PyTorch 1.5.1 to 1.5.0

* change PyTorch 1.5.1 to 1.5.0

* fix cu102
2021-11-25 09:40:29 -08:00
Kingdrone 48d4222412 [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 7e5e34b217 add MMFewShot on README (#1065) 2021-11-24 19:40:00 +08:00
MengzhangLI 765a3a117a add mmflow on README (#1052) 2021-11-18 21:57:30 +08:00
MengzhangLI b692ab740c [Benchmark] Add BiSeNetV1 COCO-Stuff 164k benchmark (#1019)
* bisenetv1 on cocostuff164k

* change config_names & delete redundant keys

* pretrain should before lr.

* remove redundancy in bisenetv1_r50-d32
2021-11-17 00:12:02 -08:00
MengzhangLI 6d88b07084 fixing dice metric in unet (#1041) 2021-11-16 20:14:17 +08:00
MengzhangLI 008856a84c [Fix] Remove `fp16` folder in `configs`. (#1031)
* remove fp16 folder

* remove fp16 in docs

* fix some typos

* fix some typos

* fix fp16 in yml
2021-11-15 19:14:57 +08:00
Junjun2016 d4ec8ef963 fix EfficientMultiheadAttention in SegFormer (#1037) 2021-11-15 04:33:28 +08:00
Rockey bc27f24109 [Fix] Fix the bug that vit cannot load pretrain properly when using i… (#999)
* [Fix] Fix the bug that vit cannot load pretrain properly when using init_cfg to specify the pretrain scheme

* [Fix] fix the coverage problem

* Update mmseg/models/backbones/vit.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>

* [Fix] make the predicate more concise and clearer

* [Fix] Modified the judgement logic

* Update tests/test_models/test_backbones/test_vit.py

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>

* add comments

Co-authored-by: Junjun2016 <hejunjun@sjtu.edu.cn>
2021-11-03 10:36:09 -07:00
MengzhangLI f013682ea0 delete benchmark_new.py (#1012) 2021-11-03 23:27:48 +08:00
Junjun2016 28627a1288 Bump v0.19.0 (#1009)
* change version to v0.19.0

* update changelog
2021-11-03 05:27:20 +08:00
Junjun2016 0a06853bb6 [Feature] Support TIMMBackbone (#998)
* add TIMMBackbone and unittests

* add timm to tests requirements

* deprecate pt1.3.1

* reduce the unittests input of timm backbone

* fix ci

* fix ci

* fix ci

* fix ci

* fix ci

* fix ci

* fix ci

* fix ci

* fix ci

* remove unittests of large models of timm backbone

* generate coverage report for all unittests env

* reduce the unittests input of timm backbone

* reduce the unittests input of timm backbone
2021-11-02 12:51:11 +08:00
MengzhangLI 4ea92ebbe6 smaller input & channels of unittest (#1004) 2021-11-01 07:47:43 -07:00
MengzhangLI 992d577783 [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
gszh d7f82e5dc8 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
lkm2835 4b69af7b13 Fix typo in usage example (#1003) 2021-10-29 01:06:49 +08:00
Yuan Haobo a33cbbf039 vit contiguous (#992) 2021-10-25 18:11:48 +08:00
Rockey b3d5cc3f17 [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
Junjun2016 32f153e272 [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
MengzhangLI d8e7cfae21 [Benchmark] Uploading FastFCN on ADE20K (#972)
* Uploading FastFCN on ADE20K

* fixing lint error
2021-10-19 20:27:33 -07:00
Junjun2016 67f1420472 [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
MengzhangLI 279b7e5d25 Change BiSeNetV2 to Method (#945) 2021-10-12 20:35:58 -07:00
MengzhangLI d81554a642 first commit (#946) 2021-10-12 20:32:41 -07:00
Rockey 7aa2680d16 fix the wrong description in the Chinese documentation (#954) 2021-10-12 20:39:02 +08:00
VVsssssk 3c6cbc4a24 [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
Shouping Shan adb1cd361b [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
Junjun2016 1ce4904fe3 Bump to v0.18.0 (#940)
* bump to v0.18.0

* replace \ with /
2021-10-07 17:37:31 +08:00
VoyagerXVoyagerXisavailable fcc1c3b166 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
MengzhangLI a9d1295bc3 [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
MengzhangLI 7db1cbb181 [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 84edf6c190 fix load ckpt bug in swin (#928) 2021-09-30 07:50:44 -07:00
MengzhangLI 0cf56d48a4 change name to EncNet (#929) 2021-09-30 22:19:23 +08:00
Jerry Jiarui XU 7edc71d532 [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
MengzhangLI e701497a36 [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
sennnnn 2800d43507 [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
Junjun2016 d96937a003 [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
MengzhangLI f82e4d6fc9 [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