2e479d346b
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
b9ba9f6ce7
Support DMNet ( #313 )
...
* Support DMNet
* fix doc and delete norm_name
2020-12-23 22:16:34 -08:00
aba6d62c65
Doc refactor ( #311 )
...
* refactor docs
* add docs
* add modelzoo
* refactor getting started
2020-12-22 18:36:49 -08:00
a5d15ae228
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
7e918e302a
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
55df29beab
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
d5712c4d38
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
b77623e783
change 'reduct' to 'reduce' ( #298 )
2020-12-13 10:06:01 -08:00
7678ecc333
Bump to 0.9.0 ( #285 )
...
* Bump to 0.9.0
* add version
2020-12-01 21:10:55 -08:00
26f4bed242
[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
e8d643fe3a
[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
1530af6533
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
d8f780c403
chmod +x stat.py ( #266 )
...
* add modelzoo statistics readthedocs
* chmod stat.py
2020-11-22 20:44:32 -08:00
886db15751
fix detail url ( #267 )
2020-11-23 11:46:22 +08:00
2b80032fa6
fixed readocs ( #264 )
2020-11-18 19:49:45 -08:00
ad09482fa2
[docs] add modelzoo statistics readthedocs ( #263 )
...
* add modelzoo statistics readthedocs
* fix
2020-11-18 15:58:02 -08:00
97f10dbb04
[Bug Fix] Add missing transforms in __init__.py ( #260 )
2020-11-18 12:22:46 +08:00
a6234975f1
[Bug Fix] Fix dropout_ratio typo ( #261 )
2020-11-18 12:22:06 +08:00
ffcdfd36d0
[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
16fae6ef7f
update test tutorial: --eval-options instead of --options to set extra options ( #251 )
2020-11-16 17:32:53 +08:00
378ea27e4f
fix _resize_seg bug in Resize data transforms ( #246 )
2020-11-16 17:28:41 +08:00
365ce9e2af
[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
0588426eaa
[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
500babf958
[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
be94bbf1cc
Bump to v0.8 ( #226 )
...
* Bump to v0.8
* add version
2020-11-04 16:18:02 -08:00
86d473002f
[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
294a1f377a
fix dataset jpg bug ( #213 )
...
* fix dataset jpg bug
* fix syntax error
2020-10-28 11:58:44 -07:00
969c7fd8e4
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
464ffb77ad
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
5956451014
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
eaefe54e8d
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
a2e9d97096
add missing 0.7 ( #188 )
2020-10-11 18:08:40 +08:00
725a01f038
Bump to 0.7 ( #177 )
...
* Bump to 0.7
* update
2020-10-10 19:19:52 +08:00
420578a304
[Doc] Correct download link of ADE20K's test set ( #181 )
2020-10-10 19:18:27 +08:00
c13e1d5e05
[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
e7240c8cf1
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
93a2456def
Remove redundancies in pytorch2onnx ( #160 )
...
* rm redundancies
* re-add some packages
2020-09-30 10:11:34 +08:00
cc01f0d07a
[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
2de01c0c60
update docker ( #149 )
...
* update docker
* Update Dockerfile
2020-09-29 19:54:32 +08:00
2610a11981
[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
e86f4377b3
[Enhance] Update url to https://download.openmmlab.com ( #165 )
2020-09-28 00:32:44 +08:00
f3f443ff71
[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
efc5c20cd0
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
e385842557
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
51e4cdefc5
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
588a2c036a
add support for 4D output ( #150 )
2020-09-23 17:01:20 +08:00
7baed6513e
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
cac4138f99
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
276e9ca75e
[Enhance] Use MMCV collect_env ( #137 )
2020-09-19 21:32:41 +08:00
7c6fa48411
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
Junjun2016
Jerry Jiarui XU
Youqing Xiaozhua
yinchimaoliang
yamengxi
Jintao Lin
Jinkun Cao
David de la Iglesia Castro
LabMemNo003
Lei Yang
Mahmoud Zidan
robin Han
sshuair
Iago González