Commit Graph

225 Commits (4686f1ec28093a8511fe6b32a36aa34b08e67f79)
 

Author SHA1 Message Date
quincylin1 27342fd709
add mmocr link (#501) 2021-04-21 16:08:59 -07:00
sennnnn 83d312e87a
Add support for Pascal Context 59 classes (#459)
* Add support for Pascal Context 59 classes (#459)

* Create PascalContextDataset59 class in mmseg/datasets/pascal_context.py;

* Set reduce_zero_label=True for train_pipeline and PascalContextDataset59;

* Add some configs for Pascal-Context 59 classes training and testing;

* Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";

* Continue(1): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";

* ignore files and folders named tempxxx;

* Continue(2): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0";

* Modify the calculation of IoU;

* Modify the CLASSES order of PascalContextDataset;

* Add "fcn", "deeplabv3", "deeplabv3+", "pspnet" config file for model training based on PascalContextDataset59;

Add some ignore items in ".gitignore";

* fix the bug "test_cfg specified in both outer field and model field " of pspnet config file;

* * Clean unnecessary codes;

* Add weighs link, config link, log link and evaluation results about PascalContextDataset59 in README.md

* Add command line argument: "-p | --port", this arg can change the transmit port when you transmit data to distributed machine.

* * Remove rebundant config files;

* Remove "-p|--port" command argument;

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-04-19 21:20:54 -07:00
Ziyi Wu 1483720cc0
add BaseSegmentor import to segmentors/__init__.py (#495) 2021-04-19 08:52:42 -07:00
sshuair 8f37e558c9
add tool pytorch2torchscript (#469)
* add tool pytorch2torchscript

* fix the assert message for pytorch version.
2021-04-19 08:51:49 -07:00
Jerry Jiarui XU 67eee62609
Fix sem_fpn and unet README.md (#492) 2021-04-18 02:56:53 -07:00
sennnnn fb24bf54b6
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 789d1a142b
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 e0e985fa85
[Improvement] Add cfg link in modelzoo (#468) 2021-04-09 14:36:03 -07:00
Jerry Jiarui XU a897a34a95
[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 0df07ab11f Fixed Colaboratory Tutorial 2021-04-07 15:19:29 -07:00
sshuair 9524f37fe6
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 4345ee3d55
Bump to v0.12 (#455) 2021-04-03 21:02:56 -07:00
Jerry Jiarui XU b9f9cf6558
refactor test organization (#440)
* refactor test organization

* fixed se layer

* update mmcv uper bound
2021-03-30 17:55:09 -07:00
lizz 455d163217
Save base_channels for UNet (#450)
Signed-off-by: lizz <lizz@sensetime.com>
2021-03-30 09:52:08 -07:00
lizz 33e8357c62
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 f7a5d53a4b
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
谢昕辰 1fc3e374e2
[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
谢昕辰 e86a87f060
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
钟晓键 340132dcf4
correct the script for installing mmcv-full (#399) 2021-03-27 21:13:06 -07:00
David de la Iglesia Castro 6acd77e345
Add opacity option to show_result (#425) 2021-03-22 20:34:38 -07:00
谢昕辰 b81894636b
add plot_logs tool (#426)
* Support plot logs

* add plot log docs
2021-03-21 22:05:32 -07:00
谢昕辰 0c31afe9eb
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
谢昕辰 7e1b24dd32
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 d0a71c1509
Fix PhotoMetricDistortion docstring (#388) 2021-02-24 18:22:00 -08:00
Kai Chen 34427f9261
add readme of chinese version (#383) 2021-02-21 11:11:28 -08:00
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