Commit Graph

228 Commits (0cf838f2949537f6cf46972cf6aa86b781db0600)
 

Author SHA1 Message Date
Jerry Jiarui XU e5007e7491 [Bug fix] fixed fp16 inference (#497)
* fixed fp16

* update fps
2021-04-24 09:57:33 -07:00
谢昕辰 b379b5a5b3 support transformer backbone (#465)
* vit backbone

* fix lint

* add docstrings and fix pretrained pos_embed dim not match prob

* add unittest for vit

* fix lint

* add vit based fcn configs

* fix import error

* support multiple resolution input images

* upsample pos_embed at init_weights

* support resize pos_embed at evaluation

* fix training errors

* add more unitest code for vit backbone

* unitest for uncovered code

* add norm_eval unittest

* refactor _pos_embeding

* minor change

* change var name

* rafactor init_weight

* load weights after resize

* ignore 'module' in pretrain checkpoint

* add with_cp

* add with_cp

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-04-21 20:19:55 -07:00
sshuair 0d99ff9c35 more docs about tools pytorch2torchscript and print_config (#499)
* Description of arguments about tools pytorch2torchscript and print_config

* fix docs lint
2021-04-21 16:09:59 -07:00
quincylin1 b03640f223 add mmocr link (#501) 2021-04-21 16:08:59 -07:00
sennnnn 38d2e969dd 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 c4049bab3e add BaseSegmentor import to segmentors/__init__.py (#495) 2021-04-19 08:52:42 -07:00
sshuair c3d9642e2d 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 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