mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/configs/ocrnet
History
sennnnn d3dc4f9583 [Enhancement] Add Dev tools to boost develop (#798) 
* Modify default work dir when training.

* Refactor gather_models.py.

* Add train and test matching list.

* Regression benchmark list.

* lower readme name to upper readme name.

* Add url check tool and model inference test tool.

* Modify tool name.

* Support duplicate mode of log json url check.

* Add regression benchmark evaluation automatic tool.

* Add train script generator.

* Only Support script running.

* Add evaluation results gather.

* Add exec Authority.

* Automatically make checkpoint root folder.

* Modify gather results save path.

* Coarse-grained train results gather tool.

* Complete benchmark train script.

* Make some little modifications.

* Fix checkpoint urls.

* Fix unet checkpoint urls.

* Fix fast scnn & fcn checkpoint url.

* Fix fast scnn checkpoint urls.

* Fix fast scnn url.

* Add differential results calculation.

* Add differential results of regression benchmark train results.

* Add an extra argument to select model.

* Update nonlocal_net & hrnet checkpoint url.

* Fix checkpoint url of hrnet and Fix some tta evaluation results and modify gather models tool.

* Modify fast scnn checkpoint url.

* Resolve new comments.

* Fix url check status code bug.

* Resolve some comments.

* Modify train scripts generator.

* Modify work_dir of regression benchmark results.

* model gather tool modification.
 		2021-09-02 09:44:51 -07:00
..
README.md [Enhancement] Add Dev tools to boost develop (#798) 2021-09-02 09:44:51 -07:00
ocrnet.yml [Enhancement] Add Dev tools to boost develop (#798) 2021-09-02 09:44:51 -07:00
ocrnet_hr18_512x512_20k_voc12aug.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr18_512x512_40k_voc12aug.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr18_512x512_80k_ade20k.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr18_512x512_160k_ade20k.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr18_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18_512x1024_160k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x512_20k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x512_40k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x512_80k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x512_160k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr18s_512x1024_160k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
ocrnet_hr48_512x512_20k_voc12aug.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x512_40k_voc12aug.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x512_80k_ade20k.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x512_160k_ade20k.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x1024_40k_cityscapes.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x1024_80k_cityscapes.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_hr48_512x1024_160k_cityscapes.py Add test tutorial (#9) 2020-07-12 23:53:56 +08:00
ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py [Improvement] Add markdown linter and fix linting errors (#171) 2020-10-07 19:50:16 +08:00
ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py [Improvement] Add markdown linter and fix linting errors (#171) 2020-10-07 19:50:16 +08:00
ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py [Improvement] Add markdown linter and fix linting errors (#171) 2020-10-07 19:50:16 +08:00

README.md

Object-Contextual Representations for Semantic Segmentation

Introduction

@article{YuanW18,
  title={Ocnet: Object context network for scene parsing},
  author={Yuhui Yuan and Jingdong Wang},
  booktitle={arXiv preprint arXiv:1809.00916},
  year={2018}
}

@article{YuanCW20,
  title={Object-Contextual Representations for Semantic Segmentation},
  author={Yuhui Yuan and Xilin Chen and Jingdong Wang},
  booktitle={ECCV},
  year={2020}
}

Results and models

Cityscapes

HRNet backbone

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
OCRNet HRNetV2p-W18-Small 512x1024 40000 3.5 10.45 74.30 75.95 config model | log
OCRNet HRNetV2p-W18 512x1024 40000 4.7 7.50 77.72 79.49 config model | log
OCRNet HRNetV2p-W48 512x1024 40000 8 4.22 80.58 81.79 config model | log
OCRNet HRNetV2p-W18-Small 512x1024 80000 - - 77.16 78.66 config model | log
OCRNet HRNetV2p-W18 512x1024 80000 - - 78.57 80.46 config model | log
OCRNet HRNetV2p-W48 512x1024 80000 - - 80.70 81.87 config model | log
OCRNet HRNetV2p-W18-Small 512x1024 160000 - - 78.45 79.97 config model | log
OCRNet HRNetV2p-W18 512x1024 160000 - - 79.47 80.91 config model | log
OCRNet HRNetV2p-W48 512x1024 160000 - - 81.35 82.70 config model | log

ResNet backbone

Method Backbone Crop Size Batch Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
OCRNet R-101-D8 512x1024 8 40000 - - 80.09 - config model | log
OCRNet R-101-D8 512x1024 16 40000 8.8 3.02 80.30 - config model | log
OCRNet R-101-D8 512x1024 16 80000 8.8 3.02 80.81 - config model | log

ADE20K

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
OCRNet HRNetV2p-W18-Small 512x512 80000 6.7 28.98 35.06 35.80 config model | log
OCRNet HRNetV2p-W18 512x512 80000 7.9 18.93 37.79 39.16 config model | log
OCRNet HRNetV2p-W48 512x512 80000 11.2 16.99 43.00 44.30 config model | log
OCRNet HRNetV2p-W18-Small 512x512 160000 - - 37.19 38.40 config model | log
OCRNet HRNetV2p-W18 512x512 160000 - - 39.32 40.80 config model | log
OCRNet HRNetV2p-W48 512x512 160000 - - 43.25 44.88 config model | log

Pascal VOC 2012 + Aug

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
OCRNet HRNetV2p-W18-Small 512x512 20000 3.5 31.55 71.70 73.84 config model | log
OCRNet HRNetV2p-W18 512x512 20000 4.7 19.91 74.75 77.11 config model | log
OCRNet HRNetV2p-W48 512x512 20000 8.1 17.83 77.72 79.87 config model | log
OCRNet HRNetV2p-W18-Small 512x512 40000 - - 72.76 74.60 config model | log
OCRNet HRNetV2p-W18 512x512 40000 - - 74.98 77.40 config model | log
OCRNet HRNetV2p-W48 512x512 40000 - - 77.14 79.71 config model | log