Commit Graph

100 Commits (0448dec5e4690ced25f8ee63e1a6b0f83f84cac8)
 

Author SHA1 Message Date
Jerry Jiarui XU 0448dec5e4
Bump v0.11 (#368)
* bump to v0.11

* update readme
2021-02-02 15:09:20 -08:00
Jerry Jiarui XU 6879b0d43c
[Bug fix] Fixed ADE20k test (#359)
* [Bug fix] Fixed ADE20k test

* fixed ade cfg
2021-01-24 02:17:59 -08:00
Jerry Jiarui XU faaf29e668
[Improvement] Move train_cfg/test_cfg inside model (#341)
* [Improvement] Move train_cfg/test_cfg inside model

* fixed config dict

* fixed config dict
2021-01-19 17:06:23 -08:00
yamengxi 0ee3016558
[New Feature]add lovasz loss (#351)
* add lovasz loss

* Modify as comments

* Modify paper url

* add unittest and remove Var

* impove unittest
2021-01-18 19:05:30 -08:00
congee 1c96a89472
update stat to classify papers (#348) 2021-01-13 22:20:02 -08:00
Junjun2016 49515e7e3c
Add more UNet-based medical segmentation benchmark (#324)
* Add UNet as backbone and FCN PSPNet DeepLabV3 as decode_head benchmark on 4 retinal vessel segmentation datasets

* adjust README of UNet
2021-01-11 00:07:59 -08:00
Jerry Jiarui XU da0af50812
[Improvement] Add tags for each README.md (#340) 2021-01-10 21:35:09 -08:00
Jerry Jiarui XU 30d0321895
[Improvement] Add OpenMMLab projects (#339) 2021-01-10 18:25:39 -08:00
yamengxi ce46d70d20
memory efficient test (#330)
* memory efficient test

* implement efficient test

* merge

* Add document and docstring

* fix unit test

* add memory usage report
2021-01-09 23:47:31 -08:00
congee 8ed47abd23
remove the model with 0 ckpts and exclude the papers without proposing any model (#336) 2021-01-08 02:00:30 -08:00
yamengxi 77e8ce3304
[Bug Fix] Fix TTA resize scale (#334)
* fix tta bug

* modify as suggested

* fix test_tta bug
2021-01-07 09:58:34 -08:00
Jerry Jiarui XU 8f8e77df96
Fix README.md (#329) 2021-01-04 23:52:40 -08:00
Jerry Jiarui XU 9f071cade8
Bump to v0.10 (#325)
* Bump to v0.10

* change version
2021-01-02 15:29:56 -08:00
Jerry Jiarui XU 2c921363d5
Add more models (#316)
* add more configs

* add more configs

* fixed backbone type

* fixed deeplabv3+ channels

* add r101

* update link

* change resnet18 link

* update aug test

* add inf time

* add mem
2020-12-29 17:56:06 -08:00
yamengxi 7fdb4002fa
[New model] Support MobileNetV3 (#268)
* delete markdownlint

* Support MobileNetV3

* fix import

* add mobilenetv3 head and configs

* Modify MobileNetV3 to semantic segmentation version

* modify mobilenetv3 configs

* add std configs

* fix Conv2dAdaptivePadding bug

* add configs

* add unitest and fix bugs

* fix lraspp unitest bugs

* restore

* fix unitest

* add MobileNetV3 docstring

* add mmcv

* add mmcv

* fix syntax bug

* fix unitest bug

* fix unitest bug

* fix unitest bugs

* fix docstring

* add configs

* restore

* delete unnecessary assert

* modify unitest

* delete benchmark
2020-12-26 00:02:50 -08:00
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