├── LICENSE ├── Readme.md ├── assets ├── gf_method.png └── latency.png ├── classification ├── .gitignore ├── classification │ └── model.py ├── configs │ └── cls │ │ ├── mbv2 │ │ ├── filt_last1_r2.yaml │ │ ├── filt_last1_r4.yaml │ │ ├── filt_last1_r7.yaml │ │ ├── filt_last2_r2.yaml │ │ ├── filt_last2_r4.yaml │ │ ├── filt_last2_r7.yaml │ │ ├── filt_last3_r2.yaml │ │ ├── filt_last3_r4.yaml │ │ ├── filt_last3_r7.yaml │ │ ├── filt_last4_r2.yaml │ │ ├── filt_last4_r4.yaml │ │ └── filt_last4_r7.yaml │ │ ├── mcunet │ │ ├── filt_last1_r2.yaml │ │ ├── filt_last1_r4.yaml │ │ ├── filt_last1_r7.yaml │ │ ├── filt_last2_r2.yaml │ │ ├── filt_last2_r4.yaml │ │ ├── filt_last2_r7.yaml │ │ ├── filt_last3_r2.yaml │ │ ├── filt_last3_r4.yaml │ │ ├── filt_last3_r7.yaml │ │ ├── filt_last4_r2.yaml │ │ ├── filt_last4_r4.yaml │ │ └── filt_last4_r7.yaml │ │ ├── res18 │ │ ├── filt_last1_r2.yaml │ │ ├── filt_last1_r4.yaml │ │ ├── filt_last1_r7.yaml │ │ ├── filt_last2_r2.yaml │ │ ├── filt_last2_r4.yaml │ │ ├── filt_last2_r7.yaml │ │ ├── filt_last3_r2.yaml │ │ ├── filt_last3_r4.yaml │ │ ├── filt_last3_r7.yaml │ │ ├── filt_last4_r2.yaml │ │ ├── filt_last4_r4.yaml │ │ └── filt_last4_r7.yaml │ │ └── res34 │ │ ├── filt_last1_r2.yaml │ │ ├── filt_last1_r4.yaml │ │ ├── filt_last1_r7.yaml │ │ ├── filt_last2_r2.yaml │ │ ├── filt_last2_r4.yaml │ │ ├── filt_last2_r7.yaml │ │ ├── filt_last3_r2.yaml │ │ ├── filt_last3_r4.yaml │ │ ├── filt_last3_r7.yaml │ │ ├── filt_last4_r2.yaml │ │ ├── filt_last4_r4.yaml │ │ └── filt_last4_r7.yaml ├── custom_op │ ├── __init__.py │ ├── conv_avg.py │ └── register.py ├── data │ ├── cifar10 │ │ └── usr_group_15.29.npy │ ├── cifar100 │ │ └── usr_group_15.82.npy │ └── imagenet │ │ └── usr_group_13.10.npy ├── dataloader │ ├── __init__.py │ ├── pl_dataset.py │ └── sampling.py ├── models │ ├── LICENSE │ └── encoders │ │ ├── __init__.py │ │ ├── _base.py │ │ ├── _preprocessing.py │ │ ├── _utils.py │ │ ├── mcunet.py │ │ ├── mobilenet.py │ │ └── resnet.py ├── scripts │ ├── mbv2 │ │ ├── mbv2_cifar100_base.sh │ │ ├── mbv2_cifar100_r2.sh │ │ ├── mbv2_cifar100_r4.sh │ │ ├── mbv2_cifar100_r7.sh │ │ ├── mbv2_cifar10_base.sh │ │ ├── mbv2_cifar10_r2.sh │ │ ├── mbv2_cifar10_r4.sh │ │ ├── mbv2_cifar10_r7.sh │ │ ├── mbv2_imagenet_base_l2.sh │ │ ├── mbv2_imagenet_base_l4.sh │ │ ├── mbv2_imagenet_filt_l2.sh │ │ └── mbv2_imagenet_filt_l4.sh │ ├── mcunet │ │ ├── mcu_cifar100_base.sh │ │ ├── mcu_cifar100_r2.sh │ │ ├── mcu_cifar100_r4.sh │ │ ├── mcu_cifar100_r7.sh │ │ ├── mcu_cifar10_base.sh │ │ ├── mcu_cifar10_r2.sh │ │ ├── mcu_cifar10_r4.sh │ │ ├── mcu_cifar10_r7.sh │ │ ├── mcu_imagenet_base_l2.sh │ │ ├── mcu_imagenet_base_l4.sh │ │ ├── mcu_imagenet_filt_l2.sh │ │ └── mcu_imagenet_filt_l4.sh │ ├── res18 │ │ ├── res18_cifar100_base.sh │ │ ├── res18_cifar100_r2.sh │ │ ├── res18_cifar100_r4.sh │ │ ├── res18_cifar100_r7.sh │ │ ├── res18_cifar10_base.sh │ │ ├── res18_cifar10_r2.sh │ │ ├── res18_cifar10_r4.sh │ │ ├── res18_cifar10_r7.sh │ │ ├── res18_imagenet_base_l2.sh │ │ ├── res18_imagenet_base_l4.sh │ │ ├── res18_imagenet_filt_l2.sh │ │ └── res18_imagenet_filt_l4.sh │ └── res34 │ │ ├── res34_cifar100_base.sh │ │ ├── res34_cifar100_r2.sh │ │ ├── res34_cifar100_r4.sh │ │ ├── res34_cifar100_r7.sh │ │ ├── res34_cifar10_base.sh │ │ ├── res34_cifar10_r2.sh │ │ ├── res34_cifar10_r4.sh │ │ ├── res34_cifar10_r7.sh │ │ ├── res34_imagenet_base_l2.sh │ │ ├── res34_imagenet_base_l4.sh │ │ ├── res34_imagenet_filt_l2.sh │ │ └── res34_imagenet_filt_l4.sh ├── trainer_cls.py └── util.py ├── latency ├── .gitignore ├── CMakeLists.txt ├── cudnn_prof.cpp ├── dnnl_prof.cpp ├── run_test.sh └── test_cfg.txt └── segmentation ├── .gitignore ├── configs ├── deeplabv3 │ ├── calib_deeplabv3_r18-d8_512x512_1k_voc12aug.py │ ├── enc_10L_deeplabv3_r18-d8_512x512_20k_voc12aug.py │ ├── enc_5L_deeplabv3_r18-d8_512x512_20k_voc12aug.py │ ├── filt_10L_deeplabv3_r18-d8_512x512_20k_voc12aug.py │ ├── filt_5L_deeplabv3_r18-d8_512x512_20k_voc12aug.py │ └── full_deeplabv3_r18-d8_512x512_20k_voc12aug.py ├── deeplabv3mv2 │ ├── calib_deeplabv3_mv2_512x1024_5k_cityscapes.py │ ├── calib_deeplabv3_mv2_512x512_5k_voc12aug.py │ ├── enc_10L_deeplabv3_mv2_512x1024_20k_cityscapes.py │ ├── enc_10L_deeplabv3_mv2_512x512_20k_voc12aug.py │ ├── enc_5L_deeplabv3_mv2_512x1024_20k_cityscapes.py │ ├── enc_5L_deeplabv3_mv2_512x512_20k_voc12aug.py │ ├── filt_10L_deeplabv3_mv2_512x1024_20k_cityscapes.py │ ├── filt_10L_deeplabv3_mv2_512x512_20k_voc12aug.py │ ├── filt_5L_deeplabv3_mv2_512x1024_20k_cityscapes.py │ ├── filt_5L_deeplabv3_mv2_512x512_20k_voc12aug.py │ ├── full_deeplabv3_mv2_512x1024_20k_cityscapes.py │ └── full_deeplabv3_mv2_512x512_20k_voc12aug.py ├── fcn │ ├── calib_fcn_r18-d8_512x512_1k_voc12aug.py │ ├── enc_10L_fcn_r18-d8_512x512_voc12aug.py │ ├── enc_5L_fcn_r18-d8_512x512_voc12aug.py │ ├── filt_10L_fcn_r18-d8_512x512_voc12aug.py │ ├── filt_5L_fcn_r18-d8_512x512_voc12aug.py │ └── full_fcn_r18-d8_512x512_20k_voc12aug.py ├── pspnet │ ├── calib_pspnet_r18-d8_512x512_1k_voc12aug.py │ ├── enc_10L_pspnet_r18-d8_512x512_20k_voc12aug.py │ ├── enc_5L_pspnet_r18-d8_512x512_20k_voc12aug.py │ ├── filt_10L_pspnet_r18-d8_512x512_20k_voc12aug.py │ ├── filt_5L_pspnet_r18-d8_512x512_20k_voc12aug.py │ └── full_pspnet_r18-d8_512x512_20k_voc12aug.py ├── pspnetmv2 │ ├── calib_pspnet_mv2-d8_512x1024_5k_cityscapes.py │ ├── calib_pspnet_mv2-d8_512x512_5k_voc12aug.py │ ├── enc_10L_pspnet_mv2-d8_512x1024_20k_cityscapes.py │ ├── enc_10L_pspnet_mv2-d8_512x512_20k_voc12aug.py │ ├── enc_5L_pspnet_mv2-d8_512x1024_20k_cityscapes.py │ ├── enc_5L_pspnet_mv2-d8_512x512_20k_voc12aug.py │ ├── filt_10L_pspnet_mv2-d8_512x1024_20k_cityscapes.py │ ├── filt_10L_pspnet_mv2-d8_512x512_20k_voc12aug.py │ ├── filt_5L_pspnet_mv2-d8_512x1024_20k_cityscapes.py │ ├── filt_5L_pspnet_mv2-d8_512x512_20k_voc12aug.py │ ├── full_pspnet_mv2-d8_512x1024_20k_cityscapes.py │ └── full_pspnet_mv2-d8_512x512_20k_voc12aug.py └── upernet │ ├── calib_upernet_r18_512x1024_1k_cityscapes.py │ ├── calib_upernet_r18_512x512_1k_voc12aug.py │ ├── enc_10L_upernet_r18_512x512_20k_cityscapes.py │ ├── enc_10L_upernet_r18_512x512_20k_voc12aug.py │ ├── enc_5L_upernet_r18_512x512_20k_cityscapes.py │ ├── enc_5L_upernet_r18_512x512_20k_voc12aug.py │ ├── filt_10L_upernet_r18_512x512_20k_cityscapes.py │ ├── filt_10L_upernet_r18_512x512_20k_voc12aug.py │ ├── filt_5L_upernet_r18_512x512_20k_cityscapes.py │ ├── filt_5L_upernet_r18_512x512_20k_voc12aug.py │ ├── full_upernet_r18_512x1024_20k_cityscapes.py │ └── full_upernet_r18_512x512_20k_voc12aug.py ├── custom_op ├── __init__.py ├── conv_avg.py ├── latency │ ├── .gitignore │ ├── CMakeLists.txt │ ├── cudnn_conv.cpp │ ├── cudnn_prof.cpp │ ├── dnnl_conv.cpp │ ├── dnnl_prof.cpp │ ├── jetson_pwr_logger.cpp │ ├── log_analyse.ipynb │ ├── speedup_mem.pdf │ ├── test.sh │ └── test_cfg.txt └── register.py ├── data └── .gitkeep ├── dist_train.sh ├── mmsegmentation ├── .circleci │ └── config.yml ├── .dev │ ├── batch_test_list.py │ ├── batch_train_list.txt │ ├── benchmark_evaluation.sh │ ├── benchmark_inference.py │ ├── benchmark_train.sh │ ├── check_urls.py │ ├── gather_benchmark_evaluation_results.py │ ├── gather_benchmark_train_results.py │ ├── gather_models.py │ ├── generate_benchmark_evaluation_script.py │ ├── generate_benchmark_train_script.py │ ├── log_collector │ │ ├── example_config.py │ │ ├── log_collector.py │ │ ├── readme.md │ │ └── utils.py │ ├── md2yml.py │ └── upload_modelzoo.py ├── .gitignore ├── .owners.yml ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── CITATION.cff ├── LICENSE ├── LICENSES.md ├── MANIFEST.in ├── README.md ├── README_zh-CN.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── ade20k_640x640.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_1024x1024.py │ │ │ ├── cityscapes_768x768.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── cityscapes_832x832.py │ │ │ ├── coco-stuff10k.py │ │ │ ├── coco-stuff164k.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── isaid.py │ │ │ ├── loveda.py │ │ │ ├── pascal_context.py │ │ │ ├── pascal_context_59.py │ │ │ ├── pascal_voc12.py │ │ │ ├── pascal_voc12_aug.py │ │ │ ├── potsdam.py │ │ │ ├── stare.py │ │ │ └── vaihingen.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── ann_r50-d8.py │ │ │ ├── apcnet_r50-d8.py │ │ │ ├── bisenetv1_r18-d32.py │ │ │ ├── bisenetv2.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 │ │ │ ├── dpt_vit-b16.py │ │ │ ├── emanet_r50-d8.py │ │ │ ├── encnet_r50-d8.py │ │ │ ├── erfnet_fcn.py │ │ │ ├── fast_scnn.py │ │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ │ ├── fcn_hr18.py │ │ │ ├── fcn_r50-d8.py │ │ │ ├── fcn_unet_s5-d16.py │ │ │ ├── fpn_r50.py │ │ │ ├── gcnet_r50-d8.py │ │ │ ├── icnet_r50-d8.py │ │ │ ├── isanet_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 │ │ │ ├── segformer_mit-b0.py │ │ │ ├── segmenter_vit-b16_mask.py │ │ │ ├── setr_mla.py │ │ │ ├── setr_naive.py │ │ │ ├── setr_pup.py │ │ │ ├── stdc.py │ │ │ ├── twins_pcpvt-s_fpn.py │ │ │ ├── twins_pcpvt-s_upernet.py │ │ │ ├── upernet_beit.py │ │ │ ├── upernet_convnext.py │ │ │ ├── upernet_mae.py │ │ │ ├── upernet_r50.py │ │ │ ├── upernet_swin.py │ │ │ └── upernet_vit-b16_ln_mln.py │ │ └── schedules │ │ │ ├── schedule_10k.py │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.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 │ ├── beit │ │ ├── README.md │ │ ├── beit.yml │ │ ├── upernet_beit-base_640x640_160k_ade20k_ms.py │ │ ├── upernet_beit-base_8x2_640x640_160k_ade20k.py │ │ ├── upernet_beit-large_fp16_640x640_160k_ade20k_ms.py │ │ └── upernet_beit-large_fp16_8x1_640x640_160k_ade20k.py │ ├── bisenetv1 │ │ ├── README.md │ │ ├── bisenetv1.yml │ │ ├── bisenetv1_r101-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv1_r101-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv1_r18-d32_4x4_1024x1024_160k_cityscapes.py │ │ ├── bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ ├── bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py │ │ ├── bisenetv1_r18-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv1_r18-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv1_r50-d32_4x4_1024x1024_160k_cityscapes.py │ │ ├── bisenetv1_r50-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ ├── bisenetv1_r50-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ └── bisenetv1_r50-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ ├── bisenetv2 │ │ ├── README.md │ │ ├── bisenetv2.yml │ │ ├── bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py │ │ ├── bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py │ │ ├── bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py │ │ └── bisenetv2_fcn_ohem_4x4_1024x1024_160k_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 │ ├── convnext │ │ ├── README.md │ │ ├── convnext.yml │ │ ├── upernet_convnext_base_fp16_512x512_160k_ade20k.py │ │ ├── upernet_convnext_base_fp16_640x640_160k_ade20k.py │ │ ├── upernet_convnext_large_fp16_640x640_160k_ade20k.py │ │ ├── upernet_convnext_small_fp16_512x512_160k_ade20k.py │ │ ├── upernet_convnext_tiny_fp16_512x512_160k_ade20k.py │ │ └── upernet_convnext_xlarge_fp16_640x640_160k_ade20k.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_4x4_160k_coco-stuff164k.py │ │ ├── deeplabv3_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ ├── deeplabv3_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ ├── deeplabv3_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ ├── deeplabv3_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ ├── deeplabv3_r101-d8_512x512_80k_ade20k.py │ │ ├── deeplabv3_r101-d8_769x769_40k_cityscapes.py │ │ ├── deeplabv3_r101-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3_r101-d8_fp16_512x1024_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_4x4_160k_coco-stuff164k.py │ │ ├── deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ ├── deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ ├── deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ ├── deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.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_4x4_512x512_80k_vaihingen.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_512x512_80k_loveda.py │ │ ├── deeplabv3plus_r101-d8_512x512_80k_potsdam.py │ │ ├── deeplabv3plus_r101-d8_769x769_40k_cityscapes.py │ │ ├── deeplabv3plus_r101-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus_r18-d8_4x4_512x512_80k_vaihingen.py │ │ ├── deeplabv3plus_r18-d8_4x4_896x896_80k_isaid.py │ │ ├── deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r18-d8_512x512_80k_loveda.py │ │ ├── deeplabv3plus_r18-d8_512x512_80k_potsdam.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_4x4_512x512_80k_vaihingen.py │ │ ├── deeplabv3plus_r50-d8_4x4_896x896_80k_isaid.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_512x512_80k_loveda.py │ │ ├── deeplabv3plus_r50-d8_512x512_80k_potsdam.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 │ ├── dpt │ │ ├── README.md │ │ ├── dpt.yml │ │ └── dpt_vit-b16_512x512_160k_ade20k.py │ ├── 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 │ ├── erfnet │ │ ├── README.md │ │ ├── erfnet.yml │ │ └── erfnet_fcn_4x4_512x1024_160k_cityscapes.py │ ├── fastfcn │ │ ├── README.md │ │ ├── fastfcn.yml │ │ ├── fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_160k_ade20k.py │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_80k_ade20k.py │ │ ├── fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_enc_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_160k_ade20k.py │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_80k_ade20k.py │ │ ├── fastfcn_r50-d32_jpu_psp_4x4_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py │ │ ├── fastfcn_r50-d32_jpu_psp_512x512_160k_ade20k.py │ │ └── fastfcn_r50-d32_jpu_psp_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_r101-d8_fp16_512x1024_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 │ ├── 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_4x4_512x512_80k_vaihingen.py │ │ ├── fcn_hr18_4x4_896x896_80k_isaid.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_hr18_512x512_80k_loveda.py │ │ ├── fcn_hr18_512x512_80k_potsdam.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_4x4_512x512_80k_vaihingen.py │ │ ├── fcn_hr18s_4x4_896x896_80k_isaid.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_hr18s_512x512_80k_loveda.py │ │ ├── fcn_hr18s_512x512_80k_potsdam.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_4x4_512x512_80k_vaihingen.py │ │ ├── fcn_hr48_4x4_896x896_80k_isaid.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 │ │ ├── fcn_hr48_512x512_80k_loveda.py │ │ ├── fcn_hr48_512x512_80k_potsdam.py │ │ └── hrnet.yml │ ├── icnet │ │ ├── README.md │ │ ├── icnet.yml │ │ ├── icnet_r101-d8_832x832_160k_cityscapes.py │ │ ├── icnet_r101-d8_832x832_80k_cityscapes.py │ │ ├── icnet_r101-d8_in1k-pre_832x832_160k_cityscapes.py │ │ ├── icnet_r101-d8_in1k-pre_832x832_80k_cityscapes.py │ │ ├── icnet_r18-d8_832x832_160k_cityscapes.py │ │ ├── icnet_r18-d8_832x832_80k_cityscapes.py │ │ ├── icnet_r18-d8_in1k-pre_832x832_160k_cityscapes.py │ │ ├── icnet_r18-d8_in1k-pre_832x832_80k_cityscapes.py │ │ ├── icnet_r50-d8_832x832_160k_cityscapes.py │ │ ├── icnet_r50-d8_832x832_80k_cityscapes.py │ │ ├── icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py │ │ └── icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py │ ├── isanet │ │ ├── README.md │ │ ├── isanet.yml │ │ ├── isanet_r101-d8_512x1024_40k_cityscapes.py │ │ ├── isanet_r101-d8_512x1024_80k_cityscapes.py │ │ ├── isanet_r101-d8_512x512_160k_ade20k.py │ │ ├── isanet_r101-d8_512x512_20k_voc12aug.py │ │ ├── isanet_r101-d8_512x512_40k_voc12aug.py │ │ ├── isanet_r101-d8_512x512_80k_ade20k.py │ │ ├── isanet_r101-d8_769x769_40k_cityscapes.py │ │ ├── isanet_r101-d8_769x769_80k_cityscapes.py │ │ ├── isanet_r50-d8_512x1024_40k_cityscapes.py │ │ ├── isanet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── isanet_r50-d8_512x512_160k_ade20k.py │ │ ├── isanet_r50-d8_512x512_20k_voc12aug.py │ │ ├── isanet_r50-d8_512x512_40k_voc12aug.py │ │ ├── isanet_r50-d8_512x512_80k_ade20k.py │ │ ├── isanet_r50-d8_769x769_40k_cityscapes.py │ │ └── isanet_r50-d8_769x769_80k_cityscapes.py │ ├── knet │ │ ├── README.md │ │ ├── knet.yml │ │ ├── knet_s3_deeplabv3_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ ├── knet_s3_fcn_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ ├── knet_s3_pspnet_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ ├── knet_s3_upernet_r50-d8_8x2_512x512_adamw_80k_ade20k.py │ │ ├── knet_s3_upernet_swin-l_8x2_512x512_adamw_80k_ade20k.py │ │ ├── knet_s3_upernet_swin-l_8x2_640x640_adamw_80k_ade20k.py │ │ └── knet_s3_upernet_swin-t_8x2_512x512_adamw_80k_ade20k.py │ ├── mae │ │ ├── README.md │ │ ├── mae.yml │ │ ├── upernet_mae-base_fp16_512x512_160k_ade20k_ms.py │ │ └── upernet_mae-base_fp16_8x2_512x512_160k_ade20k.py │ ├── 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_4x4_512x512_80k_potsdam.py │ │ ├── pspnet_r101-d8_4x4_512x512_80k_vaihingen.py │ │ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py │ │ ├── pspnet_r101-d8_512x1024_40k_dark.py │ │ ├── pspnet_r101-d8_512x1024_40k_night_driving.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_4x4_160k_coco-stuff164k.py │ │ ├── pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ ├── pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ ├── pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ ├── pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ ├── pspnet_r101-d8_512x512_80k_ade20k.py │ │ ├── pspnet_r101-d8_512x512_80k_loveda.py │ │ ├── pspnet_r101-d8_769x769_40k_cityscapes.py │ │ ├── pspnet_r101-d8_769x769_80k_cityscapes.py │ │ ├── pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ ├── pspnet_r101b-d8_512x1024_80k_cityscapes.py │ │ ├── pspnet_r101b-d8_512x1024_80k_dark.py │ │ ├── pspnet_r101b-d8_512x1024_80k_night_driving.py │ │ ├── pspnet_r101b-d8_769x769_80k_cityscapes.py │ │ ├── pspnet_r18-d8_4x4_512x512_80k_potsdam.py │ │ ├── pspnet_r18-d8_4x4_512x512_80k_vaihingen.py │ │ ├── pspnet_r18-d8_4x4_896x896_80k_isaid.py │ │ ├── pspnet_r18-d8_512x1024_80k_cityscapes.py │ │ ├── pspnet_r18-d8_512x512_80k_loveda.py │ │ ├── pspnet_r18-d8_769x769_80k_cityscapes.py │ │ ├── pspnet_r18b-d8_512x1024_80k_cityscapes.py │ │ ├── pspnet_r18b-d8_769x769_80k_cityscapes.py │ │ ├── pspnet_r50-d32_512x1024_80k_cityscapes.py │ │ ├── pspnet_r50-d32_rsb-pretrain_512x1024_adamw_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_4x4_512x512_80k_potsdam.py │ │ ├── pspnet_r50-d8_4x4_512x512_80k_vaihingen.py │ │ ├── pspnet_r50-d8_4x4_896x896_80k_isaid.py │ │ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py │ │ ├── pspnet_r50-d8_512x1024_40k_dark.py │ │ ├── pspnet_r50-d8_512x1024_40k_night_driving.py │ │ ├── pspnet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── pspnet_r50-d8_512x1024_80k_dark.py │ │ ├── pspnet_r50-d8_512x1024_80k_night_driving.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_4x4_160k_coco-stuff164k.py │ │ ├── pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ ├── pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ ├── pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ ├── pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py │ │ ├── pspnet_r50-d8_512x512_80k_ade20k.py │ │ ├── pspnet_r50-d8_512x512_80k_loveda.py │ │ ├── pspnet_r50-d8_769x769_40k_cityscapes.py │ │ ├── pspnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── pspnet_r50-d8_rsb-pretrain_512x1024_adamw_80k_cityscapes.py │ │ ├── pspnet_r50b-d32_512x1024_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 │ ├── segformer │ │ ├── README.md │ │ ├── segformer.yml │ │ ├── segformer_mit-b0_512x512_160k_ade20k.py │ │ ├── segformer_mit-b0_8x1_1024x1024_160k_cityscapes.py │ │ ├── segformer_mit-b1_512x512_160k_ade20k.py │ │ ├── segformer_mit-b1_8x1_1024x1024_160k_cityscapes.py │ │ ├── segformer_mit-b2_512x512_160k_ade20k.py │ │ ├── segformer_mit-b2_8x1_1024x1024_160k_cityscapes.py │ │ ├── segformer_mit-b3_512x512_160k_ade20k.py │ │ ├── segformer_mit-b3_8x1_1024x1024_160k_cityscapes.py │ │ ├── segformer_mit-b4_512x512_160k_ade20k.py │ │ ├── segformer_mit-b4_8x1_1024x1024_160k_cityscapes.py │ │ ├── segformer_mit-b5_512x512_160k_ade20k.py │ │ ├── segformer_mit-b5_640x640_160k_ade20k.py │ │ └── segformer_mit-b5_8x1_1024x1024_160k_cityscapes.py │ ├── segmenter │ │ ├── README.md │ │ ├── segmenter.yml │ │ ├── segmenter_vit-b_mask_8x1_512x512_160k_ade20k.py │ │ ├── segmenter_vit-l_mask_8x1_640x640_160k_ade20k.py │ │ ├── segmenter_vit-s_linear_8x1_512x512_160k_ade20k.py │ │ ├── segmenter_vit-s_mask_8x1_512x512_160k_ade20k.py │ │ └── segmenter_vit-t_mask_8x1_512x512_160k_ade20k.py │ ├── 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 │ │ ├── setr_vit-large_mla_8x1_768x768_80k_cityscapes.py │ │ ├── setr_vit-large_naive_8x1_768x768_80k_cityscapes.py │ │ └── setr_vit-large_pup_8x1_768x768_80k_cityscapes.py │ ├── stdc │ │ ├── README.md │ │ ├── stdc.yml │ │ ├── stdc1_512x1024_80k_cityscapes.py │ │ ├── stdc1_in1k-pre_512x1024_80k_cityscapes.py │ │ ├── stdc2_512x1024_80k_cityscapes.py │ │ └── stdc2_in1k-pre_512x1024_80k_cityscapes.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_large_patch4_window12_512x512_pretrain_384x384_22K_160k_ade20k.py │ │ ├── upernet_swin_large_patch4_window7_512x512_pretrain_224x224_22K_160k_ade20k.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 │ ├── twins │ │ ├── README.md │ │ ├── twins.yml │ │ ├── twins_pcpvt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ ├── twins_pcpvt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── twins_pcpvt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ ├── twins_pcpvt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ ├── twins_pcpvt-s_uperhead_8x4_512x512_160k_ade20k.py │ │ ├── twins_svt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ ├── twins_svt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── twins_svt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ ├── twins_svt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── twins_svt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ └── twins_svt-s_uperhead_8x2_512x512_160k_ade20k.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 │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_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_4x4_512x1024_160k_cityscapes.py │ │ ├── fcn_unet_s5-d16_64x64_40k_drive.py │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_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 │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_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_r18_512x1024_40k_cityscapes.py │ │ ├── upernet_r18_512x1024_80k_cityscapes.py │ │ ├── upernet_r18_512x512_160k_ade20k.py │ │ ├── upernet_r18_512x512_20k_voc12aug.py │ │ ├── upernet_r18_512x512_40k_voc12aug.py │ │ ├── upernet_r18_512x512_80k_ade20k.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 │ └── video_demo.py ├── docker │ ├── Dockerfile │ └── serve │ │ ├── Dockerfile │ │ ├── config.properties │ │ └── entrypoint.sh ├── docs │ ├── en │ │ ├── Makefile │ │ ├── _static │ │ │ ├── css │ │ │ │ └── readthedocs.css │ │ │ └── images │ │ │ │ └── mmsegmentation.png │ │ ├── api.rst │ │ ├── changelog.md │ │ ├── conf.py │ │ ├── dataset_prepare.md │ │ ├── faq.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 │ └── zh_cn │ │ ├── Makefile │ │ ├── _static │ │ ├── css │ │ │ └── readthedocs.css │ │ └── images │ │ │ └── mmsegmentation.png │ │ ├── api.rst │ │ ├── conf.py │ │ ├── dataset_prepare.md │ │ ├── faq.md │ │ ├── get_started.md │ │ ├── imgs │ │ ├── qq_group_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 │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── test.py │ │ └── train.py │ ├── core │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── class_names.py │ │ │ ├── eval_hooks.py │ │ │ └── metrics.py │ │ ├── hook │ │ │ ├── __init__.py │ │ │ └── wandblogger_hook.py │ │ ├── optimizers │ │ │ ├── __init__.py │ │ │ └── layer_decay_optimizer_constructor.py │ │ ├── seg │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ └── ohem_pixel_sampler.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── dist_util.py │ │ │ └── misc.py │ ├── datasets │ │ ├── __init__.py │ │ ├── ade.py │ │ ├── builder.py │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── coco_stuff.py │ │ ├── custom.py │ │ ├── dark_zurich.py │ │ ├── dataset_wrappers.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── isaid.py │ │ ├── isprs.py │ │ ├── loveda.py │ │ ├── night_driving.py │ │ ├── pascal_context.py │ │ ├── pipelines │ │ │ ├── __init__.py │ │ │ ├── compose.py │ │ │ ├── formating.py │ │ │ ├── formatting.py │ │ │ ├── loading.py │ │ │ ├── test_time_aug.py │ │ │ └── transforms.py │ │ ├── potsdam.py │ │ ├── samplers │ │ │ ├── __init__.py │ │ │ └── distributed_sampler.py │ │ ├── stare.py │ │ └── voc.py │ ├── models │ │ ├── __init__.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── beit.py │ │ │ ├── bisenetv1.py │ │ │ ├── bisenetv2.py │ │ │ ├── cgnet.py │ │ │ ├── erfnet.py │ │ │ ├── fast_scnn.py │ │ │ ├── hrnet.py │ │ │ ├── icnet.py │ │ │ ├── mae.py │ │ │ ├── mit.py │ │ │ ├── mobilenet_v2.py │ │ │ ├── mobilenet_v3.py │ │ │ ├── resnest.py │ │ │ ├── resnet.py │ │ │ ├── resnext.py │ │ │ ├── stdc.py │ │ │ ├── swin.py │ │ │ ├── timm_backbone.py │ │ │ ├── twins.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 │ │ │ ├── dpt_head.py │ │ │ ├── ema_head.py │ │ │ ├── enc_head.py │ │ │ ├── fcn_head.py │ │ │ ├── fpn_head.py │ │ │ ├── gc_head.py │ │ │ ├── isa_head.py │ │ │ ├── knet_head.py │ │ │ ├── lraspp_head.py │ │ │ ├── nl_head.py │ │ │ ├── ocr_head.py │ │ │ ├── point_head.py │ │ │ ├── psa_head.py │ │ │ ├── psp_head.py │ │ │ ├── segformer_head.py │ │ │ ├── segmenter_mask_head.py │ │ │ ├── sep_aspp_head.py │ │ │ ├── sep_fcn_head.py │ │ │ ├── setr_mla_head.py │ │ │ ├── setr_up_head.py │ │ │ ├── stdc_head.py │ │ │ └── uper_head.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── accuracy.py │ │ │ ├── cross_entropy_loss.py │ │ │ ├── dice_loss.py │ │ │ ├── focal_loss.py │ │ │ ├── lovasz_loss.py │ │ │ ├── tversky_loss.py │ │ │ └── utils.py │ │ ├── necks │ │ │ ├── __init__.py │ │ │ ├── featurepyramid.py │ │ │ ├── fpn.py │ │ │ ├── ic_neck.py │ │ │ ├── jpu.py │ │ │ ├── mla_neck.py │ │ │ └── multilevel_neck.py │ │ ├── segmentors │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── cascade_encoder_decoder.py │ │ │ └── encoder_decoder.py │ │ └── utils │ │ │ ├── __init__.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 │ │ ├── misc.py │ │ ├── set_env.py │ │ └── util_distribution.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 ├── setup.cfg ├── setup.py ├── tests │ ├── __init__.py │ ├── test_apis │ │ └── test_single_gpu.py │ ├── test_config.py │ ├── test_core │ │ ├── test_layer_decay_optimizer_constructor.py │ │ └── test_optimizer.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_beit.py │ │ │ ├── test_bisenetv1.py │ │ │ ├── test_bisenetv2.py │ │ │ ├── test_blocks.py │ │ │ ├── test_cgnet.py │ │ │ ├── test_erfnet.py │ │ │ ├── test_fast_scnn.py │ │ │ ├── test_hrnet.py │ │ │ ├── test_icnet.py │ │ │ ├── test_mae.py │ │ │ ├── test_mit.py │ │ │ ├── test_mobilenet_v3.py │ │ │ ├── test_resnest.py │ │ │ ├── test_resnet.py │ │ │ ├── test_resnext.py │ │ │ ├── test_stdc.py │ │ │ ├── test_swin.py │ │ │ ├── test_timm_backbone.py │ │ │ ├── test_twins.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_dpt_head.py │ │ │ ├── test_ema_head.py │ │ │ ├── test_enc_head.py │ │ │ ├── test_fcn_head.py │ │ │ ├── test_gc_head.py │ │ │ ├── test_isa_head.py │ │ │ ├── test_knet_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_segformer_head.py │ │ │ ├── test_segmenter_mask_head.py │ │ │ ├── test_setr_mla_head.py │ │ │ ├── test_setr_up_head.py │ │ │ ├── test_stdc_head.py │ │ │ ├── test_uper_head.py │ │ │ └── utils.py │ │ ├── test_losses │ │ │ ├── __init__.py │ │ │ ├── test_ce_loss.py │ │ │ ├── test_dice_loss.py │ │ │ ├── test_focal_loss.py │ │ │ ├── test_lovasz_loss.py │ │ │ ├── test_tversky_loss.py │ │ │ └── test_utils.py │ │ ├── test_necks │ │ │ ├── __init__.py │ │ │ ├── test_feature2pyramid.py │ │ │ ├── test_fpn.py │ │ │ ├── test_ic_neck.py │ │ │ ├── test_jpu.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_utils │ │ │ ├── __init__.py │ │ │ ├── test_embed.py │ │ │ └── test_shape_convert.py │ ├── test_sampler.py │ └── test_utils │ │ ├── test_misc.py │ │ ├── test_set_env.py │ │ └── test_util_distribution.py └── tools │ ├── analyze_logs.py │ ├── benchmark.py │ ├── browse_dataset.py │ ├── confusion_matrix.py │ ├── convert_datasets │ ├── chase_db1.py │ ├── cityscapes.py │ ├── coco_stuff10k.py │ ├── coco_stuff164k.py │ ├── drive.py │ ├── hrf.py │ ├── isaid.py │ ├── loveda.py │ ├── pascal_context.py │ ├── potsdam.py │ ├── stare.py │ ├── vaihingen.py │ └── voc_aug.py │ ├── deploy_test.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── get_flops.py │ ├── model_converters │ ├── beit2mmseg.py │ ├── mit2mmseg.py │ ├── stdc2mmseg.py │ ├── swin2mmseg.py │ ├── twins2mmseg.py │ ├── vit2mmseg.py │ └── vitjax2mmseg.py │ ├── onnx2tensorrt.py │ ├── print_config.py │ ├── publish_model.py │ ├── pytorch2onnx.py │ ├── pytorch2torchscript.py │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ ├── torchserve │ ├── mmseg2torchserve.py │ ├── mmseg_handler.py │ └── test_torchserve.py │ └── train.py ├── results ├── dlv3.csv ├── dlv3m.csv ├── fcn.csv ├── psp.csv ├── pspm.csv └── uper.csv ├── scripts ├── calib.sh └── example_dlv3.sh ├── slurm_train.sh ├── tools ├── compare_ckpt.py └── get_miou.py └── train.py /assets/gf_method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/assets/gf_method.png -------------------------------------------------------------------------------- /assets/latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/assets/latency.png -------------------------------------------------------------------------------- /classification/custom_op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/classification/custom_op/__init__.py -------------------------------------------------------------------------------- /classification/data/cifar10/usr_group_15.29.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/classification/data/cifar10/usr_group_15.29.npy -------------------------------------------------------------------------------- /classification/data/cifar100/usr_group_15.82.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/classification/data/cifar100/usr_group_15.82.npy -------------------------------------------------------------------------------- /classification/data/imagenet/usr_group_13.10.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/classification/data/imagenet/usr_group_13.10.npy -------------------------------------------------------------------------------- /classification/dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/classification/dataloader/__init__.py -------------------------------------------------------------------------------- /latency/.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | .vscode/ 3 | results*/ -------------------------------------------------------------------------------- /latency/run_test.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | 3 | echo CPU Test 4 | build/dnnl_prof test_cfg.txt cpu.csv 5 | sleep 0.5 6 | echo Done 7 | 8 | echo GPU Test 9 | build/cudnn_prof test_cfg.txt gpu.csv 10 | sleep 0.5 11 | echo Done 12 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3/enc_10L_deeplabv3_r18-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_deeplabv3_r18-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", "decode_head", "~decode_head.conv_seg", 5 | "~decode_head.bottleneck", "~decode_head.aspp_modules", 6 | "~backbone.layer4.1.conv2", 7 | "~backbone.layer4.1.conv1", 8 | "~backbone.layer4.0.conv2", 9 | "~backbone.layer4.0.conv1", 10 | ] 11 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3/enc_5L_deeplabv3_r18-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_deeplabv3_r18-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", "decode_head", "~decode_head.conv_seg", 5 | "~decode_head.bottleneck", "~decode_head.aspp_modules" 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3mv2/enc_10L_deeplabv3_mv2_512x1024_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | 'full_deeplabv3_mv2_512x1024_20k_cityscapes.py', 3 | ] 4 | 5 | freeze_layers = [ 6 | "backbone", "~backbone.layer7", "~backbone.layer6.2.conv.2" 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3mv2/enc_10L_deeplabv3_mv2_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | 'full_deeplabv3_mv2_512x512_20k_voc12aug.py', 3 | ] 4 | 5 | freeze_layers = [ 6 | "backbone", "~backbone.layer7", "~backbone.layer6.2.conv.2" 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3mv2/enc_5L_deeplabv3_mv2_512x1024_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | 'full_deeplabv3_mv2_512x1024_20k_cityscapes.py', 3 | ] 4 | 5 | freeze_layers = [ 6 | "backbone", "decode_head", "~decode_head.conv_seg", 7 | "~decode_head.bottleneck", "~decode_head.aspp_modules" 8 | ] 9 | -------------------------------------------------------------------------------- /segmentation/configs/deeplabv3mv2/enc_5L_deeplabv3_mv2_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | 'full_deeplabv3_mv2_512x512_20k_voc12aug.py', 3 | ] 4 | 5 | freeze_layers = [ 6 | "backbone", "decode_head", "~decode_head.conv_seg", 7 | "~decode_head.bottleneck", "~decode_head.aspp_modules" 8 | ] 9 | -------------------------------------------------------------------------------- /segmentation/configs/fcn/enc_10L_fcn_r18-d8_512x512_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_fcn_r18-d8_512x512_20k_voc12aug.py' 2 | 3 | freeze_layers = [ 4 | "backbone", 5 | "~backbone.layer4", 6 | "~backbone.layer3.1", 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/configs/fcn/enc_5L_fcn_r18-d8_512x512_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_fcn_r18-d8_512x512_20k_voc12aug.py' 2 | 3 | freeze_layers = [ 4 | "backbone", 5 | "~backbone.layer4.1", 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/configs/pspnet/enc_10L_pspnet_r18-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_r18-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", "~backbone.layer4", 5 | ] 6 | -------------------------------------------------------------------------------- /segmentation/configs/pspnet/enc_5L_pspnet_r18-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_r18-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", 5 | ] 6 | -------------------------------------------------------------------------------- /segmentation/configs/pspnet/filt_5L_pspnet_r18-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_r18-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", 5 | ] 6 | 7 | 8 | gradient_filter = dict( 9 | enable=True, 10 | filter_install=[ 11 | dict(path="decode_head.bottleneck", type='cbr', radius=4), 12 | ] 13 | ) 14 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/enc_10L_pspnet_mv2-d8_512x1024_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x1024_20k_cityscapes.py" 2 | 3 | freeze_layers = [ 4 | "backbone", "~backbone.layer7", 5 | "~backbone.layer6.2.conv.2", "~backbone.layer6.2.conv.1" 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/enc_10L_pspnet_mv2-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = [ 4 | "backbone", "~backbone.layer7", 5 | "~backbone.layer6.2.conv.2", "~backbone.layer6.2.conv.1" 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/enc_5L_pspnet_mv2-d8_512x1024_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x1024_20k_cityscapes.py" 2 | 3 | freeze_layers = ["backbone"] 4 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/enc_5L_pspnet_mv2-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = ["backbone"] 4 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/filt_5L_pspnet_mv2-d8_512x1024_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x1024_20k_cityscapes.py" 2 | 3 | freeze_layers = ["backbone"] 4 | 5 | gradient_filter = dict( 6 | enable=True, 7 | filter_install=[ 8 | dict(path="decode_head.bottleneck", type='cbr', radius=4), 9 | ] 10 | ) 11 | -------------------------------------------------------------------------------- /segmentation/configs/pspnetmv2/filt_5L_pspnet_mv2-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = "./full_pspnet_mv2-d8_512x512_20k_voc12aug.py" 2 | 3 | freeze_layers = ["backbone"] 4 | 5 | gradient_filter = dict( 6 | enable=True, 7 | filter_install=[ 8 | dict(path="decode_head.bottleneck", type='cbr', radius=4), 9 | ] 10 | ) 11 | -------------------------------------------------------------------------------- /segmentation/configs/upernet/enc_10L_upernet_r18_512x512_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_upernet_r18_512x512_20k_cityscapes.py' 2 | 3 | freeze_layers = [ 4 | 'backbone', 'decode_head', '~decode_head.conv_seg', 5 | '~decode_head.fpn_bottleneck', '~decode_head.fpn_convs', '~decode_head.bottleneck', 6 | '~decode_head.lateral_convs', '~decode_head.psp_modules.0', '~decode_head.psp_modules.1', 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/configs/upernet/enc_10L_upernet_r18_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_upernet_r18_512x512_20k_voc12aug.py' 2 | 3 | freeze_layers = [ 4 | 'backbone', 'decode_head', '~decode_head.conv_seg', 5 | '~decode_head.fpn_bottleneck', '~decode_head.fpn_convs', '~decode_head.bottleneck', 6 | '~decode_head.lateral_convs', '~decode_head.psp_modules.0', '~decode_head.psp_modules.1', 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/configs/upernet/enc_5L_upernet_r18_512x512_20k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_upernet_r18_512x512_20k_cityscapes.py' 2 | 3 | freeze_layers = [ 4 | 'backbone', 'decode_head', '~decode_head.conv_seg', 5 | '~decode_head.fpn_bottleneck', '~decode_head.fpn_convs', '~decode_head.bottleneck' 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/configs/upernet/enc_5L_upernet_r18_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = 'full_upernet_r18_512x512_20k_voc12aug.py' 2 | 3 | freeze_layers = [ 4 | 'backbone', 'decode_head', '~decode_head.conv_seg', 5 | '~decode_head.fpn_bottleneck', '~decode_head.fpn_convs', '~decode_head.bottleneck' 6 | ] 7 | -------------------------------------------------------------------------------- /segmentation/custom_op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/custom_op/__init__.py -------------------------------------------------------------------------------- /segmentation/custom_op/latency/.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | .vscode/ 3 | results*/ 4 | cudnn_samples_v8/ -------------------------------------------------------------------------------- /segmentation/custom_op/latency/speedup_mem.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/custom_op/latency/speedup_mem.pdf -------------------------------------------------------------------------------- /segmentation/custom_op/latency/test.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | 3 | echo CPU Test 4 | build/dnnl_prof test_cfg.txt office_cpu.csv 5 | sleep 0.5 6 | echo Done 7 | 8 | echo GPU Test 9 | build/cudnn_prof test_cfg.txt office_gpu.csv 10 | sleep 0.5 11 | echo Done 12 | -------------------------------------------------------------------------------- /segmentation/data/.gitkeep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/.owners.yml: -------------------------------------------------------------------------------- 1 | assign: 2 | strategy: 3 | # random 4 | # round-robin 5 | daily-shift-based 6 | assignees: 7 | - MengzhangLI 8 | - xiexinch 9 | - MeowZheng 10 | - MengzhangLI 11 | - xiexinch 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/CITATION.cff: -------------------------------------------------------------------------------- 1 | cff-version: 1.2.0 2 | message: "If you use this software, please cite it as below." 3 | authors: 4 | - name: "MMSegmentation Contributors" 5 | title: "OpenMMLab Semantic Segmentation Toolbox and Benchmark" 6 | date-released: 2020-07-10 7 | url: "https://github.com/open-mmlab/mmsegmentation" 8 | license: Apache-2.0 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r101-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r101-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c')))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r18-d32_4x4_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/bisenetv1_r18-d32.py', 3 | '../_base_/datasets/cityscapes_1024x1024.py', 4 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 5 | ] 6 | lr_config = dict(warmup='linear', warmup_iters=1000) 7 | optimizer = dict(lr=0.025) 8 | data = dict( 9 | samples_per_gpu=4, 10 | workers_per_gpu=4, 11 | ) 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py' 2 | data = dict( 3 | samples_per_gpu=8, 4 | workers_per_gpu=4, 5 | ) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r18-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet18_v1c'))), ) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r50-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r50-d32_4x4_1024x1024_160k_cityscapes.py' 2 | model = dict( 3 | type='EncoderDecoder', 4 | backbone=dict( 5 | backbone_cfg=dict( 6 | init_cfg=dict( 7 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv1/bisenetv1_r50-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r50-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | backbone_cfg=dict( 6 | init_cfg=dict( 7 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv2/bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/bisenetv2.py', 3 | '../_base_/datasets/cityscapes_1024x1024.py', 4 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 5 | ] 6 | lr_config = dict(warmup='linear', warmup_iters=1000) 7 | optimizer = dict(lr=0.05) 8 | data = dict( 9 | samples_per_gpu=4, 10 | workers_per_gpu=4, 11 | ) 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv2/bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/bisenetv2.py', 3 | '../_base_/datasets/cityscapes_1024x1024.py', 4 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 5 | ] 6 | lr_config = dict(warmup='linear', warmup_iters=1000) 7 | optimizer = dict(lr=0.05) 8 | data = dict( 9 | samples_per_gpu=8, 10 | workers_per_gpu=4, 11 | ) 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/bisenetv2/bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_160k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r50-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_160k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/coco-stuff10k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_20k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_320k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/coco-stuff10k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3/deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r101-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_4x4_512x512_80k_vaihingen.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_loveda.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c'))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_potsdam.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r18-d8_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_4x4_896x896_80k_isaid.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r18-d8_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_potsdam.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r50-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/vaihingen.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=6), auxiliary_head=dict(num_classes=6)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r50-d8_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', '../_base_/datasets/isaid.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=16), auxiliary_head=dict(num_classes=16)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', '../_base_/datasets/loveda.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=7), auxiliary_head=dict(num_classes=7)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/potsdam.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=6), auxiliary_head=dict(num_classes=6)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/erfnet/erfnet_fcn_4x4_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/erfnet_fcn.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | data = dict( 6 | samples_per_gpu=4, 7 | workers_per_gpu=4, 8 | ) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | # model settings 2 | _base_ = './fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py' 3 | data = dict( 4 | samples_per_gpu=4, 5 | workers_per_gpu=4, 6 | ) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | # model settings 2 | _base_ = './fastfcn_r50-d32_jpu_enc_512x1024_80k_cityscapes.py' 3 | data = dict( 4 | samples_per_gpu=4, 5 | workers_per_gpu=4, 6 | ) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_psp_4x4_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fastfcn_r50-d32_jpu_psp.py', 3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | data = dict( 7 | samples_per_gpu=4, 8 | workers_per_gpu=4, 9 | ) 10 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fastfcn_r50-d32_jpu_psp.py', 3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_psp_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fastfcn_r50-d32_jpu_psp.py', 3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_160k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fastfcn/fastfcn_r50-d32_jpu_psp_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fastfcn_r50-d32_jpu_psp.py', 3 | '../_base_/datasets/ade20k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/fcn/fcn_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/hrnet/fcn_hr18_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/vaihingen.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=6)) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/hrnet/fcn_hr18_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/isaid.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=16)) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/hrnet/fcn_hr18_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/loveda.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=7)) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/hrnet/fcn_hr18_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/potsdam.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=6)) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r101-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(depth=101))) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r101-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(depth=101))) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r101-d8_in1k-pre_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | depth=101, 6 | init_cfg=dict( 7 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c')))) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r101-d8_in1k-pre_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | depth=101, 6 | init_cfg=dict( 7 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c')))) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r18-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict(layer_channels=(128, 512), backbone_cfg=dict(depth=18))) 4 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r18-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict(layer_channels=(128, 512), backbone_cfg=dict(depth=18))) 4 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r18-d8_in1k-pre_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | layer_channels=(128, 512), 5 | backbone_cfg=dict( 6 | depth=18, 7 | init_cfg=dict( 8 | type='Pretrained', checkpoint='open-mmlab://resnet18_v1c')))) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r18-d8_in1k-pre_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | layer_channels=(128, 512), 5 | backbone_cfg=dict( 6 | depth=18, 7 | init_cfg=dict( 8 | type='Pretrained', checkpoint='open-mmlab://resnet18_v1c')))) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r50-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/icnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_832x832.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_160k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r50-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/icnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_832x832.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/icnet/icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/isanet/isanet_r50-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_4x4_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_potsdam.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_vaihingen.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x1024_40k_dark.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_40k_dark.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x1024_40k_night_driving.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_40k_night_driving.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_160k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_80k_loveda.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c'))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101b-d8_512x1024_80k_dark.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_dark.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r101b-d8_512x1024_80k_night_driving.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_night_driving.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r18-d8_4x4_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_potsdam.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r18-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_vaihingen.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r18-d8_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_896x896_80k_isaid.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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r18-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_80k_loveda.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=18, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet18_v1c')), 7 | decode_head=dict( 8 | in_channels=512, 9 | channels=128, 10 | ), 11 | auxiliary_head=dict(in_channels=256, channels=64)) 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d32_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 | model = dict(backbone=dict(dilations=(1, 1, 2, 4), strides=(1, 2, 2, 2))) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_4x4_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/potsdam.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=6), auxiliary_head=dict(num_classes=6)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/vaihingen.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=6), auxiliary_head=dict(num_classes=6)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/isaid.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=16), auxiliary_head=dict(num_classes=16)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_160k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/coco-stuff10k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_20k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_320k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/coco-stuff10k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', 3 | '../_base_/datasets/coco-stuff164k.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(num_classes=171), auxiliary_head=dict(num_classes=171)) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/loveda.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=7), auxiliary_head=dict(num_classes=7)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/pspnet/pspnet_r50b-d32_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 | model = dict( 6 | pretrained='torchvision://resnet50', 7 | backbone=dict(type='ResNet', dilations=(1, 1, 2, 4), strides=(1, 2, 2, 2))) 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/stdc/stdc1_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/stdc.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | lr_config = dict(warmup='linear', warmup_iters=1000) 6 | data = dict( 7 | samples_per_gpu=12, 8 | workers_per_gpu=4, 9 | ) 10 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/stdc/stdc1_in1k-pre_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/stdc/stdc1_20220308-5368626c.pth' # noqa 2 | _base_ = './stdc1_512x1024_80k_cityscapes.py' 3 | model = dict( 4 | backbone=dict( 5 | backbone_cfg=dict( 6 | init_cfg=dict(type='Pretrained', checkpoint=checkpoint)))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/stdc/stdc2_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './stdc1_512x1024_80k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(stdc_type='STDCNet2'))) 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/stdc/stdc2_in1k-pre_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/stdc/stdc2_20220308-7dbd9127.pth' # noqa 2 | _base_ = './stdc2_512x1024_80k_cityscapes.py' 3 | model = dict( 4 | backbone=dict( 5 | backbone_cfg=dict( 6 | init_cfg=dict(type='Pretrained', checkpoint=checkpoint)))) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/twins/twins_pcpvt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py'] 2 | 3 | checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/twins/pcpvt_base_20220308-0621964c.pth' # noqa 4 | 5 | model = dict( 6 | backbone=dict( 7 | init_cfg=dict(type='Pretrained', checkpoint=checkpoint), 8 | depths=[3, 4, 18, 3]), ) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/twins/twins_pcpvt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py'] 2 | 3 | checkpoint = 'https://download.openmmlab.com/mmsegmentation/v0.5/pretrain/twins/pcpvt_large_20220308-37579dc6.pth' # noqa 4 | 5 | model = dict( 6 | backbone=dict( 7 | init_cfg=dict(type='Pretrained', checkpoint=checkpoint), 8 | depths=[3, 8, 27, 3])) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/twins/twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/twins_pcpvt-s_fpn.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | 6 | optimizer = dict(_delete_=True, type='AdamW', lr=0.0001, weight_decay=0.0001) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_unet_s5-d16_128x128_40k_stare.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_unet_s5-d16_256x256_40k_hrf.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_unet_s5-d16_64x64_40k_drive.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_unet_s5-d16_128x128_40k_chase_db1.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_unet_s5-d16_128x128_40k_stare.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/fcn_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_unet_s5-d16_256x256_40k_hrf.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/fcn_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_unet_s5-d16_64x64_40k_drive.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_unet_s5-d16_128x128_40k_chase_db1.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_unet_s5-d16_128x128_40k_stare.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/pspnet_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_unet_s5-d16_256x256_40k_hrf.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/unet/pspnet_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_unet_s5-d16_64x64_40k_drive.py' 2 | model = dict( 3 | decode_head=dict(loss_decode=[ 4 | dict(type='CrossEntropyLoss', loss_name='loss_ce', loss_weight=1.0), 5 | dict(type='DiceLoss', loss_name='loss_dice', loss_weight=3.0) 6 | ])) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/upernet/upernet_r18_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x1024_40k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://resnet18_v1c', 4 | backbone=dict(depth=18), 5 | decode_head=dict(in_channels=[64, 128, 256, 512]), 6 | auxiliary_head=dict(in_channels=256)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/configs/upernet/upernet_r18_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://resnet18_v1c', 4 | backbone=dict(depth=18), 5 | decode_head=dict(in_channels=[64, 128, 256, 512]), 6 | auxiliary_head=dict(in_channels=256)) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | neck=None) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | neck=None) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1, final_norm=True)) 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | ) 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_small_patch16_224-cd65a155.pth', 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)) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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='pretrain/deit_small_patch16_224-cd65a155.pth', 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)) 9 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/demo/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/demo/demo.png -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../images/mmsegmentation.png"); 3 | background-size: 201px 40px; 4 | height: 40px; 5 | width: 201px; 6 | } 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/en/_static/images/mmsegmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/docs/en/_static/images/mmsegmentation.png -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/en/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/zh_cn/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../images/mmsegmentation.png"); 3 | background-size: 201px 40px; 4 | height: 40px; 5 | width: 201px; 6 | } 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/zh_cn/_static/images/mmsegmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/docs/zh_cn/_static/images/mmsegmentation.png -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/zh_cn/imgs/qq_group_qrcode.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/docs/zh_cn/imgs/qq_group_qrcode.jpg -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/zh_cn/imgs/zhihu_qrcode.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/docs/zh_cn/imgs/zhihu_qrcode.jpg -------------------------------------------------------------------------------- /segmentation/mmsegmentation/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/hook/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .wandblogger_hook import MMSegWandbHook 3 | 4 | __all__ = ['MMSegWandbHook'] 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/optimizers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .layer_decay_optimizer_constructor import ( 3 | LayerDecayOptimizerConstructor, LearningRateDecayOptimizerConstructor) 4 | 5 | __all__ = [ 6 | 'LearningRateDecayOptimizerConstructor', 'LayerDecayOptimizerConstructor' 7 | ] 8 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/seg/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .builder import build_pixel_sampler 3 | from .sampler import BasePixelSampler, OHEMPixelSampler 4 | 5 | __all__ = ['build_pixel_sampler', 'BasePixelSampler', 'OHEMPixelSampler'] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/seg/builder.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from mmcv.utils import Registry, build_from_cfg 3 | 4 | PIXEL_SAMPLERS = Registry('pixel sampler') 5 | 6 | 7 | def build_pixel_sampler(cfg, **default_args): 8 | """Build pixel sampler for segmentation map.""" 9 | return build_from_cfg(cfg, PIXEL_SAMPLERS, default_args) 10 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/seg/sampler/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .base_pixel_sampler import BasePixelSampler 3 | from .ohem_pixel_sampler import OHEMPixelSampler 4 | 5 | __all__ = ['BasePixelSampler', 'OHEMPixelSampler'] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/seg/sampler/base_pixel_sampler.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from abc import ABCMeta, abstractmethod 3 | 4 | 5 | class BasePixelSampler(metaclass=ABCMeta): 6 | """Base class of pixel sampler.""" 7 | 8 | def __init__(self, **kwargs): 9 | pass 10 | 11 | @abstractmethod 12 | def sample(self, seg_logit, seg_label): 13 | """Placeholder for sample function.""" 14 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/core/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .dist_util import check_dist_init, sync_random_seed 3 | from .misc import add_prefix 4 | 5 | __all__ = ['add_prefix', 'check_dist_init', 'sync_random_seed'] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | # flake8: noqa 3 | import warnings 4 | 5 | from .formatting import * 6 | 7 | warnings.warn('DeprecationWarning: mmseg.datasets.pipelines.formating will be ' 8 | 'deprecated in 2021, please replace it with ' 9 | 'mmseg.datasets.pipelines.formatting.') 10 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .distributed_sampler import DistributedSampler 3 | 4 | __all__ = ['DistributedSampler'] 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/models/necks/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .featurepyramid import Feature2Pyramid 3 | from .fpn import FPN 4 | from .ic_neck import ICNeck 5 | from .jpu import JPU 6 | from .mla_neck import MLANeck 7 | from .multilevel_neck import MultiLevelNeck 8 | 9 | __all__ = [ 10 | 'FPN', 'MultiLevelNeck', 'MLANeck', 'ICNeck', 'JPU', 'Feature2Pyramid' 11 | ] 12 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/models/segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .base import BaseSegmentor 3 | from .cascade_encoder_decoder import CascadeEncoderDecoder 4 | from .encoder_decoder import EncoderDecoder 5 | 6 | __all__ = ['BaseSegmentor', 'EncoderDecoder', 'CascadeEncoderDecoder'] 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/mmseg/ops/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .encoding import Encoding 3 | from .wrappers import Upsample, resize 4 | 5 | __all__ = ['Upsample', 'resize', 'Encoding'] 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/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 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/optional.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/docs.txt: -------------------------------------------------------------------------------- 1 | docutils==0.16.0 2 | myst-parser 3 | -e git+https://github.com/gaotongxiao/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme 4 | sphinx==4.0.2 5 | sphinx_copybutton 6 | sphinx_markdown_tables 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcls>=0.20.1 2 | mmcv-full>=1.4.4,<1.7.0 3 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | prettytable 3 | torch 4 | torchvision 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | mmcls>=0.20.1 3 | numpy 4 | packaging 5 | prettytable 6 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | codecov 2 | flake8 3 | interrogate 4 | pytest 5 | xdoctest>=0.10.0 6 | yapf 7 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/resources/3dogs.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/resources/3dogs.jpg -------------------------------------------------------------------------------- /segmentation/mmsegmentation/resources/3dogs_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/resources/3dogs_mask.png -------------------------------------------------------------------------------- /segmentation/mmsegmentation/resources/mmseg-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/resources/mmseg-logo.png -------------------------------------------------------------------------------- /segmentation/mmsegmentation/resources/seg_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SLDGroup/GradientFilter-CVPR23/33be753cdb10c88c24d736fae7acb699033e6886/segmentation/mmsegmentation/resources/seg_demo.gif -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import all_zeros, check_norm_state, is_block, is_norm 3 | 4 | __all__ = ['is_norm', 'is_block', 'all_zeros', 'check_norm_state'] 5 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_losses/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_necks/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /segmentation/mmsegmentation/tests/test_models/test_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | --------------------------------------------------------------------------------