Commit Graph

205 Commits (ee66167999212593451f9cbf0746c2b5b79cc0ab)
 

Author SHA1 Message Date
yamengxi cb2e0d3cb1
Add blood vessel dataset processing script (#184)
* Add blood vessel dataset processing script

* Fix syntax error

* Fix syntax error

* Fix syntax error

* Fix bugs

* Fix bugs

* Fix bugs

* Use safe functions and expand more apis

* Use safe functions and expand more apis

* Fix hard code and verify dataset integrity
2020-10-18 01:22:42 -07:00
Jerry Jiarui XU 5a76a715f1
add missing 0.7 (#188) 2020-10-11 18:08:40 +08:00
Jerry Jiarui XU 28f52d9d3d
Bump to 0.7 (#177)
* Bump to 0.7

* update
2020-10-10 19:19:52 +08:00
LabMemNo003 ec4b4783f1
[Doc] Correct download link of ADE20K's test set (#181) 2020-10-10 19:18:27 +08:00
Jerry Jiarui XU 0264de0bb0
[Improvement] Add markdown linter and fix linting errors (#171)
* [Improvement] Add markdown linter and fix linting errors

* fixed pip
2020-10-07 19:50:16 +08:00
David de la Iglesia Castro 118fd9ddf0
Support custom palette (#157)
* Fix split

* Update tests/test_data/test_dataset.py

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

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-30 18:02:08 +08:00
Lei Yang f70507168f
Remove redundancies in pytorch2onnx (#160)
* rm redundancies

* re-add some packages
2020-09-30 10:11:34 +08:00
Jerry Jiarui XU e1f4f51dc6
[Feature] Add OS16 DeepLab (#154)
* Add D16-MG124 models

* Use MMCV DepthSepConv

* add OHEM

* add warmup

* fixed test

* fixed test

* change to bs 16

* revert config

* add models
2020-09-29 20:00:48 +08:00
Jerry Jiarui XU 65da3e3752
update docker (#149)
* update docker

* Update Dockerfile
2020-09-29 19:54:32 +08:00
Jerry Jiarui XU f86c24d806
[Enhance] Refactor inverted residual (#164)
* [Enhance] Unifed InvertedResidual in MobileNetV2 and FastSCNN

* [Enhance] Unifed InvertedResidual in MobileNetV2 and FastSCNN
2020-09-28 00:33:51 +08:00
Jerry Jiarui XU 924571eced
[Enhance] Update url to https://download.openmmlab.com (#165) 2020-09-28 00:32:44 +08:00
Jerry Jiarui XU f95079ba8a
[Enhance] Migrate to MMCV DepthwiseSeparableConv (#158)
* Add D16-MG124 models

* Use MMCV DepthSepConv

* add OHEM

* add warmup

* fixed test

* fixed test

* change to bs 16

* revert config

* add models

* seperate
2020-09-25 19:56:10 +08:00
Mahmoud Zidan b9fd8e4bea
adding mobilenetv2 to docs/model_zoo.md and readme.md (#146)
* updating the readme with mobilenet_v2 backbone

* adding mobilenetv2 to model_zoo.md

* placing mobilenetv2 after pointrend
2020-09-25 17:44:34 +08:00
David de la Iglesia Castro 7f8bc7935c
Fix cpu inference (#152)
* Add missing map_location

* Add docstring

* Update mmseg/apis/inference.py

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

* Update inference.py

* Update inference.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-25 01:34:40 +08:00
David de la Iglesia Castro 746c8d3785
Use img_prefix and seg_prefix for loading (#153)
* Use img_prefix and seg_prefix for loading

* flake8

* Fix split
2020-09-25 00:48:16 +08:00
robin Han 08f30ea497
add support for 4D output (#150) 2020-09-23 17:01:20 +08:00
yamengxi a2738fd9be
Add Pascal Context to mmsegmentation (#133)
* Add Pascal Context to mmsegmentation

* Add benchmark result to Pascal Context

* fix mmcv version

* fix code syntax

* fix code syntax again

* Update mmseg/models/segmentors/encoder_decoder.py

update hint

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

* update comment

* fix pascal context model path

* fix model path mistake again

* fix model path mistake again

* fix model path mistakes again

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-22 14:56:13 +08:00
sshuair 768c3ee650
fix acc and iou compute nan problem (#116)
* fix acc and iou compute nan problem

* fix acc and iou compute nan problem

* add nan_to_num args for mean_iou

* add nan_to_num args for mean_iou

* add nan_to_num args for mean_iou

* add nan_to_num args for mean_iou

* add nan_to_num args for mean_iou

* Update mmseg/core/evaluation/mean_iou.py

* Update mean_iou.py

* Update mean_iou.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-22 01:04:46 +08:00
Jerry Jiarui XU e3f6d575ef
[Enhance] Use MMCV collect_env (#137) 2020-09-19 21:32:41 +08:00
Iago González b8c2f913e4
Add support for custom classes (#71)
* Support for custom classes

* Fix test

* Fix pre-commit

* Add pipeline logic for custom classes

* Fix minor issues, fix test

* Fix issues from PR review

* Fix tests

* Remove palette as str

* Rename old_to_new_ids to label_map

* Test for load_anns

* Remove get_palette function

* fixed temp

* Add subset of palette, remove palette as arg

* minor update

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-09-16 21:33:01 +08:00
Jerry Jiarui XU c3e4dbc2e0
Bump to version 0.6 (#119)
* Bump to version 0.6

* fixed format
2020-09-10 20:57:18 +08:00
Jerry Jiarui XU ff98229a3d
[Feature] Support PointRend (#109)
* [Feature] Support PointRend

* add previous test

* update modelzoo
2020-09-07 19:59:44 +08:00
Hongkai Zhang e807773a64
Fix typo (#110) 2020-09-07 17:02:00 +08:00
Han Hu 8ff866d60c
Add "disentangled non-local (DNL) neural networks" [ECCV2020] (#37)
* Add DNLHead

* add configs

* add weight decay mult

* add norm back

* Update README.md

* matched inference performance

* Fixed shape

* sep conv_out

* no norm

* add norm back

* complete model zoo

* add tests

* Add test forward

* Add more test

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-09-07 16:22:00 +08:00
Xia Li 李夏 dbca8b44a9
[Feature] Support EMANet (#34)
* add emanet

* fixed bug and typos

* add emanet config

* fixed padding

* fixed identity

* rename

* rename

* add concat_input

* fallback to update last

* Fixed concat

* update EMANet

* Add tests

* remove self-implement norm

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-09-07 13:06:59 +08:00
Jerry Jiarui XU 3c6dd9e6a4
[Feature] Support MobileNetV2 backbone (#86)
* [Feature] Support MobileNetV2 backbone

* Fixed import

* Fixed test

* Fixed test

* Fixed dilate

* upload model

* update table

* update table

* update bibtex

* update MMCV requirement
2020-09-04 15:35:52 +08:00
robin Han 0c04f52c42
Onnx upsample (#100)
* add customized Upsample which can convert to ONNX

* support multiply decode head for hrnet

* support size for Upsample
2020-09-03 19:59:13 +08:00
Jerry Jiarui XU b8f42c70fa
Add Semantic FPN (#94)
* Add Semantic FPN

* remove HRFPN
2020-09-03 19:56:36 +08:00
John Zhu 597b8a61c7
Windows Support (Experimental) (#75)
* Windows basic support

* getting_started updated for Windows.

* add experimental

* install.md restructured to seperate Windows & Linux.

* fix problems in install.md

* fix mmcv version problem.

* Fix fastscnn resize problems. (#82)

* Fix fast_scnn resize problems

* Fix fast_scnn resize problems 1

* Fix fast_scnn resize problems 2

* test for pascal voc

* [Doc] Add annotaion format note (#77)

* update pytorch version to 1.6.0 in install.md

* del fastscnn_pascal config

* del create_symlink=True

* Merge instructions for Linux & Windows

* mmcv version updated

* redundant newline deleted

* Update docs/install.md

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

* Update docs/install.md

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

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-08-28 11:34:44 +08:00
MY_ 298124566b
Update test.py (#93) 2020-08-25 21:12:39 +08:00
Jerry Jiarui XU bafc0e5db6
Fixed slide inference (#90) 2020-08-25 20:01:01 +08:00
Jerry Jiarui XU 03ba9c6c26
[Doc] Add annotaion format note (#77) 2020-08-23 14:42:07 +08:00
John Zhu e951076bf2
Fix fastscnn resize problems. (#82)
* Fix fast_scnn resize problems

* Fix fast_scnn resize problems 1

* Fix fast_scnn resize problems 2

* test for pascal voc
2020-08-23 14:41:30 +08:00
John Zhu 3e49d0ad71
Sorry for forgetting to update README.md. (#74) 2020-08-19 15:52:46 +08:00
John Zhu c1f46a69f4
Fast-SCNN implemented (#58)
* init commit: fast_scnn

* 247917iters

* 4x8_80k

* configs placed in configs_unify.  4x8_80k exp.running.

* mmseg/utils/collect_env.py modified to support Windows

* study on lr

* bug in configs_unify/***/cityscapes.py fixed.

* lr0.08_100k

* lr_power changed to 1.2

* log_config by_epoch set to False.

* lr1.2

* doc strings added

* add fast_scnn backbone  test

* 80k 0.08,0.12

* add 450k

* fast_scnn test: fix BN bug.

* Add different config files into configs/

* .gitignore recovered.

* configs_unify del

* .gitignore recovered.

* delete sub-optimal config files of fast-scnn

* Code style improved.

* add docstrings to component modules of fast-scnn

* relevant files modified according to Jerry's instructions

* relevant files modified according to Jerry's instructions

* lint problems fixed.

* fast_scnn config extremely simplified.

* InvertedResidual

* fixed padding problems

* add unit test for inverted_residual

* add unit test for inverted_residual: debug 0

* add unit test for inverted_residual: debug 1

* add unit test for inverted_residual: debug 2

* add unit test for inverted_residual: debug 3

* add unit test for sep_fcn_head: debug 0

* add unit test for sep_fcn_head: debug 1

* add unit test for sep_fcn_head: debug 2

* add unit test for sep_fcn_head: debug 3

* add unit test for sep_fcn_head: debug 4

* add unit test for sep_fcn_head: debug 5

* FastSCNN type(dwchannels) changed to tuple.

* t changed to expand_ratio.

* Spaces fixed.

* Update mmseg/models/backbones/fast_scnn.py

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

* Update mmseg/models/decode_heads/sep_fcn_head.py

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

* Update mmseg/models/decode_heads/sep_fcn_head.py

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

* Docstrings fixed.

* Docstrings fixed.

* Inverted Residual kept coherent with mmcl.

* Inverted Residual kept coherent with mmcl. Debug 0

* _make_layer parameters renamed.

* final commit

* Arg scale_factor deleted.

* Expand_ratio docstrings updated.

* final commit

* Readme for Fast-SCNN added.

* model-zoo.md modified.

* fast_scnn README updated.

* Move InvertedResidual module into mmseg/utils.

* test_inverted_residual module corrected.

* test_inverted_residual.py moved.

* encoder_decoder modified to avoid bugs when running PSPNet.
getting_started.md bug fixed.

* Revert "encoder_decoder modified to avoid bugs when running PSPNet. "

This reverts commit dd0aadfb

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-08-18 23:33:05 +08:00
Jerry Jiarui XU d9484068ce
Support ResNeSt backbone (#47)
* Support ResNeSt backbone

* fixed avg_down

* add docstring and test

* update table

* update docs and tests

* fixed test

* rename

* refactor splits
2020-08-17 00:54:01 +08:00
RainbowSecret 381eacb9a5
add more results of OCRNet (#20)
* update the HRNet-OCR & add ResNet-101-OCR

* revise the script

* add the results of resnet-101+ocr

* add cascade ocr, aspp ocr

* add comparison table

* move comparison table

* support ocr+decoder

* revise the ocrnet_sep_aspp

* update the results of ocrnet

* update the results of ocrnet

* add sep-ocr-variants

* add bs2x exp of deeplabv3/v3+

* apply sep-conv in ocr module

* update the results

* update the results

* update the results of OCRNet

* update the results of OCRNet

* correct the results

* verify the release branch

* init the release branch

* add more results of ocrnet and ocrnetplus

* resolve the conflicts

* rename OCRNetPlus as OCRNet+

* fix the format

* fix the lint issues

* fix the lint issues

* fix the lint issues

* fix the lint isort issues

* fix the lint yapf issues

* fix the format issues

* remove the changes by the master branch

* remove the changes by the master branch

* remove the changes by the master branch

* remove the changes by the master branch

* remove the changes by the master branch

* add the logs folder to .gitignore

* recover .gitignore

* update readme

* update readme

* reset the cudnn_benchmark

* revise the README of OCRNet

* revise the name

* revise the reference of OCRNet

* revise the Figure of OCRNet+

* update the results of OCR/OCR+

* update the results of OCR/OCR+

* update the results of OCR/OCR+

* fix the format issue

* fix the format issue

* remove the ocr+

* update the results

* update the results

* fix the conflicts

* fix the lint issue

* fix the lint issue

* fix the lint issue

* fix the lint issue

* fix the inconsistency

* add urls to README

* clean the code

* remove the schedule configs

* clean the custom code

* clean up

* remove ocr.png

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-14 13:16:27 +08:00
robin Han 9ba22456d9
add pytorch2onnx part (#12)
* add pytorch2onnx part

* Update according to the latest mmcv

* add docstring

* update docs

* update docs

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-14 03:28:21 +08:00
Jerry Jiarui XU 5a753811fb
Add pytorch 1.6 CI (#64) 2020-08-12 18:55:40 +08:00
Jerry Jiarui XU 4e29452dbe
Auto get version info and git hash (#55)
* Auto get version info and git hash

* bump 0.5.1 and update doc

* fixed docs

* Add change log
2020-08-11 19:23:35 +08:00
Jerry Jiarui XU 5e264c608c
Generalized OHEM (#54)
* Generalized OHEM

* remove config

* update docstring

* fixed sort prob

* fixed valid_mask
2020-08-09 23:49:23 +08:00
Evgeny Nizhibitsky 00f56eb00f
Add load-from flag (#33)
* Add load-from flag

* minor update

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-06 21:17:49 +08:00
Kai Chen 2b801dedfc
Update README.md (#38) 2020-07-27 19:02:08 +08:00
ycr 579f5f883e
Fix palette type. (#27) 2020-07-23 13:01:31 +08:00
Jerry Jiarui XU 1bedfe305b
Fixed training tricks (#26) 2020-07-20 15:17:59 +08:00
Jerry Jiarui XU 1765c12985
Support FP16 (#21)
* Support FP16

* add miss folder

* add tests

* remove useless config

* update memory

* reduce config

* migrate fp16 to mmcv

* add model link
2020-07-20 15:17:18 +08:00
Jerry Jiarui XU 1af2ad6a9f
Fixed voc aug convert (#19)
* Fixed voc aug convert

* update getting_started.md

* add class balanced doc
2020-07-17 09:55:28 +08:00
Jerry Jiarui XU 1c3f547659
Fixed long description (#16) 2020-07-14 19:35:25 +08:00
Jerry Jiarui XU 4d832e54dd
fixed name in pypi, add badge (#15)
* fixed name in pypi, add badge

* fixed name in install.md

* update install.md
2020-07-14 14:41:52 +08:00
Kai Chen 152879330b
Install wheel before building mmseg (#14) 2020-07-14 00:28:13 +08:00