Commit Graph

185 Commits (420783d007d98dd967b5ba01a672637b946f557e)
 

Author SHA1 Message Date
Junjun2016 5dacca3ea8
Add 4 retinal vessel segmentation benchmark (#315)
* add 4 retinal vessel segmentation configs of UNet

* fix flip augmentation

* add unet benchmark on 4 medical datasets

* fix hrf bug
2020-12-23 23:58:09 -08:00
Junjun2016 94e12e8d21
Support DMNet (#313)
* Support DMNet

* fix doc and delete norm_name
2020-12-23 22:16:34 -08:00
Jerry Jiarui XU feefc6a9de
Doc refactor (#311)
* refactor docs

* add docs

* add modelzoo

* refactor getting started
2020-12-22 18:36:49 -08:00
Junjun2016 e3f6f655d6
Support APCNet (#299)
* Support APCNet

* code optimization

* add apcnet configs

* add benchmark

* add readme and model zoo

* fix doc
2020-12-17 23:23:45 -08:00
Jerry Jiarui XU 5c6e65759f
update CI for pip 20.3 (#307)
* update CI for pip 20.3

* fixed 1.6 torchvision

* delete python 3.8

* fixed mmcv

* add cuda home

* change to 1.2.2

* add pip upgrade

* install cuda for all

* add missing mmcv

* switch to deprecate
2020-12-17 22:46:38 -08:00
Junjun2016 061a2955dd
Support resize data augmentation according to original image size (#291)
* Support resize data augmentation according to original image size (img_scale=None and retio_range is tuple)

* fix docstring

* fix bug

* add unittest

* img_scale=None in TTA

* fix bug

* add unittest

* fix typos

* fix bug
2020-12-14 20:23:18 -08:00
Youqing Xiaozhua 7970e0ff11
Bugfix: num of augmentations and image meta not match when run TTA on CPU (#276)
* add inference test

* fix E501 line too long (81 > 79 characters

* fix wrong config path

* fix num of augmentations (2) != num of image meta (1)

* Update test_inference.py

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-12-13 23:51:42 -08:00
yinchimaoliang 1fb54ac8c0
change 'reduct' to 'reduce' (#298) 2020-12-13 10:06:01 -08:00
Jerry Jiarui XU 90465128ce
Bump to 0.9.0 (#285)
* Bump to 0.9.0

* add version
2020-12-01 21:10:55 -08:00
yamengxi 0066ce884f
[Feature]add CLAHE transform (#229)
* add CLAHE transform

* fix syntax error

* fix syntax error

* restore

* add a test

* modify cv2 to mmcv

* add docstring

* modify

* restore

* fix mmcv.clahe error

* change mmcv version to 1.3.0

* fix bugs

* add all data transformers to __init__

* fix __init__

* fix test_transform
2020-12-01 21:08:16 -08:00
yamengxi 4dc809adf2
[Feature] add AdjustGamma transform (#232)
* add AdjustGamma transform

* restore

* change cv2 to mmcv

* simplify AdjustGamma

* fix syntax error

* modify

* fix syntax error

* change mmcv version to 1.3.0

* fix lut function name error

* fix syntax error

* fix range
2020-12-01 20:14:01 -08:00
Junjun2016 993be2523b
add dice evaluation metric (#225)
* add dice evaluation metric

* add dice evaluation metric

* add dice evaluation metric

* support 2 metrics

* support 2 metrics

* support 2 metrics

* support 2 metrics

* fix docstring

* use np.round once for all
2020-11-23 19:21:22 -08:00
Jintao Lin 90e8e38e84
chmod +x stat.py (#266)
* add modelzoo statistics readthedocs

* chmod stat.py
2020-11-22 20:44:32 -08:00
yamengxi b6f9f8e9ec
fix detail url (#267) 2020-11-23 11:46:22 +08:00
Jerry Jiarui XU 15cca882c9
fixed readocs (#264) 2020-11-18 19:49:45 -08:00
Jintao Lin f31e0e930d
[docs] add modelzoo statistics readthedocs (#263)
* add modelzoo statistics readthedocs

* fix
2020-11-18 15:58:02 -08:00
Jerry Jiarui XU 8d568fbc0d
[Bug Fix] Add missing transforms in __init__.py (#260) 2020-11-18 12:22:46 +08:00
Jerry Jiarui XU 79d03b1597
[Bug Fix] Fix dropout_ratio typo (#261) 2020-11-18 12:22:06 +08:00
Jerry Jiarui XU 61e1d5c814
[Enhancement] Support ignore_index for sigmoid BCE (#210)
* [Enhancement] Add args check for ignore_index

* Support ignore_index
2020-11-17 16:14:03 +08:00
Jinkun Cao c2608b212a
update test tutorial: --eval-options instead of --options to set extra options (#251) 2020-11-16 17:32:53 +08:00
Junjun2016 300f42d006
fix _resize_seg bug in Resize data transforms (#246) 2020-11-16 17:28:41 +08:00
yamengxi 2bd51e60fd
[Feature] Add Rerange transform (#228)
* add rerange transform

* restore

* delete rerange config

* delete rerange config hint

* add min < max assert

* restore
2020-11-10 02:30:07 -08:00
yamengxi 7c68bca595
[Feature] Add Rgb2Gray transform (#227)
* add transformer Rgb2Gray

* restore

* fix self.weights

* restore

* fix code

* restore

* fix syntax error

* restore
2020-11-08 23:23:34 -08:00
Jerry Jiarui XU 3d18775113
[Feature] Add RandomRotate transform (#215)
* add RandomRotate for transforms

* change rotation function to mmcv.imrotate

* refactor

* add unittest

* fixed test

* fixed docstring

* fixed test

* add more test

* fixed repr

* rename to prob

* fixed unittest

Co-authored-by: hkzhang95 <GodBlessZhk@outlook.com>
2020-11-07 09:53:12 -08:00
Jerry Jiarui XU 0d1092105f
Bump to v0.8 (#226)
* Bump to v0.8

* add version
2020-11-04 16:18:02 -08:00
yamengxi f5eb7f2627
[New model] Support CGNet (#223)
* added cgnet

* added testing for cgnet

* git test

* add cgnet

* fix __init__

* rename FGlo with GlobalContextExtractor

* add readme.md and rename bn with norm

* delete cg_head

* fix a language mistake

* rename cgnet_m3n21.py to cgnet.py

* modify README.md

* modify list to tuple

* add fcn_head test

* add assert to fcn_head

* blank

* fix fcn_head assert bug

* add

* add cgnet to README.md and model_zoo.md

* modify cgnet README.md

Co-authored-by: KID <wps_@mail.nankai.edu.cn>
2020-11-03 00:05:25 -08:00
yamengxi 8a174a3511
fix dataset jpg bug (#213)
* fix dataset jpg bug

* fix syntax error
2020-10-28 11:58:44 -07:00
David de la Iglesia Castro e384ef578a
Add runner type (#118)
* Add runner_type option

* pre-commit

* Fix max_iters

* Add by_epoch to EvalHook

* Add test_eval_hook for epoch runner

* Remove runner-type arg from tools/train

* Add missing every_n_iters check for epoch mode

* Bump mmcv min version

* Use build_runner

* Use interval in tests

* Update test_eval_hook.py

* Use every_n_epochs instead of every_n_iters. Update DistEvalHook

* Add test_dist_eval_hook_epoch

* Fix tests

* Add DeprecationWarning

* Update docs

* Replace DeprecationWarning with UserWarning
2020-10-25 15:18:46 -07:00
yamengxi 3bdc276888
Define blood vessel dataset and fix filename bug (#203)
* fix filename bug

* define blood vessel dataset

* redo debug

* fix small bug

* rename dataset

Co-authored-by: yamengxi <yamengxi@sensetime.com>
2020-10-22 17:27:40 -07:00
Junjun2016 651da35c0e
add unet (#161)
* add unet

* add unet

* add unet

* update test_unet

* update test_unet

* update test_unet

* update test_unet

* fix bugs

* add init method for unet

* add test of UNet init_weights method

* add registry

* merge upsample

* fix test

* Update mmseg/models/backbones/unet.py

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

* Update mmseg/models/backbones/unet.py

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

* split UpConvBlock from UNet

* use reversed

* rename upsample module

* rename upsample module

* rename upsample module

* rename upsample module

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-10-21 11:24:38 -07:00
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