├── .dev
├── gather_models.py
├── md2yml.py
└── upload_modelzoo.py
├── .github
├── CODE_OF_CONDUCT.md
├── CONTRIBUTING.md
├── ISSUE_TEMPLATE
│ ├── config.yml
│ ├── error-report.md
│ ├── feature_request.md
│ ├── general_questions.md
│ └── reimplementation_questions.md
├── pull_request_template.md
└── workflows
│ ├── build.yml
│ └── deploy.yml
├── .gitignore
├── .pre-commit-config.yaml
├── .readthedocs.yml
├── LICENSE
├── MANIFEST.in
├── README.md
├── README_MMSeg.md
├── README_zh-CN.md
├── configs
├── _base_
│ ├── datasets
│ │ ├── ade20k.py
│ │ ├── chase_db1.py
│ │ ├── cityscapes.py
│ │ ├── cityscapes_769x769.py
│ │ ├── drive.py
│ │ ├── hrf.py
│ │ ├── pascal_context.py
│ │ ├── pascal_context_59.py
│ │ ├── pascal_voc12.py
│ │ ├── pascal_voc12_aug.py
│ │ └── stare.py
│ ├── default_runtime.py
│ ├── models
│ │ ├── ann_r50-d8.py
│ │ ├── apcnet_r50-d8.py
│ │ ├── ccnet_r50-d8.py
│ │ ├── cgnet.py
│ │ ├── danet_r50-d8.py
│ │ ├── deeplabv3_r50-d8.py
│ │ ├── deeplabv3_unet_s5-d16.py
│ │ ├── deeplabv3plus_r50-d8.py
│ │ ├── dmnet_r50-d8.py
│ │ ├── dnl_r50-d8.py
│ │ ├── emanet_r50-d8.py
│ │ ├── encnet_r50-d8.py
│ │ ├── fast_scnn.py
│ │ ├── fcn-resize-concat_litehr18-with-head.py
│ │ ├── fcn-resize-concat_litehr18-without-head.py
│ │ ├── fcn-resize-concat_litehr30-with-head.py
│ │ ├── fcn-resize-concat_litehr30-without-head.py
│ │ ├── fcn_hr18.py
│ │ ├── fcn_litehr18-with-head.py
│ │ ├── fcn_litehr18-without-head.py
│ │ ├── fcn_litehr30-with-head.py
│ │ ├── fcn_litehr30-without-head.py
│ │ ├── fcn_r50-d8.py
│ │ ├── fcn_unet_s5-d16.py
│ │ ├── fpn_r50.py
│ │ ├── gcnet_r50-d8.py
│ │ ├── lraspp_m-v3-d8.py
│ │ ├── nonlocal_r50-d8.py
│ │ ├── ocrnet_hr18.py
│ │ ├── ocrnet_r50-d8.py
│ │ ├── pointrend_r50.py
│ │ ├── psanet_r50-d8.py
│ │ ├── pspnet_r50-d8.py
│ │ ├── pspnet_unet_s5-d16.py
│ │ ├── setr_mla.py
│ │ ├── setr_naive.py
│ │ ├── setr_pup.py
│ │ ├── upernet_r50.py
│ │ ├── upernet_swin.py
│ │ └── upernet_vit-b16_ln_mln.py
│ └── schedules
│ │ ├── schedule_160k.py
│ │ ├── schedule_20k.py
│ │ ├── schedule_320k.py
│ │ ├── schedule_40k.py
│ │ ├── schedule_640k.py
│ │ └── schedule_80k.py
├── ann
│ ├── README.md
│ ├── ann.yml
│ ├── ann_r101-d8_512x1024_40k_cityscapes.py
│ ├── ann_r101-d8_512x1024_80k_cityscapes.py
│ ├── ann_r101-d8_512x512_160k_ade20k.py
│ ├── ann_r101-d8_512x512_20k_voc12aug.py
│ ├── ann_r101-d8_512x512_40k_voc12aug.py
│ ├── ann_r101-d8_512x512_80k_ade20k.py
│ ├── ann_r101-d8_769x769_40k_cityscapes.py
│ ├── ann_r101-d8_769x769_80k_cityscapes.py
│ ├── ann_r50-d8_512x1024_40k_cityscapes.py
│ ├── ann_r50-d8_512x1024_80k_cityscapes.py
│ ├── ann_r50-d8_512x512_160k_ade20k.py
│ ├── ann_r50-d8_512x512_20k_voc12aug.py
│ ├── ann_r50-d8_512x512_40k_voc12aug.py
│ ├── ann_r50-d8_512x512_80k_ade20k.py
│ ├── ann_r50-d8_769x769_40k_cityscapes.py
│ └── ann_r50-d8_769x769_80k_cityscapes.py
├── apcnet
│ ├── README.md
│ ├── apcnet.yml
│ ├── apcnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── apcnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── apcnet_r101-d8_512x512_160k_ade20k.py
│ ├── apcnet_r101-d8_512x512_80k_ade20k.py
│ ├── apcnet_r101-d8_769x769_40k_cityscapes.py
│ ├── apcnet_r101-d8_769x769_80k_cityscapes.py
│ ├── apcnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── apcnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── apcnet_r50-d8_512x512_160k_ade20k.py
│ ├── apcnet_r50-d8_512x512_80k_ade20k.py
│ ├── apcnet_r50-d8_769x769_40k_cityscapes.py
│ └── apcnet_r50-d8_769x769_80k_cityscapes.py
├── ccnet
│ ├── README.md
│ ├── ccnet.yml
│ ├── ccnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── ccnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── ccnet_r101-d8_512x512_160k_ade20k.py
│ ├── ccnet_r101-d8_512x512_20k_voc12aug.py
│ ├── ccnet_r101-d8_512x512_40k_voc12aug.py
│ ├── ccnet_r101-d8_512x512_80k_ade20k.py
│ ├── ccnet_r101-d8_769x769_40k_cityscapes.py
│ ├── ccnet_r101-d8_769x769_80k_cityscapes.py
│ ├── ccnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── ccnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── ccnet_r50-d8_512x512_160k_ade20k.py
│ ├── ccnet_r50-d8_512x512_20k_voc12aug.py
│ ├── ccnet_r50-d8_512x512_40k_voc12aug.py
│ ├── ccnet_r50-d8_512x512_80k_ade20k.py
│ ├── ccnet_r50-d8_769x769_40k_cityscapes.py
│ └── ccnet_r50-d8_769x769_80k_cityscapes.py
├── cgnet
│ ├── README.md
│ ├── cgnet.yml
│ ├── cgnet_512x1024_60k_cityscapes.py
│ └── cgnet_680x680_60k_cityscapes.py
├── danet
│ ├── README.md
│ ├── danet.yml
│ ├── danet_r101-d8_512x1024_40k_cityscapes.py
│ ├── danet_r101-d8_512x1024_80k_cityscapes.py
│ ├── danet_r101-d8_512x512_160k_ade20k.py
│ ├── danet_r101-d8_512x512_20k_voc12aug.py
│ ├── danet_r101-d8_512x512_40k_voc12aug.py
│ ├── danet_r101-d8_512x512_80k_ade20k.py
│ ├── danet_r101-d8_769x769_40k_cityscapes.py
│ ├── danet_r101-d8_769x769_80k_cityscapes.py
│ ├── danet_r50-d8_512x1024_40k_cityscapes.py
│ ├── danet_r50-d8_512x1024_80k_cityscapes.py
│ ├── danet_r50-d8_512x512_160k_ade20k.py
│ ├── danet_r50-d8_512x512_20k_voc12aug.py
│ ├── danet_r50-d8_512x512_40k_voc12aug.py
│ ├── danet_r50-d8_512x512_80k_ade20k.py
│ ├── danet_r50-d8_769x769_40k_cityscapes.py
│ └── danet_r50-d8_769x769_80k_cityscapes.py
├── deeplabv3
│ ├── README.md
│ ├── deeplabv3.yml
│ ├── deeplabv3_r101-d16-mg124_512x1024_40k_cityscapes.py
│ ├── deeplabv3_r101-d16-mg124_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r101-d8_480x480_40k_pascal_context.py
│ ├── deeplabv3_r101-d8_480x480_40k_pascal_context_59.py
│ ├── deeplabv3_r101-d8_480x480_80k_pascal_context.py
│ ├── deeplabv3_r101-d8_480x480_80k_pascal_context_59.py
│ ├── deeplabv3_r101-d8_512x1024_40k_cityscapes.py
│ ├── deeplabv3_r101-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r101-d8_512x512_160k_ade20k.py
│ ├── deeplabv3_r101-d8_512x512_20k_voc12aug.py
│ ├── deeplabv3_r101-d8_512x512_40k_voc12aug.py
│ ├── deeplabv3_r101-d8_512x512_80k_ade20k.py
│ ├── deeplabv3_r101-d8_769x769_40k_cityscapes.py
│ ├── deeplabv3_r101-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3_r101b-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r101b-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3_r18-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r18-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3_r18b-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r18b-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3_r50-d8_480x480_40k_pascal_context.py
│ ├── deeplabv3_r50-d8_480x480_40k_pascal_context_59.py
│ ├── deeplabv3_r50-d8_480x480_80k_pascal_context.py
│ ├── deeplabv3_r50-d8_480x480_80k_pascal_context_59.py
│ ├── deeplabv3_r50-d8_512x1024_40k_cityscapes.py
│ ├── deeplabv3_r50-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_r50-d8_512x512_160k_ade20k.py
│ ├── deeplabv3_r50-d8_512x512_20k_voc12aug.py
│ ├── deeplabv3_r50-d8_512x512_40k_voc12aug.py
│ ├── deeplabv3_r50-d8_512x512_80k_ade20k.py
│ ├── deeplabv3_r50-d8_769x769_40k_cityscapes.py
│ ├── deeplabv3_r50-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3_r50b-d8_512x1024_80k_cityscapes.py
│ └── deeplabv3_r50b-d8_769x769_80k_cityscapes.py
├── deeplabv3plus
│ ├── README.md
│ ├── deeplabv3plus.yml
│ ├── deeplabv3plus_r101-d16-mg124_512x1024_40k_cityscapes.py
│ ├── deeplabv3plus_r101-d16-mg124_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context.py
│ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py
│ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context.py
│ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py
│ ├── deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py
│ ├── deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r101-d8_512x512_160k_ade20k.py
│ ├── deeplabv3plus_r101-d8_512x512_20k_voc12aug.py
│ ├── deeplabv3plus_r101-d8_512x512_40k_voc12aug.py
│ ├── deeplabv3plus_r101-d8_512x512_80k_ade20k.py
│ ├── deeplabv3plus_r101-d8_769x769_40k_cityscapes.py
│ ├── deeplabv3plus_r101-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r18-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context.py
│ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py
│ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context.py
│ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py
│ ├── deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py
│ ├── deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_r50-d8_512x512_160k_ade20k.py
│ ├── deeplabv3plus_r50-d8_512x512_20k_voc12aug.py
│ ├── deeplabv3plus_r50-d8_512x512_40k_voc12aug.py
│ ├── deeplabv3plus_r50-d8_512x512_80k_ade20k.py
│ ├── deeplabv3plus_r50-d8_769x769_40k_cityscapes.py
│ ├── deeplabv3plus_r50-d8_769x769_80k_cityscapes.py
│ ├── deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py
│ └── deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py
├── dmnet
│ ├── README.md
│ ├── dmnet.yml
│ ├── dmnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── dmnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── dmnet_r101-d8_512x512_160k_ade20k.py
│ ├── dmnet_r101-d8_512x512_80k_ade20k.py
│ ├── dmnet_r101-d8_769x769_40k_cityscapes.py
│ ├── dmnet_r101-d8_769x769_80k_cityscapes.py
│ ├── dmnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── dmnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── dmnet_r50-d8_512x512_160k_ade20k.py
│ ├── dmnet_r50-d8_512x512_80k_ade20k.py
│ ├── dmnet_r50-d8_769x769_40k_cityscapes.py
│ └── dmnet_r50-d8_769x769_80k_cityscapes.py
├── dnlnet
│ ├── README.md
│ ├── dnl_r101-d8_512x1024_40k_cityscapes.py
│ ├── dnl_r101-d8_512x1024_80k_cityscapes.py
│ ├── dnl_r101-d8_512x512_160k_ade20k.py
│ ├── dnl_r101-d8_512x512_80k_ade20k.py
│ ├── dnl_r101-d8_769x769_40k_cityscapes.py
│ ├── dnl_r101-d8_769x769_80k_cityscapes.py
│ ├── dnl_r50-d8_512x1024_40k_cityscapes.py
│ ├── dnl_r50-d8_512x1024_80k_cityscapes.py
│ ├── dnl_r50-d8_512x512_160k_ade20k.py
│ ├── dnl_r50-d8_512x512_80k_ade20k.py
│ ├── dnl_r50-d8_769x769_40k_cityscapes.py
│ ├── dnl_r50-d8_769x769_80k_cityscapes.py
│ └── dnlnet.yml
├── emanet
│ ├── README.md
│ ├── emanet.yml
│ ├── emanet_r101-d8_512x1024_80k_cityscapes.py
│ ├── emanet_r101-d8_769x769_80k_cityscapes.py
│ ├── emanet_r50-d8_512x1024_80k_cityscapes.py
│ └── emanet_r50-d8_769x769_80k_cityscapes.py
├── encnet
│ ├── README.md
│ ├── encnet.yml
│ ├── encnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── encnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── encnet_r101-d8_512x512_160k_ade20k.py
│ ├── encnet_r101-d8_512x512_20k_voc12aug.py
│ ├── encnet_r101-d8_512x512_40k_voc12aug.py
│ ├── encnet_r101-d8_512x512_80k_ade20k.py
│ ├── encnet_r101-d8_769x769_40k_cityscapes.py
│ ├── encnet_r101-d8_769x769_80k_cityscapes.py
│ ├── encnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── encnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── encnet_r50-d8_512x512_160k_ade20k.py
│ ├── encnet_r50-d8_512x512_20k_voc12aug.py
│ ├── encnet_r50-d8_512x512_40k_voc12aug.py
│ ├── encnet_r50-d8_512x512_80k_ade20k.py
│ ├── encnet_r50-d8_769x769_40k_cityscapes.py
│ ├── encnet_r50-d8_769x769_80k_cityscapes.py
│ └── encnet_r50s-d8_512x512_80k_ade20k.py
├── fastscnn
│ ├── README.md
│ ├── fast_scnn_lr0.12_8x4_160k_cityscapes.py
│ └── fastscnn.yml
├── fcn
│ ├── README.md
│ ├── fcn.yml
│ ├── fcn_d6_r101-d16_512x1024_40k_cityscapes.py
│ ├── fcn_d6_r101-d16_512x1024_80k_cityscapes.py
│ ├── fcn_d6_r101-d16_769x769_40k_cityscapes.py
│ ├── fcn_d6_r101-d16_769x769_80k_cityscapes.py
│ ├── fcn_d6_r101b-d16_512x1024_80k_cityscapes.py
│ ├── fcn_d6_r101b-d16_769x769_80k_cityscapes.py
│ ├── fcn_d6_r50-d16_512x1024_40k_cityscapes.py
│ ├── fcn_d6_r50-d16_512x1024_80k_cityscapes.py
│ ├── fcn_d6_r50-d16_769x769_40k_cityscapes.py
│ ├── fcn_d6_r50-d16_769x769_80k_cityscapes.py
│ ├── fcn_d6_r50b-d16_512x1024_80k_cityscapes.py
│ ├── fcn_d6_r50b-d16_769x769_80k_cityscapes.py
│ ├── fcn_r101-d8_480x480_40k_pascal_context.py
│ ├── fcn_r101-d8_480x480_40k_pascal_context_59.py
│ ├── fcn_r101-d8_480x480_80k_pascal_context.py
│ ├── fcn_r101-d8_480x480_80k_pascal_context_59.py
│ ├── fcn_r101-d8_512x1024_40k_cityscapes.py
│ ├── fcn_r101-d8_512x1024_80k_cityscapes.py
│ ├── fcn_r101-d8_512x512_160k_ade20k.py
│ ├── fcn_r101-d8_512x512_20k_voc12aug.py
│ ├── fcn_r101-d8_512x512_40k_voc12aug.py
│ ├── fcn_r101-d8_512x512_80k_ade20k.py
│ ├── fcn_r101-d8_769x769_40k_cityscapes.py
│ ├── fcn_r101-d8_769x769_80k_cityscapes.py
│ ├── fcn_r101b-d8_512x1024_80k_cityscapes.py
│ ├── fcn_r101b-d8_769x769_80k_cityscapes.py
│ ├── fcn_r18-d8_512x1024_80k_cityscapes.py
│ ├── fcn_r18-d8_769x769_80k_cityscapes.py
│ ├── fcn_r18b-d8_512x1024_80k_cityscapes.py
│ ├── fcn_r18b-d8_769x769_80k_cityscapes.py
│ ├── fcn_r50-d8_480x480_40k_pascal_context.py
│ ├── fcn_r50-d8_480x480_40k_pascal_context_59.py
│ ├── fcn_r50-d8_480x480_80k_pascal_context.py
│ ├── fcn_r50-d8_480x480_80k_pascal_context_59.py
│ ├── fcn_r50-d8_512x1024_40k_cityscapes.py
│ ├── fcn_r50-d8_512x1024_80k_cityscapes.py
│ ├── fcn_r50-d8_512x512_160k_ade20k.py
│ ├── fcn_r50-d8_512x512_20k_voc12aug.py
│ ├── fcn_r50-d8_512x512_40k_voc12aug.py
│ ├── fcn_r50-d8_512x512_80k_ade20k.py
│ ├── fcn_r50-d8_769x769_40k_cityscapes.py
│ ├── fcn_r50-d8_769x769_80k_cityscapes.py
│ ├── fcn_r50b-d8_512x1024_80k_cityscapes.py
│ └── fcn_r50b-d8_769x769_80k_cityscapes.py
├── fp16
│ ├── README.md
│ ├── deeplabv3_r101-d8_512x1024_80k_fp16_cityscapes.py
│ ├── deeplabv3plus_r101-d8_512x1024_80k_fp16_cityscapes.py
│ ├── fcn_r101-d8_512x1024_80k_fp16_cityscapes.py
│ ├── fp16.yml
│ └── pspnet_r101-d8_512x1024_80k_fp16_cityscapes.py
├── gcnet
│ ├── README.md
│ ├── gcnet.yml
│ ├── gcnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── gcnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── gcnet_r101-d8_512x512_160k_ade20k.py
│ ├── gcnet_r101-d8_512x512_20k_voc12aug.py
│ ├── gcnet_r101-d8_512x512_40k_voc12aug.py
│ ├── gcnet_r101-d8_512x512_80k_ade20k.py
│ ├── gcnet_r101-d8_769x769_40k_cityscapes.py
│ ├── gcnet_r101-d8_769x769_80k_cityscapes.py
│ ├── gcnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── gcnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── gcnet_r50-d8_512x512_160k_ade20k.py
│ ├── gcnet_r50-d8_512x512_20k_voc12aug.py
│ ├── gcnet_r50-d8_512x512_40k_voc12aug.py
│ ├── gcnet_r50-d8_512x512_80k_ade20k.py
│ ├── gcnet_r50-d8_769x769_40k_cityscapes.py
│ └── gcnet_r50-d8_769x769_80k_cityscapes.py
├── hrnet
│ ├── README.md
│ ├── fcn_hr18_480x480_40k_pascal_context.py
│ ├── fcn_hr18_480x480_40k_pascal_context_59.py
│ ├── fcn_hr18_480x480_80k_pascal_context.py
│ ├── fcn_hr18_480x480_80k_pascal_context_59.py
│ ├── fcn_hr18_512x1024_160k_cityscapes.py
│ ├── fcn_hr18_512x1024_40k_cityscapes.py
│ ├── fcn_hr18_512x1024_80k_cityscapes.py
│ ├── fcn_hr18_512x512_160k_ade20k.py
│ ├── fcn_hr18_512x512_20k_voc12aug.py
│ ├── fcn_hr18_512x512_40k_voc12aug.py
│ ├── fcn_hr18_512x512_80k_ade20k.py
│ ├── fcn_hr18s_480x480_40k_pascal_context.py
│ ├── fcn_hr18s_480x480_40k_pascal_context_59.py
│ ├── fcn_hr18s_480x480_80k_pascal_context.py
│ ├── fcn_hr18s_480x480_80k_pascal_context_59.py
│ ├── fcn_hr18s_512x1024_160k_cityscapes.py
│ ├── fcn_hr18s_512x1024_40k_cityscapes.py
│ ├── fcn_hr18s_512x1024_80k_cityscapes.py
│ ├── fcn_hr18s_512x512_160k_ade20k.py
│ ├── fcn_hr18s_512x512_20k_voc12aug.py
│ ├── fcn_hr18s_512x512_40k_voc12aug.py
│ ├── fcn_hr18s_512x512_80k_ade20k.py
│ ├── fcn_hr48_480x480_40k_pascal_context.py
│ ├── fcn_hr48_480x480_40k_pascal_context_59.py
│ ├── fcn_hr48_480x480_80k_pascal_context.py
│ ├── fcn_hr48_480x480_80k_pascal_context_59.py
│ ├── fcn_hr48_512x1024_160k_cityscapes.py
│ ├── fcn_hr48_512x1024_40k_cityscapes.py
│ ├── fcn_hr48_512x1024_80k_cityscapes.py
│ ├── fcn_hr48_512x512_160k_ade20k.py
│ ├── fcn_hr48_512x512_20k_voc12aug.py
│ ├── fcn_hr48_512x512_40k_voc12aug.py
│ ├── fcn_hr48_512x512_80k_ade20k.py
│ └── hrnet.yml
├── litehrnet
│ ├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-with-head_512x512_160k_ade20k.py
│ ├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-without-head_512x512_160k_ade20k.py
│ ├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-with-head_512x512_160k_ade20k.py
│ ├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-without-head_512x512_160k_ade20k.py
│ ├── fcn_litehr18-with-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn_litehr18-with-head_512x512_160k_ade20k.py
│ ├── fcn_litehr18-without-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn_litehr18-without-head_512x512_160k_ade20k.py
│ ├── fcn_litehr30-with-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn_litehr30-with-head_512x512_160k_ade20k.py
│ ├── fcn_litehr30-without-head_512x1024_8x2_160k_cityscapes.py
│ ├── fcn_litehr30-without-head_512x512_160k_ade20k.py
│ ├── litehr---.sh
│ ├── litehr---ade20k.sh
│ └── litehr---cityscapes.sh
├── litehrnet_320k
│ ├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-with-head_512x512_320k_ade20k.py
│ ├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-without-head_512x512_320k_ade20k.py
│ ├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-with-head_512x512_320k_ade20k.py
│ ├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-without-head_512x512_320k_ade20k.py
│ ├── fcn_litehr18-with-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn_litehr18-with-head_512x512_320k_ade20k.py
│ ├── fcn_litehr18-without-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn_litehr18-without-head_512x512_320k_ade20k.py
│ ├── fcn_litehr30-with-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn_litehr30-with-head_512x512_320k_ade20k.py
│ ├── fcn_litehr30-without-head_512x1024_8x2_320k_cityscapes.py
│ ├── fcn_litehr30-without-head_512x512_320k_ade20k.py
│ ├── litehr---ade20k.sh
│ └── litehr---cityscapes.sh
├── litehrnet_640k
│ ├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-with-head_512x512_640k_ade20k.py
│ ├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn-resize-concat_litehr18-without-head_512x512_640k_ade20k.py
│ ├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-with-head_512x512_640k_ade20k.py
│ ├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn-resize-concat_litehr30-without-head_512x512_640k_ade20k.py
│ ├── fcn_litehr18-with-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn_litehr18-with-head_512x512_640k_ade20k.py
│ ├── fcn_litehr18-without-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn_litehr18-without-head_512x512_640k_ade20k.py
│ ├── fcn_litehr30-with-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn_litehr30-with-head_512x512_640k_ade20k.py
│ ├── fcn_litehr30-without-head_512x1024_8x2_640k_cityscapes.py
│ ├── fcn_litehr30-without-head_512x512_640k_ade20k.py
│ ├── litehr---ade20k.sh
│ ├── litehr---cityscapes.sh
│ └── litehr--640k-cityscapes.sh
├── mobilenet_v2
│ ├── README.md
│ ├── deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_m-v2-d8_512x512_160k_ade20k.py
│ ├── deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py
│ ├── fcn_m-v2-d8_512x1024_80k_cityscapes.py
│ ├── fcn_m-v2-d8_512x512_160k_ade20k.py
│ ├── mobilenet_v2.yml
│ ├── pspnet_m-v2-d8_512x1024_80k_cityscapes.py
│ └── pspnet_m-v2-d8_512x512_160k_ade20k.py
├── mobilenet_v3
│ ├── README.md
│ ├── lraspp_m-v3-d8_512x1024_320k_cityscapes.py
│ ├── lraspp_m-v3-d8_scratch_512x1024_320k_cityscapes.py
│ ├── lraspp_m-v3s-d8_512x1024_320k_cityscapes.py
│ ├── lraspp_m-v3s-d8_scratch_512x1024_320k_cityscapes.py
│ └── mobilenet_v3.yml
├── nonlocal_net
│ ├── README.md
│ ├── nonlocal_net.yml
│ ├── nonlocal_r101-d8_512x1024_40k_cityscapes.py
│ ├── nonlocal_r101-d8_512x1024_80k_cityscapes.py
│ ├── nonlocal_r101-d8_512x512_160k_ade20k.py
│ ├── nonlocal_r101-d8_512x512_20k_voc12aug.py
│ ├── nonlocal_r101-d8_512x512_40k_voc12aug.py
│ ├── nonlocal_r101-d8_512x512_80k_ade20k.py
│ ├── nonlocal_r101-d8_769x769_40k_cityscapes.py
│ ├── nonlocal_r101-d8_769x769_80k_cityscapes.py
│ ├── nonlocal_r50-d8_512x1024_40k_cityscapes.py
│ ├── nonlocal_r50-d8_512x1024_80k_cityscapes.py
│ ├── nonlocal_r50-d8_512x512_160k_ade20k.py
│ ├── nonlocal_r50-d8_512x512_20k_voc12aug.py
│ ├── nonlocal_r50-d8_512x512_40k_voc12aug.py
│ ├── nonlocal_r50-d8_512x512_80k_ade20k.py
│ ├── nonlocal_r50-d8_769x769_40k_cityscapes.py
│ └── nonlocal_r50-d8_769x769_80k_cityscapes.py
├── ocrnet
│ ├── README.md
│ ├── ocrnet.yml
│ ├── ocrnet_hr18_512x1024_160k_cityscapes.py
│ ├── ocrnet_hr18_512x1024_40k_cityscapes.py
│ ├── ocrnet_hr18_512x1024_80k_cityscapes.py
│ ├── ocrnet_hr18_512x512_160k_ade20k.py
│ ├── ocrnet_hr18_512x512_20k_voc12aug.py
│ ├── ocrnet_hr18_512x512_40k_voc12aug.py
│ ├── ocrnet_hr18_512x512_80k_ade20k.py
│ ├── ocrnet_hr18s_512x1024_160k_cityscapes.py
│ ├── ocrnet_hr18s_512x1024_40k_cityscapes.py
│ ├── ocrnet_hr18s_512x1024_80k_cityscapes.py
│ ├── ocrnet_hr18s_512x512_160k_ade20k.py
│ ├── ocrnet_hr18s_512x512_20k_voc12aug.py
│ ├── ocrnet_hr18s_512x512_40k_voc12aug.py
│ ├── ocrnet_hr18s_512x512_80k_ade20k.py
│ ├── ocrnet_hr48_512x1024_160k_cityscapes.py
│ ├── ocrnet_hr48_512x1024_40k_cityscapes.py
│ ├── ocrnet_hr48_512x1024_80k_cityscapes.py
│ ├── ocrnet_hr48_512x512_160k_ade20k.py
│ ├── ocrnet_hr48_512x512_20k_voc12aug.py
│ ├── ocrnet_hr48_512x512_40k_voc12aug.py
│ ├── ocrnet_hr48_512x512_80k_ade20k.py
│ ├── ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py
│ ├── ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py
│ └── ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py
├── point_rend
│ ├── README.md
│ ├── point_rend.yml
│ ├── pointrend_r101_512x1024_80k_cityscapes.py
│ ├── pointrend_r101_512x512_160k_ade20k.py
│ ├── pointrend_r50_512x1024_80k_cityscapes.py
│ └── pointrend_r50_512x512_160k_ade20k.py
├── psanet
│ ├── README.md
│ ├── psanet.yml
│ ├── psanet_r101-d8_512x1024_40k_cityscapes.py
│ ├── psanet_r101-d8_512x1024_80k_cityscapes.py
│ ├── psanet_r101-d8_512x512_160k_ade20k.py
│ ├── psanet_r101-d8_512x512_20k_voc12aug.py
│ ├── psanet_r101-d8_512x512_40k_voc12aug.py
│ ├── psanet_r101-d8_512x512_80k_ade20k.py
│ ├── psanet_r101-d8_769x769_40k_cityscapes.py
│ ├── psanet_r101-d8_769x769_80k_cityscapes.py
│ ├── psanet_r50-d8_512x1024_40k_cityscapes.py
│ ├── psanet_r50-d8_512x1024_80k_cityscapes.py
│ ├── psanet_r50-d8_512x512_160k_ade20k.py
│ ├── psanet_r50-d8_512x512_20k_voc12aug.py
│ ├── psanet_r50-d8_512x512_40k_voc12aug.py
│ ├── psanet_r50-d8_512x512_80k_ade20k.py
│ ├── psanet_r50-d8_769x769_40k_cityscapes.py
│ └── psanet_r50-d8_769x769_80k_cityscapes.py
├── pspnet
│ ├── README.md
│ ├── pspnet.yml
│ ├── pspnet_r101-d8_480x480_40k_pascal_context.py
│ ├── pspnet_r101-d8_480x480_40k_pascal_context_59.py
│ ├── pspnet_r101-d8_480x480_80k_pascal_context.py
│ ├── pspnet_r101-d8_480x480_80k_pascal_context_59.py
│ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py
│ ├── pspnet_r101-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_r101-d8_512x512_160k_ade20k.py
│ ├── pspnet_r101-d8_512x512_20k_voc12aug.py
│ ├── pspnet_r101-d8_512x512_40k_voc12aug.py
│ ├── pspnet_r101-d8_512x512_80k_ade20k.py
│ ├── pspnet_r101-d8_769x769_40k_cityscapes.py
│ ├── pspnet_r101-d8_769x769_80k_cityscapes.py
│ ├── pspnet_r101b-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_r101b-d8_769x769_80k_cityscapes.py
│ ├── pspnet_r18-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_r18-d8_769x769_80k_cityscapes.py
│ ├── pspnet_r18b-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_r18b-d8_769x769_80k_cityscapes.py
│ ├── pspnet_r50-d8_480x480_40k_pascal_context.py
│ ├── pspnet_r50-d8_480x480_40k_pascal_context_59.py
│ ├── pspnet_r50-d8_480x480_80k_pascal_context.py
│ ├── pspnet_r50-d8_480x480_80k_pascal_context_59.py
│ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py
│ ├── pspnet_r50-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_r50-d8_512x512_160k_ade20k.py
│ ├── pspnet_r50-d8_512x512_20k_voc12aug.py
│ ├── pspnet_r50-d8_512x512_40k_voc12aug.py
│ ├── pspnet_r50-d8_512x512_80k_ade20k.py
│ ├── pspnet_r50-d8_769x769_40k_cityscapes.py
│ ├── pspnet_r50-d8_769x769_80k_cityscapes.py
│ ├── pspnet_r50b-d8_512x1024_80k_cityscapes.py
│ └── pspnet_r50b-d8_769x769_80k_cityscapes.py
├── resnest
│ ├── README.md
│ ├── deeplabv3_s101-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3_s101-d8_512x512_160k_ade20k.py
│ ├── deeplabv3plus_s101-d8_512x1024_80k_cityscapes.py
│ ├── deeplabv3plus_s101-d8_512x512_160k_ade20k.py
│ ├── fcn_s101-d8_512x1024_80k_cityscapes.py
│ ├── fcn_s101-d8_512x512_160k_ade20k.py
│ ├── pspnet_s101-d8_512x1024_80k_cityscapes.py
│ ├── pspnet_s101-d8_512x512_160k_ade20k.py
│ └── resnest.yml
├── sem_fpn
│ ├── README.md
│ ├── fpn_r101_512x1024_80k_cityscapes.py
│ ├── fpn_r101_512x512_160k_ade20k.py
│ ├── fpn_r50_512x1024_80k_cityscapes.py
│ ├── fpn_r50_512x512_160k_ade20k.py
│ └── sem_fpn.yml
├── setr
│ ├── README.md
│ ├── setr.yml
│ ├── setr_mla_512x512_160k_b16_ade20k.py
│ ├── setr_mla_512x512_160k_b8_ade20k.py
│ ├── setr_naive_512x512_160k_b16_ade20k.py
│ └── setr_pup_512x512_160k_b16_ade20k.py
├── swin
│ ├── README.md
│ ├── swin.yml
│ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_1K.py
│ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py
│ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py
│ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py
│ ├── upernet_swin_small_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py
│ └── upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py
├── unet
│ ├── README.md
│ ├── deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py
│ ├── deeplabv3_unet_s5-d16_128x128_40k_stare.py
│ ├── deeplabv3_unet_s5-d16_256x256_40k_hrf.py
│ ├── deeplabv3_unet_s5-d16_64x64_40k_drive.py
│ ├── fcn_unet_s5-d16_128x128_40k_chase_db1.py
│ ├── fcn_unet_s5-d16_128x128_40k_stare.py
│ ├── fcn_unet_s5-d16_256x256_40k_hrf.py
│ ├── fcn_unet_s5-d16_64x64_40k_drive.py
│ ├── pspnet_unet_s5-d16_128x128_40k_chase_db1.py
│ ├── pspnet_unet_s5-d16_128x128_40k_stare.py
│ ├── pspnet_unet_s5-d16_256x256_40k_hrf.py
│ ├── pspnet_unet_s5-d16_64x64_40k_drive.py
│ └── unet.yml
├── upernet
│ ├── README.md
│ ├── upernet.yml
│ ├── upernet_r101_512x1024_40k_cityscapes.py
│ ├── upernet_r101_512x1024_80k_cityscapes.py
│ ├── upernet_r101_512x512_160k_ade20k.py
│ ├── upernet_r101_512x512_20k_voc12aug.py
│ ├── upernet_r101_512x512_40k_voc12aug.py
│ ├── upernet_r101_512x512_80k_ade20k.py
│ ├── upernet_r101_769x769_40k_cityscapes.py
│ ├── upernet_r101_769x769_80k_cityscapes.py
│ ├── upernet_r50_512x1024_40k_cityscapes.py
│ ├── upernet_r50_512x1024_80k_cityscapes.py
│ ├── upernet_r50_512x512_160k_ade20k.py
│ ├── upernet_r50_512x512_20k_voc12aug.py
│ ├── upernet_r50_512x512_40k_voc12aug.py
│ ├── upernet_r50_512x512_80k_ade20k.py
│ ├── upernet_r50_769x769_40k_cityscapes.py
│ └── upernet_r50_769x769_80k_cityscapes.py
└── vit
│ ├── README.md
│ ├── upernet_deit-b16_512x512_160k_ade20k.py
│ ├── upernet_deit-b16_512x512_80k_ade20k.py
│ ├── upernet_deit-b16_ln_mln_512x512_160k_ade20k.py
│ ├── upernet_deit-b16_mln_512x512_160k_ade20k.py
│ ├── upernet_deit-s16_512x512_160k_ade20k.py
│ ├── upernet_deit-s16_512x512_80k_ade20k.py
│ ├── upernet_deit-s16_ln_mln_512x512_160k_ade20k.py
│ ├── upernet_deit-s16_mln_512x512_160k_ade20k.py
│ ├── upernet_vit-b16_ln_mln_512x512_160k_ade20k.py
│ ├── upernet_vit-b16_mln_512x512_160k_ade20k.py
│ ├── upernet_vit-b16_mln_512x512_80k_ade20k.py
│ └── vit.yml
├── demo
├── MMSegmentation_Tutorial.ipynb
├── demo.png
├── image_demo.py
└── inference_demo.ipynb
├── docker
├── Dockerfile
└── serve
│ ├── Dockerfile
│ ├── config.properties
│ └── entrypoint.sh
├── docs
├── Makefile
├── api.rst
├── changelog.md
├── conf.py
├── dataset_prepare.md
├── get_started.md
├── index.rst
├── inference.md
├── make.bat
├── model_zoo.md
├── stat.py
├── switch_language.md
├── train.md
├── tutorials
│ ├── config.md
│ ├── customize_datasets.md
│ ├── customize_models.md
│ ├── customize_runtime.md
│ ├── data_pipeline.md
│ ├── index.rst
│ └── training_tricks.md
└── useful_tools.md
├── docs_zh-CN
├── Makefile
├── api.rst
├── conf.py
├── dataset_prepare.md
├── get_started.md
├── imgs
│ ├── qq_group_qrcode.jpg
│ ├── seggroup_qrcode.jpg
│ └── zhihu_qrcode.jpg
├── index.rst
├── inference.md
├── make.bat
├── model_zoo.md
├── stat.py
├── switch_language.md
├── train.md
├── tutorials
│ ├── config.md
│ ├── customize_datasets.md
│ ├── customize_models.md
│ ├── customize_runtime.md
│ ├── data_pipeline.md
│ ├── index.rst
│ └── training_tricks.md
└── useful_tools.md
├── mmseg
├── .mim
│ ├── configs
│ ├── model-index.yml
│ └── tools
├── __init__.py
├── apis
│ ├── __init__.py
│ ├── inference.py
│ ├── test.py
│ └── train.py
├── core
│ ├── __init__.py
│ ├── evaluation
│ │ ├── __init__.py
│ │ ├── class_names.py
│ │ ├── eval_hooks.py
│ │ └── metrics.py
│ ├── seg
│ │ ├── __init__.py
│ │ ├── builder.py
│ │ └── sampler
│ │ │ ├── __init__.py
│ │ │ ├── base_pixel_sampler.py
│ │ │ └── ohem_pixel_sampler.py
│ └── utils
│ │ ├── __init__.py
│ │ └── misc.py
├── datasets
│ ├── __init__.py
│ ├── ade.py
│ ├── builder.py
│ ├── chase_db1.py
│ ├── cityscapes.py
│ ├── custom.py
│ ├── dataset_wrappers.py
│ ├── drive.py
│ ├── hrf.py
│ ├── pascal_context.py
│ ├── pipelines
│ │ ├── __init__.py
│ │ ├── compose.py
│ │ ├── formating.py
│ │ ├── loading.py
│ │ ├── test_time_aug.py
│ │ └── transforms.py
│ ├── stare.py
│ └── voc.py
├── models
│ ├── __init__.py
│ ├── backbones
│ │ ├── __init__.py
│ │ ├── cgnet.py
│ │ ├── fast_scnn.py
│ │ ├── hrnet.py
│ │ ├── litehrnet.py
│ │ ├── mit.py
│ │ ├── mobilenet_v2.py
│ │ ├── mobilenet_v3.py
│ │ ├── resnest.py
│ │ ├── resnet.py
│ │ ├── resnext.py
│ │ ├── swin.py
│ │ ├── unet.py
│ │ └── vit.py
│ ├── builder.py
│ ├── decode_heads
│ │ ├── __init__.py
│ │ ├── ann_head.py
│ │ ├── apc_head.py
│ │ ├── aspp_head.py
│ │ ├── cascade_decode_head.py
│ │ ├── cc_head.py
│ │ ├── da_head.py
│ │ ├── decode_head.py
│ │ ├── dm_head.py
│ │ ├── dnl_head.py
│ │ ├── ema_head.py
│ │ ├── enc_head.py
│ │ ├── fcn_head.py
│ │ ├── fpn_head.py
│ │ ├── gc_head.py
│ │ ├── lraspp_head.py
│ │ ├── nl_head.py
│ │ ├── ocr_head.py
│ │ ├── point_head.py
│ │ ├── psa_head.py
│ │ ├── psp_head.py
│ │ ├── sep_aspp_head.py
│ │ ├── sep_fcn_head.py
│ │ ├── setr_mla_head.py
│ │ ├── setr_up_head.py
│ │ └── uper_head.py
│ ├── losses
│ │ ├── __init__.py
│ │ ├── accuracy.py
│ │ ├── cross_entropy_loss.py
│ │ ├── dice_loss.py
│ │ ├── lovasz_loss.py
│ │ └── utils.py
│ ├── necks
│ │ ├── __init__.py
│ │ ├── fpn.py
│ │ ├── mla_neck.py
│ │ └── multilevel_neck.py
│ ├── segmentors
│ │ ├── __init__.py
│ │ ├── base.py
│ │ ├── cascade_encoder_decoder.py
│ │ └── encoder_decoder.py
│ └── utils
│ │ ├── __init__.py
│ │ ├── channel_shuffle.py
│ │ ├── ckpt_convert.py
│ │ ├── embed.py
│ │ ├── inverted_residual.py
│ │ ├── make_divisible.py
│ │ ├── res_layer.py
│ │ ├── se_layer.py
│ │ ├── self_attention_block.py
│ │ ├── shape_convert.py
│ │ └── up_conv_block.py
├── ops
│ ├── __init__.py
│ ├── encoding.py
│ └── wrappers.py
├── utils
│ ├── __init__.py
│ ├── collect_env.py
│ └── logger.py
└── version.py
├── model-index.yml
├── pytest.ini
├── requirements.txt
├── requirements
├── docs.txt
├── mminstall.txt
├── optional.txt
├── readthedocs.txt
├── runtime.txt
└── tests.txt
├── resources
├── 3dogs.jpg
├── 3dogs_mask.png
├── mmseg-logo.png
└── seg_demo.gif
├── results
├── litehr-ade20k_results_20210901_010033.csv
├── litehr-ade20k_results_20210908_224220.csv
├── litehr-cityscapes_results_20210901_010014.csv
└── litehr-cityscapes_results_20210908_224254.csv
├── setup.cfg
├── setup.py
├── tests
├── __init__.py
├── data
│ ├── color.jpg
│ ├── gray.jpg
│ ├── pseudo_dataset
│ │ ├── gts
│ │ │ ├── 00000_gt.png
│ │ │ ├── 00001_gt.png
│ │ │ ├── 00002_gt.png
│ │ │ ├── 00003_gt.png
│ │ │ └── 00004_gt.png
│ │ ├── imgs
│ │ │ ├── 00000_img.jpg
│ │ │ ├── 00001_img.jpg
│ │ │ ├── 00002_img.jpg
│ │ │ ├── 00003_img.jpg
│ │ │ └── 00004_img.jpg
│ │ └── splits
│ │ │ ├── train.txt
│ │ │ └── val.txt
│ └── seg.png
├── test_config.py
├── test_data
│ ├── test_dataset.py
│ ├── test_dataset_builder.py
│ ├── test_loading.py
│ ├── test_transform.py
│ └── test_tta.py
├── test_digit_version.py
├── test_eval_hook.py
├── test_inference.py
├── test_metrics.py
├── test_models
│ ├── __init__.py
│ ├── test_backbones
│ │ ├── __init__.py
│ │ ├── test_blocks.py
│ │ ├── test_cgnet.py
│ │ ├── test_fast_scnn.py
│ │ ├── test_hrnet.py
│ │ ├── test_litehrnet.py
│ │ ├── test_mit.py
│ │ ├── test_mobilenet_v3.py
│ │ ├── test_resnest.py
│ │ ├── test_resnet.py
│ │ ├── test_resnext.py
│ │ ├── test_swin.py
│ │ ├── test_unet.py
│ │ ├── test_vit.py
│ │ └── utils.py
│ ├── test_forward.py
│ ├── test_heads
│ │ ├── __init__.py
│ │ ├── test_ann_head.py
│ │ ├── test_apc_head.py
│ │ ├── test_aspp_head.py
│ │ ├── test_cc_head.py
│ │ ├── test_da_head.py
│ │ ├── test_decode_head.py
│ │ ├── test_dm_head.py
│ │ ├── test_dnl_head.py
│ │ ├── test_ema_head.py
│ │ ├── test_enc_head.py
│ │ ├── test_fcn_head.py
│ │ ├── test_gc_head.py
│ │ ├── test_lraspp_head.py
│ │ ├── test_nl_head.py
│ │ ├── test_ocr_head.py
│ │ ├── test_point_head.py
│ │ ├── test_psa_head.py
│ │ ├── test_psp_head.py
│ │ ├── test_setr_mla_head.py
│ │ ├── test_setr_up_head.py
│ │ ├── test_uper_head.py
│ │ └── utils.py
│ ├── test_losses
│ │ ├── __init__.py
│ │ ├── test_ce_loss.py
│ │ ├── test_dice_loss.py
│ │ ├── test_lovasz_loss.py
│ │ └── test_utils.py
│ ├── test_necks
│ │ ├── __init__.py
│ │ ├── test_fpn.py
│ │ ├── test_mla_neck.py
│ │ └── test_multilevel_neck.py
│ └── test_segmentors
│ │ ├── __init__.py
│ │ ├── test_cascade_encoder_decoder.py
│ │ ├── test_encoder_decoder.py
│ │ └── utils.py
└── test_sampler.py
├── tools
├── analyze_logs.py
├── benchmark.py
├── convert_datasets
│ ├── chase_db1.py
│ ├── cityscapes.py
│ ├── drive.py
│ ├── hrf.py
│ ├── pascal_context.py
│ ├── stare.py
│ └── voc_aug.py
├── deploy_test.py
├── dist_test.sh
├── dist_train.sh
├── get_flops.py
├── mmseg2torchserve.py
├── mmseg_handler.py
├── onnx2tensorrt.py
├── print_config.py
├── publish_model.py
├── pytorch2onnx.py
├── pytorch2torchscript.py
├── slurm_test.sh
├── slurm_train.sh
├── test.py
└── train.py
└── work_dirs
├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_160k_cityscapes
├── 20210814_130915.log
├── 20210814_130915.log.json
└── fcn-resize-concat_litehr18-with-head_512x1024_8x2_160k_cityscapes.py
├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_320k_cityscapes
├── 20210819_165303.log
├── 20210819_165303.log.json
└── fcn-resize-concat_litehr18-with-head_512x1024_8x2_320k_cityscapes.py
├── fcn-resize-concat_litehr18-with-head_512x1024_8x2_640k_cityscapes
├── 20210827_033607.log
├── 20210827_033607.log.json
└── fcn-resize-concat_litehr18-with-head_512x1024_8x2_640k_cityscapes.py
├── fcn-resize-concat_litehr18-with-head_512x512_160k_ade20k
├── 20210815_232322.log
├── 20210815_232322.log.json
├── 20210816_104233.log
├── 20210816_104233.log.json
└── fcn-resize-concat_litehr18-with-head_512x512_160k_ade20k.py
├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_160k_cityscapes
├── 20210813_113814.log
├── 20210813_113814.log.json
└── fcn-resize-concat_litehr18-without-head_512x1024_8x2_160k_cityscapes.py
├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_320k_cityscapes
├── 20210818_202535.log
├── 20210818_202535.log.json
└── fcn-resize-concat_litehr18-without-head_512x1024_8x2_320k_cityscapes.py
├── fcn-resize-concat_litehr18-without-head_512x1024_8x2_640k_cityscapes
├── 20210827_165824.log
├── 20210827_165824.log.json
└── fcn-resize-concat_litehr18-without-head_512x1024_8x2_640k_cityscapes.py
├── fcn-resize-concat_litehr18-without-head_512x512_160k_ade20k
├── 20210816_110434.log
├── 20210816_110434.log.json
└── fcn-resize-concat_litehr18-without-head_512x512_160k_ade20k.py
├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_160k_cityscapes
├── 20210814_071016.log
├── 20210814_071016.log.json
└── fcn-resize-concat_litehr30-with-head_512x1024_8x2_160k_cityscapes.py
├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_320k_cityscapes
├── 20210819_163934.log
├── 20210819_163934.log.json
└── fcn-resize-concat_litehr30-with-head_512x1024_8x2_320k_cityscapes.py
├── fcn-resize-concat_litehr30-with-head_512x1024_8x2_640k_cityscapes
├── 20210826_154110.log
├── 20210826_154110.log.json
└── fcn-resize-concat_litehr30-with-head_512x1024_8x2_640k_cityscapes.py
├── fcn-resize-concat_litehr30-with-head_512x512_160k_ade20k
├── 20210816_171800.log
├── 20210816_171800.log.json
└── fcn-resize-concat_litehr30-with-head_512x512_160k_ade20k.py
├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_160k_cityscapes
├── 20210813_110906.log
├── 20210813_110906.log.json
└── fcn-resize-concat_litehr30-without-head_512x1024_8x2_160k_cityscapes.py
├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_320k_cityscapes
├── 20210819_115746.log
├── 20210819_115746.log.json
└── fcn-resize-concat_litehr30-without-head_512x1024_8x2_320k_cityscapes.py
├── fcn-resize-concat_litehr30-without-head_512x1024_8x2_640k_cityscapes
├── 20210827_170041.log
├── 20210827_170041.log.json
└── fcn-resize-concat_litehr30-without-head_512x1024_8x2_640k_cityscapes.py
├── fcn-resize-concat_litehr30-without-head_512x512_160k_ade20k
├── 20210816_151151.log
├── 20210816_151151.log.json
└── fcn-resize-concat_litehr30-without-head_512x512_160k_ade20k.py
├── fcn_litehr18-with-head_512x1024_8x2_160k_cityscapes
├── 20210814_151622.log
├── 20210814_151622.log.json
└── fcn_litehr18-with-head_512x1024_8x2_160k_cityscapes.py
├── fcn_litehr18-with-head_512x1024_8x2_320k_cityscapes
├── 20210818_214038.log
├── 20210818_214038.log.json
└── fcn_litehr18-with-head_512x1024_8x2_320k_cityscapes.py
├── fcn_litehr18-with-head_512x1024_8x2_640k_cityscapes
├── 20210827_033607.log
├── 20210827_033607.log.json
└── fcn_litehr18-with-head_512x1024_8x2_640k_cityscapes.py
├── fcn_litehr18-with-head_512x512_160k_ade20k
├── 20210816_010157.log
├── 20210816_010157.log.json
├── 20210816_111804.log
├── 20210816_111804.log.json
└── fcn_litehr18-with-head_512x512_160k_ade20k.py
├── fcn_litehr18-without-head_512x1024_8x2_160k_cityscapes
├── 20210814_144602.log
├── 20210814_144602.log.json
└── fcn_litehr18-without-head_512x1024_8x2_160k_cityscapes.py
├── fcn_litehr18-without-head_512x1024_8x2_320k_cityscapes
├── 20210818_180309.log
├── 20210818_180309.log.json
└── fcn_litehr18-without-head_512x1024_8x2_320k_cityscapes.py
├── fcn_litehr18-without-head_512x1024_8x2_640k_cityscapes
├── 20210827_095527.log
├── 20210827_095527.log.json
└── fcn_litehr18-without-head_512x1024_8x2_640k_cityscapes.py
├── fcn_litehr18-without-head_512x512_160k_ade20k
├── 20210815_232922.log
├── 20210815_232922.log.json
├── 20210816_103133.log
├── 20210816_103133.log.json
└── fcn_litehr18-without-head_512x512_160k_ade20k.py
├── fcn_litehr30-with-head_512x1024_8x2_160k_cityscapes
├── 20210814_053600.log
├── 20210814_053600.log.json
└── fcn_litehr30-with-head_512x1024_8x2_160k_cityscapes.py
├── fcn_litehr30-with-head_512x1024_8x2_320k_cityscapes
├── 20210818_221830.log
├── 20210818_221830.log.json
└── fcn_litehr30-with-head_512x1024_8x2_320k_cityscapes.py
├── fcn_litehr30-with-head_512x1024_8x2_640k_cityscapes
├── 20210826_205643.log
├── 20210826_205643.log.json
└── fcn_litehr30-with-head_512x1024_8x2_640k_cityscapes.py
├── fcn_litehr30-with-head_512x512_160k_ade20k
├── 20210816_133101.log
├── 20210816_133101.log.json
└── fcn_litehr30-with-head_512x512_160k_ade20k.py
├── fcn_litehr30-without-head_512x1024_8x2_160k_cityscapes
├── 20210814_011910.log
├── 20210814_011910.log.json
└── fcn_litehr30-without-head_512x1024_8x2_160k_cityscapes.py
├── fcn_litehr30-without-head_512x1024_8x2_320k_cityscapes
├── 20210819_110256.log
├── 20210819_110256.log.json
└── fcn_litehr30-without-head_512x1024_8x2_320k_cityscapes.py
├── fcn_litehr30-without-head_512x1024_8x2_640k_cityscapes
├── 20210828_065543.log
├── 20210828_065543.log.json
└── fcn_litehr30-without-head_512x1024_8x2_640k_cityscapes.py
└── fcn_litehr30-without-head_512x512_160k_ade20k
├── 20210816_121228.log
├── 20210816_121228.log.json
└── fcn_litehr30-without-head_512x512_160k_ade20k.py
/.github/ISSUE_TEMPLATE/config.yml:
--------------------------------------------------------------------------------
1 | blank_issues_enabled: false
2 |
3 | contact_links:
4 | - name: MMSegmentation Documentation
5 | url: https://mmsegmentation.readthedocs.io
6 | about: Check the docs and FAQ to see if you question is already anwsered.
7 |
--------------------------------------------------------------------------------
/.github/ISSUE_TEMPLATE/general_questions.md:
--------------------------------------------------------------------------------
1 | ---
2 | name: General questions
3 | about: Ask general questions to get help
4 | title: ''
5 | labels: ''
6 | assignees: ''
7 |
8 | ---
9 |
--------------------------------------------------------------------------------
/.readthedocs.yml:
--------------------------------------------------------------------------------
1 | version: 2
2 |
3 | formats: all
4 |
5 | python:
6 | version: 3.7
7 | install:
8 | - requirements: requirements/docs.txt
9 | - requirements: requirements/readthedocs.txt
10 |
--------------------------------------------------------------------------------
/MANIFEST.in:
--------------------------------------------------------------------------------
1 | include requirements/*.txt
2 | include mmseg/.mim/model-index.yml
3 | recursive-include mmseg/.mim/configs *.py *.yml
4 | recursive-include mmseg/.mim/tools *.py *.sh
5 |
--------------------------------------------------------------------------------
/configs/_base_/datasets/pascal_voc12_aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './pascal_voc12.py'
2 | # dataset settings
3 | data = dict(
4 | train=dict(
5 | ann_dir=['SegmentationClass', 'SegmentationClassAug'],
6 | split=[
7 | 'ImageSets/Segmentation/train.txt',
8 | 'ImageSets/Segmentation/aug.txt'
9 | ]))
10 |
--------------------------------------------------------------------------------
/configs/_base_/default_runtime.py:
--------------------------------------------------------------------------------
1 | # yapf:disable
2 | log_config = dict(
3 | interval=50,
4 | hooks=[
5 | dict(type='TextLoggerHook', by_epoch=False),
6 | # dict(type='TensorboardLoggerHook')
7 | ])
8 | # yapf:enable
9 | dist_params = dict(backend='nccl')
10 | log_level = 'INFO'
11 | load_from = None
12 | resume_from = None
13 | workflow = [('train', 1)]
14 | cudnn_benchmark = True
15 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn-resize-concat_litehr18-with-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn-resize-concat_litehr18-without-head.py'
3 | model = dict(
4 | backbone=dict(extra=dict(with_head=True)),
5 | decode_head=dict(
6 | in_channels=[40, 40, 80, 160],
7 | channels=sum([40, 40, 80, 160]),
8 | ))
9 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn-resize-concat_litehr30-with-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn-resize-concat_litehr18-with-head.py'
3 | model = dict(
4 | backbone=dict(extra=dict(stages_spec=dict(num_modules=(3, 8, 3)))))
5 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn-resize-concat_litehr30-without-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn-resize-concat_litehr18-without-head.py'
3 | model = dict(
4 | backbone=dict(extra=dict(stages_spec=dict(num_modules=(3, 8, 3)))))
5 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn_litehr18-with-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn_litehr18-without-head.py'
3 | model = dict(backbone=dict(extra=dict(with_head=True)))
4 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn_litehr18-without-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn-resize-concat_litehr18-without-head.py'
3 | model = dict(
4 | decode_head=dict(
5 | in_channels=40,
6 | in_index=0,
7 | channels=40,
8 | input_transform=None,
9 | kernel_size=3,
10 | num_convs=2,
11 | concat_input=True,
12 | dropout_ratio=0.1))
13 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn_litehr30-with-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn_litehr18-with-head.py'
3 | model = dict(
4 | backbone=dict(extra=dict(stages_spec=dict(num_modules=(3, 8, 3)))))
5 |
--------------------------------------------------------------------------------
/configs/_base_/models/fcn_litehr30-without-head.py:
--------------------------------------------------------------------------------
1 | # model settings
2 | _base_ = './fcn_litehr18-without-head.py'
3 | model = dict(
4 | backbone=dict(extra=dict(stages_spec=dict(num_modules=(3, 8, 3)))))
5 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_160k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=160000)
8 | checkpoint_config = dict(by_epoch=False, interval=16000)
9 | evaluation = dict(interval=16000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_20k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=20000)
8 | checkpoint_config = dict(by_epoch=False, interval=2000)
9 | evaluation = dict(interval=2000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_320k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=320000)
8 | checkpoint_config = dict(by_epoch=False, interval=32000)
9 | evaluation = dict(interval=32000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_40k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=40000)
8 | checkpoint_config = dict(by_epoch=False, interval=4000)
9 | evaluation = dict(interval=4000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_640k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=640000)
8 | checkpoint_config = dict(by_epoch=False, interval=64000)
9 | evaluation = dict(interval=64000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/_base_/schedules/schedule_80k.py:
--------------------------------------------------------------------------------
1 | # optimizer
2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005)
3 | optimizer_config = dict()
4 | # learning policy
5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False)
6 | # runtime settings
7 | runner = dict(type='IterBasedRunner', max_iters=80000)
8 | checkpoint_config = dict(by_epoch=False, interval=8000)
9 | evaluation = dict(interval=8000, metric='mIoU')
10 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ann_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_20k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
7 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
7 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/ann/ann_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ann_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './apcnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/apcnet/apcnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/apcnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ccnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/ccnet/ccnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ccnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './danet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/danet/danet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/danet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d16-mg124_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet101_v1c',
4 | backbone=dict(
5 | depth=101,
6 | dilations=(1, 1, 1, 2),
7 | strides=(1, 2, 2, 1),
8 | multi_grid=(1, 2, 4)),
9 | decode_head=dict(
10 | dilations=(1, 6, 12, 18),
11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000)))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d16-mg124_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet101_v1c',
4 | backbone=dict(
5 | depth=101,
6 | dilations=(1, 1, 1, 2),
7 | strides=(1, 2, 2, 1),
8 | multi_grid=(1, 2, 4)),
9 | decode_head=dict(
10 | dilations=(1, 6, 12, 18),
11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000)))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_480x480_40k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_480x480_40k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_480x480_80k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_480x480_80k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r101b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r18-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r18-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r18b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r18b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3/deeplabv3_r50b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d16-mg124_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet101_v1c',
4 | backbone=dict(
5 | depth=101,
6 | dilations=(1, 1, 1, 2),
7 | strides=(1, 2, 2, 1),
8 | multi_grid=(1, 2, 4)),
9 | decode_head=dict(
10 | dilations=(1, 6, 12, 18),
11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000)))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d16-mg124_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet101_v1c',
4 | backbone=dict(
5 | depth=101,
6 | dilations=(1, 1, 1, 2),
7 | strides=(1, 2, 2, 1),
8 | multi_grid=(1, 2, 4)),
9 | decode_head=dict(
10 | dilations=(1, 6, 12, 18),
11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000)))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_480x480_40k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_480x480_80k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | c1_in_channels=64,
7 | c1_channels=12,
8 | in_channels=512,
9 | channels=128,
10 | ),
11 | auxiliary_head=dict(in_channels=256, channels=64))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r18-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | c1_in_channels=64,
7 | c1_channels=12,
8 | in_channels=512,
9 | channels=128,
10 | ),
11 | auxiliary_head=dict(in_channels=256, channels=64))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | c1_in_channels=64,
7 | c1_channels=12,
8 | in_channels=512,
9 | channels=128,
10 | ),
11 | auxiliary_head=dict(in_channels=256, channels=64))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | c1_in_channels=64,
7 | c1_channels=12,
8 | in_channels=512,
9 | channels=128,
10 | ),
11 | auxiliary_head=dict(in_channels=256, channels=64))
12 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3plus_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/deeplabv3plus/deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dmnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/dmnet/dmnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dmnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './dnl_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/dnlnet/dnl_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/dnl_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/emanet/emanet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './emanet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/emanet/emanet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './emanet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/emanet/emanet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/emanet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/emanet/emanet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/emanet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './encnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/encnet/encnet_r50s-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | backbone=dict(stem_channels=128),
7 | decode_head=dict(num_classes=150),
8 | auxiliary_head=dict(num_classes=150))
9 |
--------------------------------------------------------------------------------
/configs/fastscnn/fast_scnn_lr0.12_8x4_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fast_scnn.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 |
6 | # Re-config the data sampler.
7 | data = dict(samples_per_gpu=4, workers_per_gpu=4)
8 |
9 | # Re-config the optimizer.
10 | optimizer = dict(type='SGD', lr=0.12, momentum=0.9, weight_decay=4e-5)
11 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101-d16_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101-d16_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101-d16_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101-d16_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101b-d16_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50b-d16_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r101b-d16_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50b-d16_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r50-d16_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | backbone=dict(dilations=(1, 1, 1, 2), strides=(1, 2, 2, 1)),
7 | decode_head=dict(dilation=6),
8 | auxiliary_head=dict(dilation=6))
9 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r50-d16_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | backbone=dict(dilations=(1, 1, 1, 2), strides=(1, 2, 2, 1)),
7 | decode_head=dict(dilation=6),
8 | auxiliary_head=dict(dilation=6))
9 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r50b-d16_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_d6_r50b-d16_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_d6_r50-d16_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_480x480_40k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_480x480_40k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_480x480_80k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_480x480_80k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r101b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r18-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r18-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r18b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r18b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_context.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=60),
7 | auxiliary_head=dict(num_classes=60),
8 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
9 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_context.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=60),
7 | auxiliary_head=dict(num_classes=60),
8 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
9 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_20k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
7 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
7 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/fcn/fcn_r50b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/fp16/deeplabv3_r101-d8_512x1024_80k_fp16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py'
2 | # fp16 settings
3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
4 | # fp16 placeholder
5 | fp16 = dict()
6 |
--------------------------------------------------------------------------------
/configs/fp16/deeplabv3plus_r101-d8_512x1024_80k_fp16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3plus/deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py'
2 | # fp16 settings
3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
4 | # fp16 placeholder
5 | fp16 = dict()
6 |
--------------------------------------------------------------------------------
/configs/fp16/fcn_r101-d8_512x1024_80k_fp16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../fcn/fcn_r101-d8_512x1024_80k_cityscapes.py'
2 | # fp16 settings
3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
4 | # fp16 placeholder
5 | fp16 = dict()
6 |
--------------------------------------------------------------------------------
/configs/fp16/pspnet_r101-d8_512x1024_80k_fp16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py'
2 | # fp16 settings
3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
4 | # fp16 placeholder
5 | fp16 = dict()
6 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './gcnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/gcnet/gcnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/gcnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_context.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=60),
7 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
8 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
9 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_context_59.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=59),
7 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
8 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
9 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_context.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=60),
7 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
8 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
9 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_context_59.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=59),
7 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
8 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
9 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x1024_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(decode_head=dict(num_classes=150))
6 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_20k.py'
4 | ]
5 | model = dict(decode_head=dict(num_classes=21))
6 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_voc12_aug.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(decode_head=dict(num_classes=21))
6 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(decode_head=dict(num_classes=150))
6 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_40k_pascal_context.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_40k_pascal_context_59.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_80k_pascal_context.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_80k_pascal_context_59.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x1024_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_160k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_40k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_20k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_40k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr18s_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_80k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_40k_pascal_context.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_40k_pascal_context_59.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_80k_pascal_context.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_480x480_80k_pascal_context_59.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x1024_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_160k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_40k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_20k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_40k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/hrnet/fcn_hr48_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fcn_hr18_512x512_80k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w48',
4 | backbone=dict(
5 | extra=dict(
6 | stage2=dict(num_channels=(48, 96)),
7 | stage3=dict(num_channels=(48, 96, 192)),
8 | stage4=dict(num_channels=(48, 96, 192, 384)))),
9 | decode_head=dict(
10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384])))
11 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr18-with-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr18-with-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr18-without-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr18-without-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr30-with-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr30-with-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr30-without-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn-resize-concat_litehr30-without-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr18-with-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr18-with-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr18-without-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr18-without-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr30-with-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr30-with-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr30-without-head_512x1024_8x2_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet/fcn_litehr30-without-head_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_160k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr18-with-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr18-with-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr18-without-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr18-without-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr30-with-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr30-with-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr30-without-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn-resize-concat_litehr30-without-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr18-with-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr18-with-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr18-without-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr18-without-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr30-with-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr30-with-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr30-without-head_512x1024_8x2_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_320k/fcn_litehr30-without-head_512x512_320k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_320k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr18-with-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr18-with-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr18-without-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr18-without-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr30-with-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr30-with-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr30-without-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn-resize-concat_litehr30-without-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn-resize-concat_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr18-with-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr18-with-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr18-without-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr18-without-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr18-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr30-with-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr30-with-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-with-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr30-without-head_512x1024_8x2_640k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | find_unused_parameters = True
7 |
--------------------------------------------------------------------------------
/configs/litehrnet_640k/fcn_litehr30-without-head_512x512_640k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_litehr30-without-head.py',
3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_640k.py'
5 | ]
6 | model = dict(decode_head=dict(num_classes=150))
7 | find_unused_parameters = True
8 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/deeplabv3_m-v2-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3/deeplabv3_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/fcn_m-v2-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../fcn/fcn_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/fcn_m-v2-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../fcn/fcn_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/pspnet_m-v2-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v2/pspnet_m-v2-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../pspnet/pspnet_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='mmcls://mobilenet_v2',
4 | backbone=dict(
5 | _delete_=True,
6 | type='MobileNetV2',
7 | widen_factor=1.,
8 | strides=(1, 2, 2, 1, 1, 1, 1),
9 | dilations=(1, 1, 1, 2, 2, 4, 4),
10 | out_indices=(1, 2, 4, 6)),
11 | decode_head=dict(in_channels=320),
12 | auxiliary_head=dict(in_channels=96))
13 |
--------------------------------------------------------------------------------
/configs/mobilenet_v3/lraspp_m-v3-d8_512x1024_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 |
6 | model = dict(pretrained='open-mmlab://contrib/mobilenet_v3_large')
7 |
8 | # Re-config the data sampler.
9 | data = dict(samples_per_gpu=4, workers_per_gpu=4)
10 |
11 | runner = dict(type='IterBasedRunner', max_iters=320000)
12 |
--------------------------------------------------------------------------------
/configs/mobilenet_v3/lraspp_m-v3-d8_scratch_512x1024_320k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 |
6 | # Re-config the data sampler.
7 | data = dict(samples_per_gpu=4, workers_per_gpu=4)
8 |
9 | runner = dict(type='IterBasedRunner', max_iters=320000)
10 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './nonlocal_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/nonlocal_net/nonlocal_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/nonlocal_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18_512x1024_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x1024_160k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x1024_160k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x1024_40k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x512_20k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x512_40k_voc12aug.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_hr18s_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './ocrnet_hr18_512x512_80k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://msra/hrnetv2_w18_small',
4 | backbone=dict(
5 | extra=dict(
6 | stage1=dict(num_blocks=(2, )),
7 | stage2=dict(num_blocks=(2, 2)),
8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))
10 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
6 | optimizer = dict(lr=0.02)
7 | lr_config = dict(min_lr=2e-4)
8 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
6 |
--------------------------------------------------------------------------------
/configs/ocrnet/ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/ocrnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
6 | optimizer = dict(lr=0.02)
7 | lr_config = dict(min_lr=2e-4)
8 |
--------------------------------------------------------------------------------
/configs/point_rend/pointrend_r101_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pointrend_r50_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/point_rend/pointrend_r101_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './pointrend_r50_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/point_rend/pointrend_r50_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pointrend_r50.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | lr_config = dict(warmup='linear', warmup_iters=200)
6 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './psanet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(mask_size=(66, 66), num_classes=150),
7 | auxiliary_head=dict(num_classes=150))
8 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(mask_size=(66, 66), num_classes=150),
7 | auxiliary_head=dict(num_classes=150))
8 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/psanet/psanet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/psanet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_480x480_40k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_480x480_40k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_480x480_80k_pascal_context.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_480x480_80k_pascal_context_59.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r101b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet101',
4 | backbone=dict(type='ResNet', depth=101))
5 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r18-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r18-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnet18_v1c',
4 | backbone=dict(depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r18b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r18b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='torchvision://resnet18',
4 | backbone=dict(type='ResNet', depth=18),
5 | decode_head=dict(
6 | in_channels=512,
7 | channels=128,
8 | ),
9 | auxiliary_head=dict(in_channels=256, channels=64))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_480x480_40k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_480x480_40k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_480x480_80k_pascal_context.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_context.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=60),
8 | auxiliary_head=dict(num_classes=60),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_480x480_80k_pascal_context_59.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_context_59.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=59),
8 | auxiliary_head=dict(num_classes=59),
9 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320)))
10 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001)
11 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_r50-d8.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50b-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/pspnet/pspnet_r50b-d8_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet'))
3 |
--------------------------------------------------------------------------------
/configs/resnest/deeplabv3_s101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/deeplabv3_s101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3/deeplabv3_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/deeplabv3plus_s101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3plus/deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/deeplabv3plus_s101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../deeplabv3plus/deeplabv3plus_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/fcn_s101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../fcn/fcn_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/fcn_s101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../fcn/fcn_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/pspnet_s101-d8_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = '../pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/resnest/pspnet_s101-d8_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = '../pspnet/pspnet_r101-d8_512x512_160k_ade20k.py'
2 | model = dict(
3 | pretrained='open-mmlab://resnest101',
4 | backbone=dict(
5 | type='ResNeSt',
6 | stem_channels=128,
7 | radix=2,
8 | reduction_factor=4,
9 | avg_down_stride=True))
10 |
--------------------------------------------------------------------------------
/configs/sem_fpn/fpn_r101_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './fpn_r50_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/sem_fpn/fpn_r101_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './fpn_r50_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/sem_fpn/fpn_r50_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fpn_r50.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/sem_fpn/fpn_r50_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fpn_r50.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(decode_head=dict(num_classes=150))
6 |
--------------------------------------------------------------------------------
/configs/setr/setr_mla_512x512_160k_b16_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = ['./setr_mla_512x512_160k_b8_ade20k.py']
2 |
3 | # num_gpus: 8 -> batch_size: 16
4 | data = dict(samples_per_gpu=2)
5 |
--------------------------------------------------------------------------------
/configs/swin/upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | './upernet_swin_base_patch4_window12_512x512_160k_ade20k_'
3 | 'pretrain_384x384_1K.py'
4 | ]
5 | model = dict(
6 | pretrained=\
7 | 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384_22k.pth', # noqa
8 | )
9 |
--------------------------------------------------------------------------------
/configs/swin/upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | './upernet_swin_base_patch4_window7_512x512_160k_ade20k_'
3 | 'pretrain_224x224_1K.py'
4 | ]
5 | model = dict(
6 | pretrained=\
7 | 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window7_224_22k.pth', # noqa
8 | )
9 |
--------------------------------------------------------------------------------
/configs/unet/deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_unet_s5-d16.py',
3 | '../_base_/datasets/chase_db1.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
7 | evaluation = dict(metric='mDice')
8 |
--------------------------------------------------------------------------------
/configs/unet/deeplabv3_unet_s5-d16_128x128_40k_stare.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_unet_s5-d16.py', '../_base_/datasets/stare.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/deeplabv3_unet_s5-d16_256x256_40k_hrf.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_unet_s5-d16.py', '../_base_/datasets/hrf.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(256, 256), stride=(170, 170)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/deeplabv3_unet_s5-d16_64x64_40k_drive.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/deeplabv3_unet_s5-d16.py', '../_base_/datasets/drive.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(64, 64), stride=(42, 42)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/fcn_unet_s5-d16_128x128_40k_chase_db1.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/chase_db1.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/fcn_unet_s5-d16_128x128_40k_stare.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/stare.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/fcn_unet_s5-d16_256x256_40k_hrf.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/hrf.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(256, 256), stride=(170, 170)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/fcn_unet_s5-d16_64x64_40k_drive.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/fcn_unet_s5-d16.py', '../_base_/datasets/drive.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(64, 64), stride=(42, 42)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/pspnet_unet_s5-d16_128x128_40k_chase_db1.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_unet_s5-d16.py',
3 | '../_base_/datasets/chase_db1.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
7 | evaluation = dict(metric='mDice')
8 |
--------------------------------------------------------------------------------
/configs/unet/pspnet_unet_s5-d16_128x128_40k_stare.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_unet_s5-d16.py', '../_base_/datasets/stare.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(128, 128), stride=(85, 85)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/pspnet_unet_s5-d16_256x256_40k_hrf.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_unet_s5-d16.py', '../_base_/datasets/hrf.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(256, 256), stride=(170, 170)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/unet/pspnet_unet_s5-d16_64x64_40k_drive.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/pspnet_unet_s5-d16.py', '../_base_/datasets/drive.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 | model = dict(test_cfg=dict(crop_size=(64, 64), stride=(42, 42)))
6 | evaluation = dict(metric='mDice')
7 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x1024_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x1024_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x512_160k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x512_20k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x512_40k_voc12aug.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_512x512_80k_ade20k.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_769x769_40k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r101_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_r50_769x769_80k_cityscapes.py'
2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101))
3 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x1024_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x1024_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/cityscapes.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x512_20k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_20k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x512_40k_voc12aug.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py',
3 | '../_base_/datasets/pascal_voc12_aug.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(num_classes=21), auxiliary_head=dict(num_classes=21))
8 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/ade20k.py',
3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
4 | ]
5 | model = dict(
6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150))
7 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_769x769_40k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_40k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/upernet/upernet_r50_769x769_80k_cityscapes.py:
--------------------------------------------------------------------------------
1 | _base_ = [
2 | '../_base_/models/upernet_r50.py',
3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py',
4 | '../_base_/schedules/schedule_80k.py'
5 | ]
6 | model = dict(
7 | decode_head=dict(align_corners=True),
8 | auxiliary_head=dict(align_corners=True),
9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513)))
10 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-b16_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_base_patch16_224-b5f2ef4d.pth', # noqa
5 | backbone=dict(drop_path_rate=0.1),
6 | neck=None) # yapf: disable
7 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-b16_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_80k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_base_patch16_224-b5f2ef4d.pth', # noqa
5 | backbone=dict(drop_path_rate=0.1),
6 | neck=None) # yapf: disable
7 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-b16_ln_mln_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_base_patch16_224-b5f2ef4d.pth', # noqa
5 | backbone=dict(drop_path_rate=0.1, final_norm=True)) # yapf: disable
6 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-b16_mln_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_base_patch16_224-b5f2ef4d.pth', # noqa
5 | backbone=dict(drop_path_rate=0.1),) # yapf: disable
6 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-s16_512x512_160k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_small_patch16_224-cd65a155.pth', # noqa
5 | backbone=dict(num_heads=6, embed_dims=384, drop_path_rate=0.1),
6 | decode_head=dict(num_classes=150, in_channels=[384, 384, 384, 384]),
7 | neck=None,
8 | auxiliary_head=dict(num_classes=150, in_channels=384)) # yapf: disable
9 |
--------------------------------------------------------------------------------
/configs/vit/upernet_deit-s16_512x512_80k_ade20k.py:
--------------------------------------------------------------------------------
1 | _base_ = './upernet_vit-b16_mln_512x512_80k_ade20k.py'
2 |
3 | model = dict(
4 | pretrained='https://dl.fbaipublicfiles.com/deit/deit_small_patch16_224-cd65a155.pth', # noqa
5 | backbone=dict(num_heads=6, embed_dims=384, drop_path_rate=0.1),
6 | decode_head=dict(num_classes=150, in_channels=[384, 384, 384, 384]),
7 | neck=None,
8 | auxiliary_head=dict(num_classes=150, in_channels=384)) # yapf: disable
9 |
--------------------------------------------------------------------------------
/demo/demo.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/demo/demo.png
--------------------------------------------------------------------------------
/docker/serve/config.properties:
--------------------------------------------------------------------------------
1 | inference_address=http://0.0.0.0:8080
2 | management_address=http://0.0.0.0:8081
3 | metrics_address=http://0.0.0.0:8082
4 | model_store=/home/model-server/model-store
5 | load_models=all
6 |
--------------------------------------------------------------------------------
/docker/serve/entrypoint.sh:
--------------------------------------------------------------------------------
1 | #!/bin/bash
2 | set -e
3 |
4 | if [[ "$1" = "serve" ]]; then
5 | shift 1
6 | torchserve --start --ts-config /home/model-server/config.properties
7 | else
8 | eval "$@"
9 | fi
10 |
11 | # prevent docker exit
12 | tail -f /dev/null
13 |
--------------------------------------------------------------------------------
/docs/switch_language.md:
--------------------------------------------------------------------------------
1 | ## English
2 |
3 | ## 简体中文
4 |
--------------------------------------------------------------------------------
/docs/tutorials/index.rst:
--------------------------------------------------------------------------------
1 | .. toctree::
2 | :maxdepth: 2
3 |
4 | config.md
5 | customize_datasets.md
6 | data_pipeline.md
7 | customize_models.md
8 | training_tricks.md
9 | customize_runtime.md
10 |
--------------------------------------------------------------------------------
/docs_zh-CN/imgs/qq_group_qrcode.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/docs_zh-CN/imgs/qq_group_qrcode.jpg
--------------------------------------------------------------------------------
/docs_zh-CN/imgs/seggroup_qrcode.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/docs_zh-CN/imgs/seggroup_qrcode.jpg
--------------------------------------------------------------------------------
/docs_zh-CN/imgs/zhihu_qrcode.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/docs_zh-CN/imgs/zhihu_qrcode.jpg
--------------------------------------------------------------------------------
/docs_zh-CN/switch_language.md:
--------------------------------------------------------------------------------
1 | ## English
2 |
3 | ## 简体中文
4 |
--------------------------------------------------------------------------------
/docs_zh-CN/tutorials/index.rst:
--------------------------------------------------------------------------------
1 | .. toctree::
2 | :maxdepth: 2
3 |
4 | config.md
5 | customize_datasets.md
6 | data_pipeline.md
7 | customize_models.md
8 | training_tricks.md
9 | customize_runtime.md
10 |
--------------------------------------------------------------------------------
/mmseg/.mim/configs:
--------------------------------------------------------------------------------
1 | ../../configs
--------------------------------------------------------------------------------
/mmseg/.mim/model-index.yml:
--------------------------------------------------------------------------------
1 | ../../model-index.yml
--------------------------------------------------------------------------------
/mmseg/.mim/tools:
--------------------------------------------------------------------------------
1 | ../../tools
--------------------------------------------------------------------------------
/mmseg/apis/__init__.py:
--------------------------------------------------------------------------------
1 | from .inference import inference_segmentor, init_segmentor, show_result_pyplot
2 | from .test import multi_gpu_test, single_gpu_test
3 | from .train import get_root_logger, set_random_seed, train_segmentor
4 |
5 | __all__ = [
6 | 'get_root_logger', 'set_random_seed', 'train_segmentor', 'init_segmentor',
7 | 'inference_segmentor', 'multi_gpu_test', 'single_gpu_test',
8 | 'show_result_pyplot'
9 | ]
10 |
--------------------------------------------------------------------------------
/mmseg/core/__init__.py:
--------------------------------------------------------------------------------
1 | from .evaluation import * # noqa: F401, F403
2 | from .seg import * # noqa: F401, F403
3 | from .utils import * # noqa: F401, F403
4 |
--------------------------------------------------------------------------------
/mmseg/core/evaluation/__init__.py:
--------------------------------------------------------------------------------
1 | from .class_names import get_classes, get_palette
2 | from .eval_hooks import DistEvalHook, EvalHook
3 | from .metrics import eval_metrics, mean_dice, mean_fscore, mean_iou
4 |
5 | __all__ = [
6 | 'EvalHook', 'DistEvalHook', 'mean_dice', 'mean_iou', 'mean_fscore',
7 | 'eval_metrics', 'get_classes', 'get_palette'
8 | ]
9 |
--------------------------------------------------------------------------------
/mmseg/core/seg/__init__.py:
--------------------------------------------------------------------------------
1 | from .builder import build_pixel_sampler
2 | from .sampler import BasePixelSampler, OHEMPixelSampler
3 |
4 | __all__ = ['build_pixel_sampler', 'BasePixelSampler', 'OHEMPixelSampler']
5 |
--------------------------------------------------------------------------------
/mmseg/core/seg/builder.py:
--------------------------------------------------------------------------------
1 | from mmcv.utils import Registry, build_from_cfg
2 |
3 | PIXEL_SAMPLERS = Registry('pixel sampler')
4 |
5 |
6 | def build_pixel_sampler(cfg, **default_args):
7 | """Build pixel sampler for segmentation map."""
8 | return build_from_cfg(cfg, PIXEL_SAMPLERS, default_args)
9 |
--------------------------------------------------------------------------------
/mmseg/core/seg/sampler/__init__.py:
--------------------------------------------------------------------------------
1 | from .base_pixel_sampler import BasePixelSampler
2 | from .ohem_pixel_sampler import OHEMPixelSampler
3 |
4 | __all__ = ['BasePixelSampler', 'OHEMPixelSampler']
5 |
--------------------------------------------------------------------------------
/mmseg/core/seg/sampler/base_pixel_sampler.py:
--------------------------------------------------------------------------------
1 | from abc import ABCMeta, abstractmethod
2 |
3 |
4 | class BasePixelSampler(metaclass=ABCMeta):
5 | """Base class of pixel sampler."""
6 |
7 | def __init__(self, **kwargs):
8 | pass
9 |
10 | @abstractmethod
11 | def sample(self, seg_logit, seg_label):
12 | """Placeholder for sample function."""
13 |
--------------------------------------------------------------------------------
/mmseg/core/utils/__init__.py:
--------------------------------------------------------------------------------
1 | from .misc import add_prefix
2 |
3 | __all__ = ['add_prefix']
4 |
--------------------------------------------------------------------------------
/mmseg/core/utils/misc.py:
--------------------------------------------------------------------------------
1 | def add_prefix(inputs, prefix):
2 | """Add prefix for dict.
3 |
4 | Args:
5 | inputs (dict): The input dict with str keys.
6 | prefix (str): The prefix to add.
7 |
8 | Returns:
9 |
10 | dict: The dict with keys updated with ``prefix``.
11 | """
12 |
13 | outputs = dict()
14 | for name, value in inputs.items():
15 | outputs[f'{prefix}.{name}'] = value
16 |
17 | return outputs
18 |
--------------------------------------------------------------------------------
/mmseg/models/necks/__init__.py:
--------------------------------------------------------------------------------
1 | from .fpn import FPN
2 | from .mla_neck import MLANeck
3 | from .multilevel_neck import MultiLevelNeck
4 |
5 | __all__ = ['FPN', 'MultiLevelNeck', 'MLANeck']
6 |
--------------------------------------------------------------------------------
/mmseg/models/segmentors/__init__.py:
--------------------------------------------------------------------------------
1 | from .base import BaseSegmentor
2 | from .cascade_encoder_decoder import CascadeEncoderDecoder
3 | from .encoder_decoder import EncoderDecoder
4 |
5 | __all__ = ['BaseSegmentor', 'EncoderDecoder', 'CascadeEncoderDecoder']
6 |
--------------------------------------------------------------------------------
/mmseg/ops/__init__.py:
--------------------------------------------------------------------------------
1 | from .encoding import Encoding
2 | from .wrappers import Upsample, resize
3 |
4 | __all__ = ['Upsample', 'resize', 'Encoding']
5 |
--------------------------------------------------------------------------------
/mmseg/utils/__init__.py:
--------------------------------------------------------------------------------
1 | from .collect_env import collect_env
2 | from .logger import get_root_logger
3 |
4 | __all__ = ['get_root_logger', 'collect_env']
5 |
--------------------------------------------------------------------------------
/pytest.ini:
--------------------------------------------------------------------------------
1 | [pytest]
2 | addopts = --xdoctest --xdoctest-style=auto
3 | norecursedirs = .git ignore build __pycache__ data docker docs .eggs
4 |
5 | filterwarnings= default
6 | ignore:.*No cfgstr given in Cacher constructor or call.*:Warning
7 | ignore:.*Define the __nice__ method for.*:Warning
8 |
--------------------------------------------------------------------------------
/requirements.txt:
--------------------------------------------------------------------------------
1 | -r requirements/optional.txt
2 | -r requirements/runtime.txt
3 | -r requirements/tests.txt
4 |
--------------------------------------------------------------------------------
/requirements/docs.txt:
--------------------------------------------------------------------------------
1 | recommonmark
2 | sphinx
3 | sphinx_markdown_tables
4 | sphinx_rtd_theme
5 |
--------------------------------------------------------------------------------
/requirements/mminstall.txt:
--------------------------------------------------------------------------------
1 | mmcv-full>=1.3.1,<=1.4.0
2 |
--------------------------------------------------------------------------------
/requirements/optional.txt:
--------------------------------------------------------------------------------
1 | cityscapesscripts
2 |
--------------------------------------------------------------------------------
/requirements/readthedocs.txt:
--------------------------------------------------------------------------------
1 | mmcv
2 | prettytable
3 | torch
4 | torchvision
5 |
--------------------------------------------------------------------------------
/requirements/runtime.txt:
--------------------------------------------------------------------------------
1 | matplotlib
2 | numpy
3 | packaging
4 | prettytable
5 |
--------------------------------------------------------------------------------
/requirements/tests.txt:
--------------------------------------------------------------------------------
1 | codecov
2 | flake8
3 | interrogate
4 | isort==4.3.21
5 | pytest
6 | xdoctest>=0.10.0
7 | yapf
8 |
--------------------------------------------------------------------------------
/resources/3dogs.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/resources/3dogs.jpg
--------------------------------------------------------------------------------
/resources/3dogs_mask.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/resources/3dogs_mask.png
--------------------------------------------------------------------------------
/resources/mmseg-logo.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/resources/mmseg-logo.png
--------------------------------------------------------------------------------
/resources/seg_demo.gif:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/resources/seg_demo.gif
--------------------------------------------------------------------------------
/setup.cfg:
--------------------------------------------------------------------------------
1 | [yapf]
2 | based_on_style = pep8
3 | blank_line_before_nested_class_or_def = true
4 | split_before_expression_after_opening_paren = true
5 |
6 | [isort]
7 | line_length = 79
8 | multi_line_output = 0
9 | known_standard_library = setuptools
10 | known_first_party = mmseg
11 | known_third_party = PIL,cityscapesscripts,cv2,detail,matplotlib,mmcv,numpy,onnxruntime,oss2,packaging,prettytable,pytest,scipy,seaborn,torch,ts
12 | no_lines_before = STDLIB,LOCALFOLDER
13 | default_section = THIRDPARTY
14 |
--------------------------------------------------------------------------------
/tests/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/__init__.py
--------------------------------------------------------------------------------
/tests/data/color.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/color.jpg
--------------------------------------------------------------------------------
/tests/data/gray.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/gray.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/gts/00000_gt.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/gts/00000_gt.png
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/gts/00001_gt.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/gts/00001_gt.png
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/gts/00002_gt.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/gts/00002_gt.png
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/gts/00003_gt.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/gts/00003_gt.png
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/gts/00004_gt.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/gts/00004_gt.png
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/imgs/00000_img.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/imgs/00000_img.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/imgs/00001_img.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/imgs/00001_img.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/imgs/00002_img.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/imgs/00002_img.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/imgs/00003_img.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/imgs/00003_img.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/imgs/00004_img.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/pseudo_dataset/imgs/00004_img.jpg
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/splits/train.txt:
--------------------------------------------------------------------------------
1 | 00000
2 | 00001
3 | 00002
4 | 00003
5 |
--------------------------------------------------------------------------------
/tests/data/pseudo_dataset/splits/val.txt:
--------------------------------------------------------------------------------
1 | 00004
2 |
--------------------------------------------------------------------------------
/tests/data/seg.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/data/seg.png
--------------------------------------------------------------------------------
/tests/test_models/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/test_models/__init__.py
--------------------------------------------------------------------------------
/tests/test_models/test_backbones/__init__.py:
--------------------------------------------------------------------------------
1 | from .utils import all_zeros, check_norm_state, is_block, is_norm
2 |
3 | __all__ = ['is_norm', 'is_block', 'all_zeros', 'check_norm_state']
4 |
--------------------------------------------------------------------------------
/tests/test_models/test_heads/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/test_models/test_heads/__init__.py
--------------------------------------------------------------------------------
/tests/test_models/test_losses/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/test_models/test_losses/__init__.py
--------------------------------------------------------------------------------
/tests/test_models/test_necks/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/test_models/test_necks/__init__.py
--------------------------------------------------------------------------------
/tests/test_models/test_segmentors/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Junjun2016/LiteHRNet/e2b13de52e970215be566067cab7bd880010f062/tests/test_models/test_segmentors/__init__.py
--------------------------------------------------------------------------------
/tools/dist_test.sh:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env bash
2 |
3 | CONFIG=$1
4 | CHECKPOINT=$2
5 | GPUS=$3
6 | PORT=${PORT:-29500}
7 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
8 | python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
9 | $(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
10 |
--------------------------------------------------------------------------------
/tools/dist_train.sh:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env bash
2 |
3 | CONFIG=$1
4 | GPUS=$2
5 | PORT=${PORT:-29500}
6 |
7 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
8 | python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
9 | $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3}
10 |
--------------------------------------------------------------------------------