Jerry Jiarui XU
2cb5e25183
Fix sem_fpn and unet README.md ( #492 )
2021-04-18 02:56:53 -07:00
sennnnn
d3603e5335
Fix mIoU calculatiton range ( #471 )
...
* Fix fence(IoU) = 0 when training on PascalContextDataset59;
* Add a test case in test_metrics() of tests/test_metrics.py to test the bug caused by torch.histc;
* Update tests/test_metrics.py
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-04-14 08:37:23 -07:00
q.yao
bc2dc1277a
add dynamic export and visualize to pytorch2onnx ( #463 )
...
* add dynamic export and visualize to pytorch2onnx
* update document
* fix lint
* fix dynamic error and add visualization
* fix lint
* update docstring
* update doc
* Update help info for --show
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* fix lint
Co-authored-by: maningsheng <maningsheng@sensetime.com>
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-04-12 11:54:59 -07:00
Jerry Jiarui XU
d003f661cc
[Improvement] Add cfg link in modelzoo ( #468 )
2021-04-09 14:36:03 -07:00
Jerry Jiarui XU
d283ca6fea
[Improvement] Add more CI ( #460 )
...
* add more CI
* fix missing
* remove python-version
* fixed acc
2021-04-08 19:29:00 -07:00
Jerry Jiarui XU
99ab9bd2d9
Fixed Colaboratory Tutorial
2021-04-07 15:19:29 -07:00
sshuair
fd486cdbc3
add print model graph args for tools/print_config.py ( #451 )
...
* add print model graph for print_config tool
* add print model graph for print_config tool
* fix double quoted
* fix iosort
2021-04-06 15:14:54 -07:00
Jerry Jiarui XU
ac2aab74e9
Bump to v0.12 ( #455 )
2021-04-03 21:02:56 -07:00
Jerry Jiarui XU
3150dd0ce4
refactor test organization ( #440 )
...
* refactor test organization
* fixed se layer
* update mmcv uper bound
2021-03-30 17:55:09 -07:00
lizz
fcad6df7a6
Save base_channels for UNet ( #450 )
...
Signed-off-by: lizz <lizz@sensetime.com>
2021-03-30 09:52:08 -07:00
lizz
bbb9f074f6
Fix typo: upsampe_cfg -> upsample_cfg ( #449 )
...
* Fix typo: upsampe_cfg -> upsample_cfg
Signed-off-by: lizz <lizz@sensetime.com>
* convoluton -> convolution
Signed-off-by: lizz <lizz@sensetime.com>
* more
Signed-off-by: lizz <lizz@sensetime.com>
* ok
Signed-off-by: lizz <lizz@sensetime.com>
2021-03-30 09:51:38 -07:00
sshuair
7c329fa775
Include each class metrics in logs ( #445 )
...
* Include each class metrics in logs
* format
* fix the unitest
* fix the custom int class_name
* fix the custom int class_name
2021-03-29 09:53:54 -07:00
谢昕辰
71be1c2793
[Bug fixed]Fix dice_loss errors ( #417 )
...
* fix training bugs
* fix unitest error
* fix error in num_classes==2 case
* delete comments
2021-03-29 09:49:54 -07:00
谢昕辰
d474cfde4b
pytorch metrics implementation ( #430 )
...
* pytorch metrics impl and test
* support list[str] input, delete unused test code and delete numpy version
* modify input data type
* add docstring and unitest of filename inputs
* add indents in docstring and use tempfile lib to create dir
* using with statement
2021-03-29 09:49:14 -07:00
钟晓键
15faf716de
correct the script for installing mmcv-full ( #399 )
2021-03-27 21:13:06 -07:00
David de la Iglesia Castro
9cbb4b1288
Add opacity option to show_result ( #425 )
2021-03-22 20:34:38 -07:00
谢昕辰
1722010396
add plot_logs tool ( #426 )
...
* Support plot logs
* add plot log docs
2021-03-21 22:05:32 -07:00
谢昕辰
bac0714347
Support fcn dilate 6 ( #400 )
...
* Support fcn dilate 6
* Support dilate in FCNHead
* configs for cityscapse dataset
* add configs for pytorch pretrained model
* update README
* add fps test results
* add memory test results and links
* modify log names
* Update mmseg/models/decode_heads/fcn_head.py
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-03-20 10:32:04 -07:00
谢昕辰
a3b523e3fd
dice loss ( #396 )
...
* dice loss
* format code, add docstring and calculate denominator without valid_mask
* minor change
* restore
2021-03-10 18:32:37 -08:00
Jerry Jiarui XU
96ddb41c6e
Fix PhotoMetricDistortion docstring ( #388 )
2021-02-24 18:22:00 -08:00
Kai Chen
67d71504a7
add readme of chinese version ( #383 )
2021-02-21 11:11:28 -08:00
Jerry Jiarui XU
45f07af4c8
Bump v0.11 ( #368 )
...
* bump to v0.11
* update readme
2021-02-02 15:09:20 -08:00
Jerry Jiarui XU
f3cf96b63e
[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
4423d32702
[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
c8bbd3fa95
[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
11471ad9bb
update stat to classify papers ( #348 )
2021-01-13 22:20:02 -08:00
Junjun2016
d56f6823a2
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
ce232deb09
[Improvement] Add tags for each README.md ( #340 )
2021-01-10 21:35:09 -08:00
Jerry Jiarui XU
c8e6a82efb
[Improvement] Add OpenMMLab projects ( #339 )
2021-01-10 18:25:39 -08:00
yamengxi
7e156454a2
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
7a6b1eba3d
remove the model with 0 ckpts and exclude the papers without proposing any model ( #336 )
2021-01-08 02:00:30 -08:00
yamengxi
022b055a66
[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
7c4e505e7d
Fix README.md ( #329 )
2021-01-04 23:52:40 -08:00
Jerry Jiarui XU
5eafe8f6df
Bump to v0.10 ( #325 )
...
* Bump to v0.10
* change version
2021-01-02 15:29:56 -08:00
Jerry Jiarui XU
95cd2700d7
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
25d8d77fab
[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
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
Junjun2016
b9ba9f6ce7
Support DMNet ( #313 )
...
* Support DMNet
* fix doc and delete norm_name
2020-12-23 22:16:34 -08:00
Jerry Jiarui XU
aba6d62c65
Doc refactor ( #311 )
...
* refactor docs
* add docs
* add modelzoo
* refactor getting started
2020-12-22 18:36:49 -08:00
Junjun2016
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
Jerry Jiarui XU
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
Junjun2016
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
Youqing Xiaozhua
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
yinchimaoliang
b77623e783
change 'reduct' to 'reduce' ( #298 )
2020-12-13 10:06:01 -08:00
Jerry Jiarui XU
7678ecc333
Bump to 0.9.0 ( #285 )
...
* Bump to 0.9.0
* add version
2020-12-01 21:10:55 -08:00
yamengxi
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
yamengxi
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
Junjun2016
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
Jintao Lin
d8f780c403
chmod +x stat.py ( #266 )
...
* add modelzoo statistics readthedocs
* chmod stat.py
2020-11-22 20:44:32 -08:00
yamengxi
886db15751
fix detail url ( #267 )
2020-11-23 11:46:22 +08:00