Commit Graph

154 Commits (5d463148444c3030bf68ba847520fbcf535dcfa2)
 

Author SHA1 Message Date
Jerry Jiarui XU 5d46314844
Bump to v0.14 (#580) 2021-06-02 18:43:08 -07:00
Yinhao Li 9dd3e15f83
[feature]: Able to use save_best option (#575)
* Add  save_best option in eval_hook.

* Update meta to fix best model can not test bug

* refactor with _do_evaluate

* remove redundent

* add meta

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-06-02 16:25:26 -07:00
谢昕辰 a95f6d8173
[Feature] support mim (#549)
* dice loss

* format code, add docstring and calculate denominator without valid_mask

* minor change

* restore

* add metafile

* add manifest.in and add config at setup.py

* add requirements

* modify manifest

* modify manifest

* Update MANIFEST.in

* add metafile

* add metadata

* fix typo

* Update metafile.yml

* Update metafile.yml

* minor change

* Update metafile.yml

* add subfix

* fix mmshow

* add more  metafile

* add config to model_zoo

* fix bug

* Update mminstall.txt

* [fix] Add models

* [Fix] Add collections

* [fix] Modify collection name

* [Fix] Set datasets to unet metafile

* [Fix] Modify collection names

* complement inference time
2021-05-31 15:07:24 -07:00
q.yao dc5d53b7f3
[Feature] Update deploy test tools (#553)
* add trt test tool

* create deploy_test, update document

* fix with isort

* move import inside __init__

* remove comment, fix doc

* update document
2021-05-24 20:37:46 -07:00
Edward 66b0525378
DeepLab V3 did not present in CVPR 2017 (#560)
https://openaccess.thecvf.com/CVPR2017 does not contain DeepLabV3
2021-05-18 09:44:41 -07:00
sennnnn 63a9516b94
Add compatible MMSegmentation and MMCV version table. (#558) 2021-05-17 10:43:13 -07:00
sennnnn 524425cf93
[Feature] Add results2img, format_results for ade dataset (#544)
* [Feature] Add results2img, format_results for ade dataset.

* clean rebundant code.
2021-05-16 19:29:28 -07:00
q.yao 8558163753
fix verify bugs (#547)
* fix verify bugs

* rename args
2021-05-12 21:14:41 -07:00
q.yao 805eda81ea
add onnx to tensorrt tools (#542) 2021-05-11 20:02:27 -07:00
谢昕辰 584f5a7def
support loading deit weights (#538) 2021-05-08 20:34:18 -07:00
Ziyi Wu db44d16e02
infer batch size using len(result) in test function (#532) 2021-05-06 08:16:46 -07:00
RangiLyu 4403923db1
fix typo (#533) 2021-05-06 08:16:06 -07:00
sennnnn aa9b609f11
Add option for output shape of ViT (#530)
* Add arg: final_reshape to control if converting output feature information from NLC to NCHW;

* Fix the default value of final_reshape;

* Modify arg: final_reshape to arg: out_shape;

* Fix some unit test bug;
2021-05-05 22:49:28 -07:00
Jerry Jiarui XU f884489120
Bump to v0.13 (#529) 2021-05-05 16:56:19 -07:00
谢昕辰 f9336032a5
Use MMCV's EvalHook in MMSegmentation (#438)
* mmcv eval hook

* mmcv evalhook compatible

* add warnings

* inherit from base class

* fix unitest

* adapt to mmcv 1.3.1

* fixed unittest

* set by_epoch=False

* fixed efficient test

* update docstring

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-05-05 16:19:54 -07:00
sshuair e16e0e303b
add metric mFscore (#509)
* add mFscore and refactor the metrics return value

* fix linting

* some docstring and name fix
2021-04-30 11:34:57 -07:00
sennnnn cf2cb542f7
Adjust vision transformer backbone architectures (#524)
* Adjust vision transformer backbone architectures;

* Add DropPath, trunc_normal_ for VisionTransformer implementation;

* Add class token buring intermediate period and remove it during final period;

* Fix some parameters loss bug;

* * Store intermediate token features and impose no processes on them;

* Remove class token and reshape entire token feature from NLC to NCHW;

* Fix some doc error

* Add a arg for VisionTransformer backbone to control if input class token into transformer;

* Add stochastic depth decay rule for DropPath;

* * Fix output bug when input_cls_token=False;

* Add related unit test;

* * Add arg: out_indices to control model output;

* Add unit test for DropPath;

* Apply suggestions from code review

Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2021-04-30 10:37:47 -07:00
Ziyi Wu cef8a4f611
[Enhance] Support reading class_weight from file in loss functions to help MMDet3D (#513)
* support reading class_weight from file in loss function

* add unit test of loss with class_weight from file

* minor fix

* move get_class_weight to utils
2021-04-29 01:04:15 -07:00
Ziyi Wu 768f704517
[Enhance] Replace data_dict calling 'img' key to support MMDet3D (#514)
* remove dict calling img key for compatibility

* fix unit test

* infer batch size using len(result) to be consistent with mmcv
2021-04-29 01:01:34 -07:00
q.yao bdf5adf2ce
[Feature] add onnxruntime test tool (#498)
* add onnxruntime test tool, update pytorch2onnx to support slice export

* onnx convert with custom output shape, update test code

* update pytorch2onnx, add rescale_shape support, add document

* update doc for lint error fixing

* remove cpu flag in ort_test.py

* change class name, fix cuda error

* remote comment

* fix bug of torch2onnx

* mIOU to mIoU
2021-04-28 20:38:01 -07:00
Jerry Jiarui XU d568d06e75
[Refactor] Use MMCV MODEL_REGISTRY (#515)
* [Refactor] Use MMCV MODEL_REGISTRY

* fixed args
2021-04-27 23:51:09 -07:00
Jerry Jiarui XU 2da3da47ed
Add QR Code in Chinese README.md (#506) 2021-04-25 19:15:13 -07:00
谢昕辰 98ef5ac705
add upsample neck (#512)
* init

* upsample v1.0

* fix errors

* change to in_channels list

* add unittest, docstring, norm/act config and rename

Co-authored-by: xiexinch <test767803@foxmail.com>
2021-04-24 21:22:09 -07:00
sennnnn 84fb600d47
Fix "the output num_classes of auxiliary head is not same as num_classes of ground truth seg_map". (#488)
* Fix "the output num_classes of auxiliary head is not match num_classes of ground truth seg_map".

* Fix spelling mistake;

* Modify "model download link", "log link";
2021-04-24 15:31:29 -07:00
Jerry Jiarui XU dc3dcb0b34
comment tag (#505) 2021-04-24 09:58:59 -07:00
sennnnn 838b6472d0
Add some publish information of implemented models (#508)
* Add some publish information

* Fix some readme omission.
2021-04-24 09:58:15 -07:00
Jerry Jiarui XU 89f8a6a410
[Bug fix] fixed fp16 inference (#497)
* fixed fp16

* update fps
2021-04-24 09:57:33 -07:00
谢昕辰 5b33faa146
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 03f2a254dc
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 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