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
谢昕辰 725d5aa002 [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
..
README.md comment tag (#505) 2021-04-24 09:58:59 -07:00
metafile.yml [Feature] support mim (#549) 2021-05-31 15:07:24 -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