├── README.md ├── convert2onnx.py ├── latency.png ├── sea_attention.png ├── seaformer-cls ├── .gitattributes ├── .github │ ├── FUNDING.yml │ ├── ISSUE_TEMPLATE │ │ ├── bug_report.md │ │ ├── config.yml │ │ └── feature_request.md │ └── workflows │ │ └── tests.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── avg_checkpoints.py ├── benchmark.py ├── clean_checkpoint.py ├── convert │ ├── convert_from_mxnet.py │ └── convert_nest_flax.py ├── distributed_train.sh ├── docs │ ├── archived_changes.md │ ├── changes.md │ ├── feature_extraction.md │ ├── index.md │ ├── javascripts │ │ └── tables.js │ ├── models.md │ ├── models │ │ ├── .pages │ │ ├── .templates │ │ │ ├── code_snippets.md │ │ │ ├── generate_readmes.py │ │ │ └── models │ │ │ │ ├── adversarial-inception-v3.md │ │ │ │ ├── advprop.md │ │ │ │ ├── big-transfer.md │ │ │ │ ├── csp-darknet.md │ │ │ │ ├── csp-resnet.md │ │ │ │ ├── csp-resnext.md │ │ │ │ ├── densenet.md │ │ │ │ ├── dla.md │ │ │ │ ├── dpn.md │ │ │ │ ├── ecaresnet.md │ │ │ │ ├── efficientnet-pruned.md │ │ │ │ ├── efficientnet.md │ │ │ │ ├── ensemble-adversarial.md │ │ │ │ ├── ese-vovnet.md │ │ │ │ ├── fbnet.md │ │ │ │ ├── gloun-inception-v3.md │ │ │ │ ├── gloun-resnet.md │ │ │ │ ├── gloun-resnext.md │ │ │ │ ├── gloun-senet.md │ │ │ │ ├── gloun-seresnext.md │ │ │ │ ├── gloun-xception.md │ │ │ │ ├── hrnet.md │ │ │ │ ├── ig-resnext.md │ │ │ │ ├── inception-resnet-v2.md │ │ │ │ ├── inception-v3.md │ │ │ │ ├── inception-v4.md │ │ │ │ ├── legacy-se-resnet.md │ │ │ │ ├── legacy-se-resnext.md │ │ │ │ ├── legacy-senet.md │ │ │ │ ├── mixnet.md │ │ │ │ ├── mnasnet.md │ │ │ │ ├── mobilenet-v2.md │ │ │ │ ├── mobilenet-v3.md │ │ │ │ ├── nasnet.md │ │ │ │ ├── noisy-student.md │ │ │ │ ├── pnasnet.md │ │ │ │ ├── regnetx.md │ │ │ │ ├── regnety.md │ │ │ │ ├── res2net.md │ │ │ │ ├── res2next.md │ │ │ │ ├── resnest.md │ │ │ │ ├── resnet-d.md │ │ │ │ ├── resnet.md │ │ │ │ ├── resnext.md │ │ │ │ ├── rexnet.md │ │ │ │ ├── se-resnet.md │ │ │ │ ├── selecsls.md │ │ │ │ ├── seresnext.md │ │ │ │ ├── skresnet.md │ │ │ │ ├── skresnext.md │ │ │ │ ├── spnasnet.md │ │ │ │ ├── ssl-resnet.md │ │ │ │ ├── ssl-resnext.md │ │ │ │ ├── swsl-resnet.md │ │ │ │ ├── swsl-resnext.md │ │ │ │ ├── tf-efficientnet-condconv.md │ │ │ │ ├── tf-efficientnet-lite.md │ │ │ │ ├── tf-efficientnet.md │ │ │ │ ├── tf-inception-v3.md │ │ │ │ ├── tf-mixnet.md │ │ │ │ ├── tf-mobilenet-v3.md │ │ │ │ ├── tresnet.md │ │ │ │ ├── vision-transformer.md │ │ │ │ ├── wide-resnet.md │ │ │ │ └── xception.md │ │ ├── adversarial-inception-v3.md │ │ ├── advprop.md │ │ ├── big-transfer.md │ │ ├── csp-darknet.md │ │ ├── csp-resnet.md │ │ ├── csp-resnext.md │ │ ├── densenet.md │ │ ├── dla.md │ │ ├── dpn.md │ │ ├── ecaresnet.md │ │ ├── efficientnet-pruned.md │ │ ├── efficientnet.md │ │ ├── ensemble-adversarial.md │ │ ├── ese-vovnet.md │ │ ├── fbnet.md │ │ ├── gloun-inception-v3.md │ │ ├── gloun-resnet.md │ │ ├── gloun-resnext.md │ │ ├── gloun-senet.md │ │ ├── gloun-seresnext.md │ │ ├── gloun-xception.md │ │ ├── hrnet.md │ │ ├── ig-resnext.md │ │ ├── inception-resnet-v2.md │ │ ├── inception-v3.md │ │ ├── inception-v4.md │ │ ├── legacy-se-resnet.md │ │ ├── legacy-se-resnext.md │ │ ├── legacy-senet.md │ │ ├── mixnet.md │ │ ├── mnasnet.md │ │ ├── mobilenet-v2.md │ │ ├── mobilenet-v3.md │ │ ├── nasnet.md │ │ ├── noisy-student.md │ │ ├── pnasnet.md │ │ ├── regnetx.md │ │ ├── regnety.md │ │ ├── res2net.md │ │ ├── res2next.md │ │ ├── resnest.md │ │ ├── resnet-d.md │ │ ├── resnet.md │ │ ├── resnext.md │ │ ├── rexnet.md │ │ ├── se-resnet.md │ │ ├── selecsls.md │ │ ├── seresnext.md │ │ ├── skresnet.md │ │ ├── skresnext.md │ │ ├── spnasnet.md │ │ ├── ssl-resnet.md │ │ ├── ssl-resnext.md │ │ ├── swsl-resnet.md │ │ ├── swsl-resnext.md │ │ ├── tf-efficientnet-condconv.md │ │ ├── tf-efficientnet-lite.md │ │ ├── tf-efficientnet.md │ │ ├── tf-inception-v3.md │ │ ├── tf-mixnet.md │ │ ├── tf-mobilenet-v3.md │ │ ├── tresnet.md │ │ ├── vision-transformer.md │ │ ├── wide-resnet.md │ │ └── xception.md │ ├── results.md │ ├── scripts.md │ └── training_hparam_examples.md ├── hubconf.py ├── inference.py ├── mkdocs.yml ├── model-index.yml ├── requirements-docs.txt ├── requirements-modelindex.txt ├── requirements.txt ├── results │ ├── README.md │ ├── benchmark-infer-amp-nchw-pt110-cu113-rtx3090.csv │ ├── benchmark-infer-amp-nchw-pt111-cu113-rtx3090.csv │ ├── benchmark-infer-amp-nhwc-pt110-cu113-rtx3090.csv │ ├── benchmark-infer-amp-nhwc-pt111-cu113-rtx3090.csv │ ├── benchmark-train-amp-nchw-pt110-cu113-rtx3090.csv │ ├── benchmark-train-amp-nchw-pt111-cu113-rtx3090.csv │ ├── benchmark-train-amp-nhwc-pt110-cu113-rtx3090.csv │ ├── benchmark-train-amp-nhwc-pt111-cu113-rtx3090.csv │ ├── generate_csv_results.py │ ├── imagenet21k_goog_synsets.txt │ ├── imagenet_a_indices.txt │ ├── imagenet_a_synsets.txt │ ├── imagenet_r_indices.txt │ ├── imagenet_r_synsets.txt │ ├── imagenet_real_labels.json │ ├── imagenet_synsets.txt │ ├── model_metadata-in1k.csv │ ├── results-imagenet-a-clean.csv │ ├── results-imagenet-a.csv │ ├── results-imagenet-r-clean.csv │ ├── results-imagenet-r.csv │ ├── results-imagenet-real.csv │ ├── results-imagenet.csv │ ├── results-imagenetv2-matched-frequency.csv │ └── results-sketch.csv ├── seaformer.py ├── setup.cfg ├── setup.py ├── tests │ ├── __init__.py │ ├── test_layers.py │ ├── test_models.py │ ├── test_optim.py │ └── test_utils.py ├── timm │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── auto_augment.py │ │ ├── config.py │ │ ├── constants.py │ │ ├── dataset.py │ │ ├── dataset_factory.py │ │ ├── distributed_sampler.py │ │ ├── loader.py │ │ ├── mixup.py │ │ ├── parsers │ │ │ ├── __init__.py │ │ │ ├── class_map.py │ │ │ ├── constants.py │ │ │ ├── parser.py │ │ │ ├── parser_factory.py │ │ │ ├── parser_image_folder.py │ │ │ ├── parser_image_in_tar.py │ │ │ ├── parser_image_tar.py │ │ │ └── parser_tfds.py │ │ ├── random_erasing.py │ │ ├── real_labels.py │ │ ├── tf_preprocessing.py │ │ ├── transforms.py │ │ └── transforms_factory.py │ ├── loss │ │ ├── __init__.py │ │ ├── asymmetric_loss.py │ │ ├── binary_cross_entropy.py │ │ ├── cross_entropy.py │ │ └── jsd.py │ ├── models │ │ ├── __init__.py │ │ ├── beit.py │ │ ├── byoanet.py │ │ ├── byobnet.py │ │ ├── cait.py │ │ ├── coat.py │ │ ├── convit.py │ │ ├── convmixer.py │ │ ├── convnext.py │ │ ├── crossvit.py │ │ ├── cspnet.py │ │ ├── deit.py │ │ ├── densenet.py │ │ ├── dla.py │ │ ├── dpn.py │ │ ├── efficientnet.py │ │ ├── efficientnet_blocks.py │ │ ├── efficientnet_builder.py │ │ ├── factory.py │ │ ├── features.py │ │ ├── fx_features.py │ │ ├── ghostnet.py │ │ ├── gluon_resnet.py │ │ ├── gluon_xception.py │ │ ├── hardcorenas.py │ │ ├── helpers.py │ │ ├── hrnet.py │ │ ├── hub.py │ │ ├── inception_resnet_v2.py │ │ ├── inception_v3.py │ │ ├── inception_v4.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── activations.py │ │ │ ├── activations_jit.py │ │ │ ├── activations_me.py │ │ │ ├── adaptive_avgmax_pool.py │ │ │ ├── attention_pool2d.py │ │ │ ├── blur_pool.py │ │ │ ├── bottleneck_attn.py │ │ │ ├── cbam.py │ │ │ ├── classifier.py │ │ │ ├── cond_conv2d.py │ │ │ ├── config.py │ │ │ ├── conv2d_same.py │ │ │ ├── conv_bn_act.py │ │ │ ├── create_act.py │ │ │ ├── create_attn.py │ │ │ ├── create_conv2d.py │ │ │ ├── create_norm_act.py │ │ │ ├── drop.py │ │ │ ├── eca.py │ │ │ ├── evo_norm.py │ │ │ ├── filter_response_norm.py │ │ │ ├── gather_excite.py │ │ │ ├── global_context.py │ │ │ ├── halo_attn.py │ │ │ ├── helpers.py │ │ │ ├── inplace_abn.py │ │ │ ├── lambda_layer.py │ │ │ ├── linear.py │ │ │ ├── median_pool.py │ │ │ ├── mixed_conv2d.py │ │ │ ├── ml_decoder.py │ │ │ ├── mlp.py │ │ │ ├── non_local_attn.py │ │ │ ├── norm.py │ │ │ ├── norm_act.py │ │ │ ├── padding.py │ │ │ ├── patch_embed.py │ │ │ ├── pool2d_same.py │ │ │ ├── pos_embed.py │ │ │ ├── selective_kernel.py │ │ │ ├── separable_conv.py │ │ │ ├── space_to_depth.py │ │ │ ├── split_attn.py │ │ │ ├── split_batchnorm.py │ │ │ ├── squeeze_excite.py │ │ │ ├── std_conv.py │ │ │ ├── test_time_pool.py │ │ │ ├── trace_utils.py │ │ │ └── weight_init.py │ │ ├── levit.py │ │ ├── mlp_mixer.py │ │ ├── mobilenetv3.py │ │ ├── mobilevit.py │ │ ├── nasnet.py │ │ ├── nest.py │ │ ├── nfnet.py │ │ ├── pit.py │ │ ├── pnasnet.py │ │ ├── poolformer.py │ │ ├── pruned │ │ │ ├── ecaresnet101d_pruned.txt │ │ │ ├── ecaresnet50d_pruned.txt │ │ │ ├── efficientnet_b1_pruned.txt │ │ │ ├── efficientnet_b2_pruned.txt │ │ │ └── efficientnet_b3_pruned.txt │ │ ├── registry.py │ │ ├── regnet.py │ │ ├── res2net.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ ├── resnetv2.py │ │ ├── rexnet.py │ │ ├── selecsls.py │ │ ├── senet.py │ │ ├── sknet.py │ │ ├── swin_transformer.py │ │ ├── swin_transformer_v2_cr.py │ │ ├── tnt.py │ │ ├── tresnet.py │ │ ├── twins.py │ │ ├── vgg.py │ │ ├── visformer.py │ │ ├── vision_transformer.py │ │ ├── vision_transformer_hybrid.py │ │ ├── volo.py │ │ ├── vovnet.py │ │ ├── xception.py │ │ ├── xception_aligned.py │ │ └── xcit.py │ ├── optim │ │ ├── __init__.py │ │ ├── adabelief.py │ │ ├── adafactor.py │ │ ├── adahessian.py │ │ ├── adamp.py │ │ ├── adamw.py │ │ ├── lamb.py │ │ ├── lars.py │ │ ├── lookahead.py │ │ ├── madgrad.py │ │ ├── nadam.py │ │ ├── nvnovograd.py │ │ ├── optim_factory.py │ │ ├── radam.py │ │ ├── rmsprop_tf.py │ │ └── sgdp.py │ ├── scheduler │ │ ├── __init__.py │ │ ├── cosine_lr.py │ │ ├── multistep_lr.py │ │ ├── plateau_lr.py │ │ ├── poly_lr.py │ │ ├── scheduler.py │ │ ├── scheduler_factory.py │ │ ├── step_lr.py │ │ └── tanh_lr.py │ ├── utils │ │ ├── __init__.py │ │ ├── agc.py │ │ ├── checkpoint_saver.py │ │ ├── clip_grad.py │ │ ├── cuda.py │ │ ├── distributed.py │ │ ├── jit.py │ │ ├── log.py │ │ ├── metrics.py │ │ ├── misc.py │ │ ├── model.py │ │ ├── model_ema.py │ │ ├── random.py │ │ └── summary.py │ └── version.py ├── train.py └── validate.py ├── seaformer-seg ├── CITATION.cff ├── LICENSE ├── MANIFEST.in ├── README.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_1024x1024.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── cityscapes_832x832.py │ │ │ ├── coco-stuff10k.py │ │ │ ├── coco-stuff164k.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── loveda.py │ │ │ ├── pascal_context.py │ │ │ ├── pascal_context_59.py │ │ │ ├── pascal_voc12.py │ │ │ ├── pascal_voc12_aug.py │ │ │ └── stare.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── ann_r50-d8.py │ │ │ ├── apcnet_r50-d8.py │ │ │ ├── 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 │ │ │ ├── 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 │ │ │ ├── setr_mla.py │ │ │ ├── setr_naive.py │ │ │ ├── setr_pup.py │ │ │ ├── upernet_r50.py │ │ │ ├── upernet_swin.py │ │ │ └── upernet_vit-b16_ln_mln.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_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 │ ├── 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 │ ├── 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_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_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_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r18-d8_512x512_80k_loveda.py │ │ ├── deeplabv3plus_r18-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context.py │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context.py │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py │ │ ├── deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py │ │ ├── deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_r50-d8_512x512_160k_ade20k.py │ │ ├── deeplabv3plus_r50-d8_512x512_20k_voc12aug.py │ │ ├── deeplabv3plus_r50-d8_512x512_40k_voc12aug.py │ │ ├── deeplabv3plus_r50-d8_512x512_80k_ade20k.py │ │ ├── deeplabv3plus_r50-d8_512x512_80k_loveda.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 │ ├── 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_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_hr18s_480x480_40k_pascal_context.py │ │ ├── fcn_hr18s_480x480_40k_pascal_context_59.py │ │ ├── fcn_hr18s_480x480_80k_pascal_context.py │ │ ├── fcn_hr18s_480x480_80k_pascal_context_59.py │ │ ├── fcn_hr18s_512x1024_160k_cityscapes.py │ │ ├── fcn_hr18s_512x1024_40k_cityscapes.py │ │ ├── fcn_hr18s_512x1024_80k_cityscapes.py │ │ ├── fcn_hr18s_512x512_160k_ade20k.py │ │ ├── fcn_hr18s_512x512_20k_voc12aug.py │ │ ├── fcn_hr18s_512x512_40k_voc12aug.py │ │ ├── fcn_hr18s_512x512_80k_ade20k.py │ │ ├── fcn_hr18s_512x512_80k_loveda.py │ │ ├── fcn_hr48_480x480_40k_pascal_context.py │ │ ├── fcn_hr48_480x480_40k_pascal_context_59.py │ │ ├── fcn_hr48_480x480_80k_pascal_context.py │ │ ├── fcn_hr48_480x480_80k_pascal_context_59.py │ │ ├── fcn_hr48_512x1024_160k_cityscapes.py │ │ ├── fcn_hr48_512x1024_40k_cityscapes.py │ │ ├── fcn_hr48_512x1024_80k_cityscapes.py │ │ ├── fcn_hr48_512x512_160k_ade20k.py │ │ ├── fcn_hr48_512x512_20k_voc12aug.py │ │ ├── fcn_hr48_512x512_40k_voc12aug.py │ │ ├── fcn_hr48_512x512_80k_ade20k.py │ │ ├── fcn_hr48_512x512_80k_loveda.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 │ ├── mobilenet_v2 │ │ ├── README.md │ │ ├── deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3_m-v2-d8_512x512_160k_ade20k.py │ │ ├── deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py │ │ ├── fcn_m-v2-d8_512x1024_80k_cityscapes.py │ │ ├── fcn_m-v2-d8_512x512_160k_ade20k.py │ │ ├── mobilenet_v2.yml │ │ ├── pspnet_m-v2-d8_512x1024_80k_cityscapes.py │ │ └── pspnet_m-v2-d8_512x512_160k_ade20k.py │ ├── mobilenet_v3 │ │ ├── README.md │ │ ├── lraspp_m-v3-d8_512x1024_320k_cityscapes.py │ │ ├── lraspp_m-v3-d8_scratch_512x1024_320k_cityscapes.py │ │ ├── lraspp_m-v3s-d8_512x1024_320k_cityscapes.py │ │ ├── lraspp_m-v3s-d8_scratch_512x1024_320k_cityscapes.py │ │ └── mobilenet_v3.yml │ ├── nonlocal_net │ │ ├── README.md │ │ ├── nonlocal_net.yml │ │ ├── nonlocal_r101-d8_512x1024_40k_cityscapes.py │ │ ├── nonlocal_r101-d8_512x1024_80k_cityscapes.py │ │ ├── nonlocal_r101-d8_512x512_160k_ade20k.py │ │ ├── nonlocal_r101-d8_512x512_20k_voc12aug.py │ │ ├── nonlocal_r101-d8_512x512_40k_voc12aug.py │ │ ├── nonlocal_r101-d8_512x512_80k_ade20k.py │ │ ├── nonlocal_r101-d8_769x769_40k_cityscapes.py │ │ ├── nonlocal_r101-d8_769x769_80k_cityscapes.py │ │ ├── nonlocal_r50-d8_512x1024_40k_cityscapes.py │ │ ├── nonlocal_r50-d8_512x1024_80k_cityscapes.py │ │ ├── nonlocal_r50-d8_512x512_160k_ade20k.py │ │ ├── nonlocal_r50-d8_512x512_20k_voc12aug.py │ │ ├── nonlocal_r50-d8_512x512_40k_voc12aug.py │ │ ├── nonlocal_r50-d8_512x512_80k_ade20k.py │ │ ├── nonlocal_r50-d8_769x769_40k_cityscapes.py │ │ └── nonlocal_r50-d8_769x769_80k_cityscapes.py │ ├── ocrnet │ │ ├── README.md │ │ ├── ocrnet.yml │ │ ├── ocrnet_hr18_512x1024_160k_cityscapes.py │ │ ├── ocrnet_hr18_512x1024_40k_cityscapes.py │ │ ├── ocrnet_hr18_512x1024_80k_cityscapes.py │ │ ├── ocrnet_hr18_512x512_160k_ade20k.py │ │ ├── ocrnet_hr18_512x512_20k_voc12aug.py │ │ ├── ocrnet_hr18_512x512_40k_voc12aug.py │ │ ├── ocrnet_hr18_512x512_80k_ade20k.py │ │ ├── ocrnet_hr18s_512x1024_160k_cityscapes.py │ │ ├── ocrnet_hr18s_512x1024_40k_cityscapes.py │ │ ├── ocrnet_hr18s_512x1024_80k_cityscapes.py │ │ ├── ocrnet_hr18s_512x512_160k_ade20k.py │ │ ├── ocrnet_hr18s_512x512_20k_voc12aug.py │ │ ├── ocrnet_hr18s_512x512_40k_voc12aug.py │ │ ├── ocrnet_hr18s_512x512_80k_ade20k.py │ │ ├── ocrnet_hr48_512x1024_160k_cityscapes.py │ │ ├── ocrnet_hr48_512x1024_40k_cityscapes.py │ │ ├── ocrnet_hr48_512x1024_80k_cityscapes.py │ │ ├── ocrnet_hr48_512x512_160k_ade20k.py │ │ ├── ocrnet_hr48_512x512_20k_voc12aug.py │ │ ├── ocrnet_hr48_512x512_40k_voc12aug.py │ │ ├── ocrnet_hr48_512x512_80k_ade20k.py │ │ ├── ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py │ │ ├── ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py │ │ └── ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py │ ├── point_rend │ │ ├── README.md │ │ ├── point_rend.yml │ │ ├── pointrend_r101_512x1024_80k_cityscapes.py │ │ ├── pointrend_r101_512x512_160k_ade20k.py │ │ ├── pointrend_r50_512x1024_80k_cityscapes.py │ │ └── pointrend_r50_512x512_160k_ade20k.py │ ├── psanet │ │ ├── README.md │ │ ├── psanet.yml │ │ ├── psanet_r101-d8_512x1024_40k_cityscapes.py │ │ ├── psanet_r101-d8_512x1024_80k_cityscapes.py │ │ ├── psanet_r101-d8_512x512_160k_ade20k.py │ │ ├── psanet_r101-d8_512x512_20k_voc12aug.py │ │ ├── psanet_r101-d8_512x512_40k_voc12aug.py │ │ ├── psanet_r101-d8_512x512_80k_ade20k.py │ │ ├── psanet_r101-d8_769x769_40k_cityscapes.py │ │ ├── psanet_r101-d8_769x769_80k_cityscapes.py │ │ ├── psanet_r50-d8_512x1024_40k_cityscapes.py │ │ ├── psanet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── psanet_r50-d8_512x512_160k_ade20k.py │ │ ├── psanet_r50-d8_512x512_20k_voc12aug.py │ │ ├── psanet_r50-d8_512x512_40k_voc12aug.py │ │ ├── psanet_r50-d8_512x512_80k_ade20k.py │ │ ├── psanet_r50-d8_769x769_40k_cityscapes.py │ │ └── psanet_r50-d8_769x769_80k_cityscapes.py │ ├── pspnet │ │ ├── README.md │ │ ├── pspnet.yml │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context.py │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context_59.py │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context.py │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context_59.py │ │ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py │ │ ├── pspnet_r101-d8_512x1024_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_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-d8_480x480_40k_pascal_context.py │ │ ├── pspnet_r50-d8_480x480_40k_pascal_context_59.py │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context.py │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context_59.py │ │ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py │ │ ├── pspnet_r50-d8_512x1024_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_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-b1_512x512_160k_ade20k.py │ │ ├── segformer_mit-b2_512x512_160k_ade20k.py │ │ ├── segformer_mit-b3_512x512_160k_ade20k.py │ │ ├── segformer_mit-b4_512x512_160k_ade20k.py │ │ ├── segformer_mit-b5_512x512_160k_ade20k.py │ │ └── segformer_mit-b5_640x640_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 │ ├── swin │ │ ├── README.md │ │ ├── swin.yml │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_1K.py │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py │ │ ├── upernet_swin_small_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ └── upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ ├── unet │ │ ├── README.md │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_stare.py │ │ ├── deeplabv3_unet_s5-d16_256x256_40k_hrf.py │ │ ├── deeplabv3_unet_s5-d16_64x64_40k_drive.py │ │ ├── fcn_unet_s5-d16_128x128_40k_chase_db1.py │ │ ├── fcn_unet_s5-d16_128x128_40k_stare.py │ │ ├── fcn_unet_s5-d16_256x256_40k_hrf.py │ │ ├── fcn_unet_s5-d16_64x64_40k_drive.py │ │ ├── pspnet_unet_s5-d16_128x128_40k_chase_db1.py │ │ ├── pspnet_unet_s5-d16_128x128_40k_stare.py │ │ ├── pspnet_unet_s5-d16_256x256_40k_hrf.py │ │ ├── pspnet_unet_s5-d16_64x64_40k_drive.py │ │ └── unet.yml │ ├── upernet │ │ ├── README.md │ │ ├── upernet.yml │ │ ├── upernet_r101_512x1024_40k_cityscapes.py │ │ ├── upernet_r101_512x1024_80k_cityscapes.py │ │ ├── upernet_r101_512x512_160k_ade20k.py │ │ ├── upernet_r101_512x512_20k_voc12aug.py │ │ ├── upernet_r101_512x512_40k_voc12aug.py │ │ ├── upernet_r101_512x512_80k_ade20k.py │ │ ├── upernet_r101_769x769_40k_cityscapes.py │ │ ├── upernet_r101_769x769_80k_cityscapes.py │ │ ├── upernet_r50_512x1024_40k_cityscapes.py │ │ ├── upernet_r50_512x1024_80k_cityscapes.py │ │ ├── upernet_r50_512x512_160k_ade20k.py │ │ ├── upernet_r50_512x512_20k_voc12aug.py │ │ ├── upernet_r50_512x512_40k_voc12aug.py │ │ ├── upernet_r50_512x512_80k_ade20k.py │ │ ├── upernet_r50_769x769_40k_cityscapes.py │ │ └── upernet_r50_769x769_80k_cityscapes.py │ └── vit │ │ ├── README.md │ │ ├── upernet_deit-b16_512x512_160k_ade20k.py │ │ ├── upernet_deit-b16_512x512_80k_ade20k.py │ │ ├── upernet_deit-b16_ln_mln_512x512_160k_ade20k.py │ │ ├── upernet_deit-b16_mln_512x512_160k_ade20k.py │ │ ├── upernet_deit-s16_512x512_160k_ade20k.py │ │ ├── upernet_deit-s16_512x512_80k_ade20k.py │ │ ├── upernet_deit-s16_ln_mln_512x512_160k_ade20k.py │ │ ├── upernet_deit-s16_mln_512x512_160k_ade20k.py │ │ ├── upernet_vit-b16_ln_mln_512x512_160k_ade20k.py │ │ ├── upernet_vit-b16_mln_512x512_160k_ade20k.py │ │ ├── upernet_vit-b16_mln_512x512_80k_ade20k.py │ │ └── vit.yml ├── demo │ ├── MMSegmentation_Tutorial.ipynb │ ├── demo.png │ ├── image_demo.py │ └── inference_demo.ipynb ├── docker │ ├── Dockerfile │ └── serve │ │ ├── Dockerfile │ │ ├── config.properties │ │ └── entrypoint.sh ├── imgs │ ├── speed_iou.jpg │ └── speed_iou.pdf ├── local_configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_1024x1024.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── cityscapes_832x832.py │ │ │ ├── coco-stuff10k.py │ │ │ ├── coco-stuff164k.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── pascal_context.py │ │ │ ├── pascal_context_59.py │ │ │ ├── pascal_voc12.py │ │ │ ├── pascal_voc12_aug.py │ │ │ └── stare.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── ann_r50-d8.py │ │ │ ├── apcnet_r50-d8.py │ │ │ ├── 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 │ │ │ ├── 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 │ │ │ ├── setr_mla.py │ │ │ ├── setr_naive.py │ │ │ ├── setr_pup.py │ │ │ ├── upernet_r50.py │ │ │ ├── upernet_swin.py │ │ │ └── upernet_vit-b16_ln_mln.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ ├── seaformer │ │ ├── seaformer_base.py │ │ ├── seaformer_base_1024x1024_160k_1x8city.py │ │ ├── seaformer_base_1024x512_160k_1x8city.py │ │ ├── seaformer_base_480x480_80k_2x8_pascalcontext.py │ │ ├── seaformer_base_480x480_80k_2x8_pascalcontext_59.py │ │ ├── seaformer_base_512x512_160k_2x8_ade20k.py │ │ ├── seaformer_base_512x512_160k_4x8_ade20k.py │ │ ├── seaformer_base_512x512_80k_2x8_cocostuff10k.py │ │ ├── seaformer_large.py │ │ ├── seaformer_large_512x512_160k_2x8_ade20k.py │ │ ├── seaformer_large_512x512_160k_4x8_ade20k.py │ │ ├── seaformer_small.py │ │ ├── seaformer_small_1024x1024_160k_1x8city.py │ │ ├── seaformer_small_1024x512_160k_1x8city.py │ │ ├── seaformer_small_480x480_80k_2x8_pascalcontext.py │ │ ├── seaformer_small_480x480_80k_2x8_pascalcontext_59.py │ │ ├── seaformer_small_512x512_160k_2x8_ade20k.py │ │ ├── seaformer_small_512x512_160k_4x8_ade20k.py │ │ ├── seaformer_small_512x512_80k_2x8_cocostuff10k.py │ │ ├── seaformer_tiny.py │ │ ├── seaformer_tiny_480x480_80k_2x8_pascalcontext.py │ │ ├── seaformer_tiny_480x480_80k_2x8_pascalcontext_59.py │ │ ├── seaformer_tiny_512x512_160k_2x8_ade20k.py │ │ ├── seaformer_tiny_512x512_160k_4x8_ade20k.py │ │ └── seaformer_tiny_512x512_80k_2x8_cocostuff10k.py │ └── topformer │ │ ├── topformer_base.py │ │ ├── topformer_base_1024x512_80k_2x8city.py │ │ ├── topformer_base_480x480_80k_2x8_pascalcontext.py │ │ ├── topformer_base_480x480_80k_2x8_pascalcontext_59.py │ │ ├── topformer_base_512x512_160k_2x8_ade20k.py │ │ ├── topformer_base_512x512_160k_4x8_ade20k.py │ │ ├── topformer_base_512x512_80k_2x8_cocostuff10k.py │ │ ├── topformer_small.py │ │ ├── topformer_small_480x480_80k_2x8_pascalcontext.py │ │ ├── topformer_small_480x480_80k_2x8_pascalcontext_59.py │ │ ├── topformer_small_512x512_160k_2x8_ade20k.py │ │ ├── topformer_small_512x512_80k_2x8_cocostuff10k.py │ │ ├── topformer_tiny.py │ │ ├── topformer_tiny_1024x512_80k_2x8_city.py │ │ ├── topformer_tiny_448x448_160k_2x8_ade20k.py │ │ ├── topformer_tiny_480x480_80k_2x8_pascalcontext.py │ │ ├── topformer_tiny_480x480_80k_2x8_pascalcontext_59.py │ │ ├── topformer_tiny_512x512_160k_2x8_ade20k.py │ │ └── topformer_tiny_512x512_80k_2x8_cocostuff10k.py ├── mmseg │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── test.py │ │ └── train.py │ ├── core │ │ ├── __init__.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── class_names.py │ │ │ ├── eval_hooks.py │ │ │ └── metrics.py │ │ ├── seg │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── sampler │ │ │ │ ├── __init__.py │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ └── ohem_pixel_sampler.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── misc.py │ ├── datasets │ │ ├── __init__.py │ │ ├── ade.py │ │ ├── builder.py │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── coco_stuff.py │ │ ├── custom.py │ │ ├── dark_zurich.py │ │ ├── dataset_wrappers.py │ │ ├── drive.py │ │ ├── hrf.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 │ │ ├── stare.py │ │ └── voc.py │ ├── models │ │ ├── __init__.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── bisenetv1.py │ │ │ ├── bisenetv2.py │ │ │ ├── cgnet.py │ │ │ ├── fast_scnn.py │ │ │ ├── hrnet.py │ │ │ ├── icnet.py │ │ │ ├── mit.py │ │ │ ├── mobilenet_v2.py │ │ │ ├── mobilenet_v3.py │ │ │ ├── resnest.py │ │ │ ├── resnet.py │ │ │ ├── resnext.py │ │ │ ├── seaformer.py │ │ │ ├── swin.py │ │ │ ├── timm_backbone.py │ │ │ ├── topformer.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 │ │ │ ├── light_head.py │ │ │ ├── lraspp_head.py │ │ │ ├── nl_head.py │ │ │ ├── ocr_head.py │ │ │ ├── point_head.py │ │ │ ├── psa_head.py │ │ │ ├── psp_head.py │ │ │ ├── segformer_head.py │ │ │ ├── sep_aspp_head.py │ │ │ ├── sep_fcn_head.py │ │ │ ├── setr_mla_head.py │ │ │ ├── setr_up_head.py │ │ │ ├── simple_head.py │ │ │ └── uper_head.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── accuracy.py │ │ │ ├── cross_entropy_loss.py │ │ │ ├── dice_loss.py │ │ │ ├── lovasz_loss.py │ │ │ └── utils.py │ │ ├── necks │ │ │ ├── __init__.py │ │ │ ├── fpn.py │ │ │ ├── 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 │ └── version.py ├── model-index.yml ├── pytest.ini ├── requirements.txt ├── requirements │ ├── docs.txt │ ├── mminstall.txt │ ├── optional.txt │ ├── readthedocs.txt │ ├── runtime.txt │ └── tests.txt ├── setup.cfg ├── setup.py └── tools │ ├── analyze_logs.py │ ├── benchmark.py │ ├── browse_dataset.py │ ├── convert_datasets │ ├── chase_db1.py │ ├── cityscapes.py │ ├── coco_stuff10k.py │ ├── coco_stuff164k.py │ ├── drive.py │ ├── hrf.py │ ├── loveda.py │ ├── pascal_context.py │ ├── stare.py │ └── voc_aug.py │ ├── convert_state_dict.py │ ├── deploy_test.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── get_flops.py │ ├── model_converters │ ├── mit2mmseg.py │ ├── swin2mmseg.py │ └── vit2mmseg.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 └── seaformer.png /latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/latency.png -------------------------------------------------------------------------------- /sea_attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/sea_attention.png -------------------------------------------------------------------------------- /seaformer-cls/.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-documentation 2 | -------------------------------------------------------------------------------- /seaformer-cls/.github/FUNDING.yml: -------------------------------------------------------------------------------- 1 | # These are supported funding model platforms 2 | github: rwightman 3 | -------------------------------------------------------------------------------- /seaformer-cls/.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | contact_links: 3 | - name: Community Discussions 4 | url: https://github.com/rwightman/pytorch-image-models/discussions 5 | about: Issues are for features and bugs. Questions can be asked in Discussions. 6 | -------------------------------------------------------------------------------- /seaformer-cls/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include timm/models/pruned/*.txt 2 | 3 | -------------------------------------------------------------------------------- /seaformer-cls/distributed_train.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | NUM_PROC=$1 3 | shift 4 | python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@" 5 | 6 | -------------------------------------------------------------------------------- /seaformer-cls/docs/javascripts/tables.js: -------------------------------------------------------------------------------- 1 | app.location$.subscribe(function() { 2 | var tables = document.querySelectorAll("article table") 3 | tables.forEach(function(table) { 4 | new Tablesort(table) 5 | }) 6 | }) -------------------------------------------------------------------------------- /seaformer-cls/docs/models/.pages: -------------------------------------------------------------------------------- 1 | title: Model Pages -------------------------------------------------------------------------------- /seaformer-cls/hubconf.py: -------------------------------------------------------------------------------- 1 | dependencies = ['torch'] 2 | from timm.models import registry 3 | 4 | globals().update(registry._model_entrypoints) 5 | -------------------------------------------------------------------------------- /seaformer-cls/requirements-docs.txt: -------------------------------------------------------------------------------- 1 | mkdocs 2 | mkdocs-material 3 | mdx_truly_sane_lists 4 | mkdocs-awesome-pages-plugin -------------------------------------------------------------------------------- /seaformer-cls/requirements-modelindex.txt: -------------------------------------------------------------------------------- 1 | model-index==0.1.10 2 | jinja2==2.11.3 3 | -------------------------------------------------------------------------------- /seaformer-cls/requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.4.0 2 | torchvision>=0.5.0 3 | pyyaml 4 | -------------------------------------------------------------------------------- /seaformer-cls/setup.cfg: -------------------------------------------------------------------------------- 1 | [dist_conda] 2 | 3 | conda_name_differences = 'torch:pytorch' 4 | channels = pytorch 5 | noarch = True 6 | -------------------------------------------------------------------------------- /seaformer-cls/tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/seaformer-cls/tests/__init__.py -------------------------------------------------------------------------------- /seaformer-cls/timm/__init__.py: -------------------------------------------------------------------------------- 1 | from .version import __version__ 2 | from .models import create_model, list_models, is_model, list_modules, model_entrypoint, \ 3 | is_scriptable, is_exportable, set_scriptable, set_exportable, has_pretrained_cfg_key, is_pretrained_cfg_key, \ 4 | get_pretrained_cfg_value, is_model_pretrained 5 | -------------------------------------------------------------------------------- /seaformer-cls/timm/data/constants.py: -------------------------------------------------------------------------------- 1 | DEFAULT_CROP_PCT = 0.875 2 | IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406) 3 | IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225) 4 | IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5) 5 | IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5) 6 | IMAGENET_DPN_MEAN = (124 / 255, 117 / 255, 104 / 255) 7 | IMAGENET_DPN_STD = tuple([1 / (.0167 * 255)] * 3) 8 | -------------------------------------------------------------------------------- /seaformer-cls/timm/data/parsers/__init__.py: -------------------------------------------------------------------------------- 1 | from .parser_factory import create_parser 2 | -------------------------------------------------------------------------------- /seaformer-cls/timm/data/parsers/constants.py: -------------------------------------------------------------------------------- 1 | IMG_EXTENSIONS = ('.png', '.jpg', '.jpeg') 2 | -------------------------------------------------------------------------------- /seaformer-cls/timm/loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .asymmetric_loss import AsymmetricLossMultiLabel, AsymmetricLossSingleLabel 2 | from .binary_cross_entropy import BinaryCrossEntropy 3 | from .cross_entropy import LabelSmoothingCrossEntropy, SoftTargetCrossEntropy 4 | from .jsd import JsdCrossEntropy 5 | -------------------------------------------------------------------------------- /seaformer-cls/timm/models/layers/trace_utils.py: -------------------------------------------------------------------------------- 1 | try: 2 | from torch import _assert 3 | except ImportError: 4 | def _assert(condition: bool, message: str): 5 | assert condition, message 6 | 7 | 8 | def _float_to_int(x: float) -> int: 9 | """ 10 | Symbolic tracing helper to substitute for inbuilt `int`. 11 | Hint: Inbuilt `int` can't accept an argument of type `Proxy` 12 | """ 13 | return int(x) 14 | -------------------------------------------------------------------------------- /seaformer-cls/timm/scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | from .cosine_lr import CosineLRScheduler 2 | from .multistep_lr import MultiStepLRScheduler 3 | from .plateau_lr import PlateauLRScheduler 4 | from .poly_lr import PolyLRScheduler 5 | from .step_lr import StepLRScheduler 6 | from .tanh_lr import TanhLRScheduler 7 | 8 | from .scheduler_factory import create_scheduler 9 | -------------------------------------------------------------------------------- /seaformer-cls/timm/utils/random.py: -------------------------------------------------------------------------------- 1 | import random 2 | import numpy as np 3 | import torch 4 | 5 | 6 | def random_seed(seed=42, rank=0): 7 | torch.manual_seed(seed + rank) 8 | np.random.seed(seed + rank) 9 | random.seed(seed + rank) 10 | -------------------------------------------------------------------------------- /seaformer-cls/timm/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.6.1' 2 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- 1 | # yapf:disable 2 | log_config = dict( 3 | interval=50, 4 | hooks=[ 5 | dict(type='TextLoggerHook', by_epoch=False), 6 | # dict(type='TensorboardLoggerHook') 7 | ]) 8 | # yapf:enable 9 | dist_params = dict(backend='nccl') 10 | log_level = 'INFO' 11 | load_from = None 12 | resume_from = None 13 | workflow = [('train', 1)] 14 | cudnn_benchmark = True 15 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=160000) 8 | checkpoint_config = dict(by_epoch=False, interval=16000) 9 | evaluation = dict(interval=16000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=20000) 8 | checkpoint_config = dict(by_epoch=False, interval=2000) 9 | evaluation = dict(interval=2000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=320000) 8 | checkpoint_config = dict(by_epoch=False, interval=32000) 9 | evaluation = dict(interval=32000, metric='mIoU') 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=40000) 8 | checkpoint_config = dict(by_epoch=False, interval=4000) 9 | evaluation = dict(interval=4000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=80000) 8 | checkpoint_config = dict(by_epoch=False, interval=8000) 9 | evaluation = dict(interval=8000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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=8, 5 | ) 6 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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=8, 11 | ) 12 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3/deeplabv3_r101-d16-mg124_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://resnet101_v1c', 4 | backbone=dict( 5 | depth=101, 6 | dilations=(1, 1, 1, 2), 7 | strides=(1, 2, 2, 1), 8 | multi_grid=(1, 2, 4)), 9 | decode_head=dict( 10 | dilations=(1, 6, 12, 18), 11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000))) 12 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3/deeplabv3_r101-d16-mg124_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://resnet101_v1c', 4 | backbone=dict( 5 | depth=101, 6 | dilations=(1, 1, 1, 2), 7 | strides=(1, 2, 2, 1), 8 | multi_grid=(1, 2, 4)), 9 | decode_head=dict( 10 | dilations=(1, 6, 12, 18), 11 | sampler=dict(type='OHEMPixelSampler', min_kept=100000))) 12 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3/deeplabv3_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3/deeplabv3_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3plus/deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet18', 4 | backbone=dict(type='ResNet', depth=18), 5 | decode_head=dict( 6 | c1_in_channels=64, 7 | c1_channels=12, 8 | in_channels=512, 9 | channels=128, 10 | ), 11 | auxiliary_head=dict(in_channels=256, channels=64)) 12 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3plus/deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet18', 4 | backbone=dict(type='ResNet', depth=18), 5 | decode_head=dict( 6 | c1_in_channels=64, 7 | c1_channels=12, 8 | in_channels=512, 9 | channels=128, 10 | ), 11 | auxiliary_head=dict(in_channels=256, channels=64)) 12 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3plus/deeplabv3plus_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/deeplabv3plus/deeplabv3plus_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/fcn/fcn_r50-d8_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_context.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=60), 7 | auxiliary_head=dict(num_classes=60), 8 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320))) 9 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/fcn/fcn_r50-d8_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/pascal_context.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=60), 7 | auxiliary_head=dict(num_classes=60), 8 | test_cfg=dict(mode='slide', crop_size=(480, 480), stride=(320, 320))) 9 | optimizer = dict(type='SGD', lr=0.004, momentum=0.9, weight_decay=0.0001) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_480x480_40k_pascal_context.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_480x480_40k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_480x480_40k_pascal_context_59.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_480x480_80k_pascal_context.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_480x480_80k_pascal_context_59.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x1024_160k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x1024_40k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_160k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_20k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_40k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr18s_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_80k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr48_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_160k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w48', 4 | backbone=dict( 5 | extra=dict( 6 | stage2=dict(num_channels=(48, 96)), 7 | stage3=dict(num_channels=(48, 96, 192)), 8 | stage4=dict(num_channels=(48, 96, 192, 384)))), 9 | decode_head=dict( 10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384]))) 11 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr48_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_20k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w48', 4 | backbone=dict( 5 | extra=dict( 6 | stage2=dict(num_channels=(48, 96)), 7 | stage3=dict(num_channels=(48, 96, 192)), 8 | stage4=dict(num_channels=(48, 96, 192, 384)))), 9 | decode_head=dict( 10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384]))) 11 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr48_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_40k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w48', 4 | backbone=dict( 5 | extra=dict( 6 | stage2=dict(num_channels=(48, 96)), 7 | stage3=dict(num_channels=(48, 96, 192)), 8 | stage4=dict(num_channels=(48, 96, 192, 384)))), 9 | decode_head=dict( 10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384]))) 11 | -------------------------------------------------------------------------------- /seaformer-seg/configs/hrnet/fcn_hr48_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_hr18_512x512_80k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w48', 4 | backbone=dict( 5 | extra=dict( 6 | stage2=dict(num_channels=(48, 96)), 7 | stage3=dict(num_channels=(48, 96, 192)), 8 | stage4=dict(num_channels=(48, 96, 192, 384)))), 9 | decode_head=dict( 10 | in_channels=[48, 96, 192, 384], channels=sum([48, 96, 192, 384]))) 11 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/isanet/isanet_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/isanet/isanet_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/mobilenet_v3/lraspp_m-v3-d8_512x1024_320k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/lraspp_m-v3-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | 6 | model = dict(pretrained='open-mmlab://contrib/mobilenet_v3_large') 7 | 8 | # Re-config the data sampler. 9 | data = dict(samples_per_gpu=4, workers_per_gpu=4) 10 | 11 | runner = dict(type='IterBasedRunner', max_iters=320000) 12 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/nonlocal_net/nonlocal_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/nonlocal_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/nonlocal_net/nonlocal_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/nonlocal_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x1024_160k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x1024_40k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x512_160k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x512_20k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x512_40k_voc12aug.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/ocrnet/ocrnet_hr18s_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ocrnet_hr18_512x512_80k_ade20k.py' 2 | model = dict( 3 | pretrained='open-mmlab://msra/hrnetv2_w18_small', 4 | backbone=dict( 5 | extra=dict( 6 | stage1=dict(num_blocks=(2, )), 7 | stage2=dict(num_blocks=(2, 2)), 8 | stage3=dict(num_modules=3, num_blocks=(2, 2, 2)), 9 | stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2))))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/psanet/psanet_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/psanet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/psanet/psanet_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/psanet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/pspnet/pspnet_r50-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/configs/pspnet/pspnet_r50-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_769x769.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | model = dict( 7 | decode_head=dict(align_corners=True), 8 | auxiliary_head=dict(align_corners=True), 9 | test_cfg=dict(mode='slide', crop_size=(769, 769), stride=(513, 513))) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/segformer/segformer_mit-b1_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./segformer_mit-b0_512x512_160k_ade20k.py'] 2 | 3 | # model settings 4 | model = dict( 5 | pretrained='pretrain/mit_b1.pth', 6 | backbone=dict( 7 | embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[2, 2, 2, 2]), 8 | decode_head=dict(in_channels=[64, 128, 320, 512])) 9 | -------------------------------------------------------------------------------- /seaformer-seg/configs/segformer/segformer_mit-b2_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./segformer_mit-b0_512x512_160k_ade20k.py'] 2 | 3 | # model settings 4 | model = dict( 5 | pretrained='pretrain/mit_b2.pth', 6 | backbone=dict( 7 | embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 4, 6, 3]), 8 | decode_head=dict(in_channels=[64, 128, 320, 512])) 9 | -------------------------------------------------------------------------------- /seaformer-seg/configs/segformer/segformer_mit-b3_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./segformer_mit-b0_512x512_160k_ade20k.py'] 2 | 3 | # model settings 4 | model = dict( 5 | pretrained='pretrain/mit_b3.pth', 6 | backbone=dict( 7 | embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 4, 18, 3]), 8 | decode_head=dict(in_channels=[64, 128, 320, 512])) 9 | -------------------------------------------------------------------------------- /seaformer-seg/configs/segformer/segformer_mit-b4_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./segformer_mit-b0_512x512_160k_ade20k.py'] 2 | 3 | # model settings 4 | model = dict( 5 | pretrained='pretrain/mit_b4.pth', 6 | backbone=dict( 7 | embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 8, 27, 3]), 8 | decode_head=dict(in_channels=[64, 128, 320, 512])) 9 | -------------------------------------------------------------------------------- /seaformer-seg/configs/segformer/segformer_mit-b5_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./segformer_mit-b0_512x512_160k_ade20k.py'] 2 | 3 | # model settings 4 | model = dict( 5 | pretrained='pretrain/mit_b5.pth', 6 | backbone=dict( 7 | embed_dims=64, num_heads=[1, 2, 5, 8], num_layers=[3, 6, 40, 3]), 8 | decode_head=dict(in_channels=[64, 128, 320, 512])) 9 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/swin/upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './upernet_swin_base_patch4_window12_512x512_160k_ade20k_' 3 | 'pretrain_384x384_1K.py' 4 | ] 5 | model = dict(pretrained='pretrain/swin_base_patch4_window12_384_22k.pth') 6 | -------------------------------------------------------------------------------- /seaformer-seg/configs/swin/upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './upernet_swin_base_patch4_window7_512x512_160k_ade20k_' 3 | 'pretrain_224x224_1K.py' 4 | ] 5 | model = dict(pretrained='pretrain/swin_base_patch4_window7_224_22k.pth') 6 | -------------------------------------------------------------------------------- /seaformer-seg/configs/swin/upernet_swin_small_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_' 3 | 'pretrain_224x224_1K.py' 4 | ] 5 | model = dict( 6 | pretrained='pretrain/swin_small_patch4_window7_224.pth', 7 | backbone=dict(depths=[2, 2, 18, 2]), 8 | decode_head=dict(in_channels=[96, 192, 384, 768], num_classes=150), 9 | auxiliary_head=dict(in_channels=384, num_classes=150)) 10 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/configs/vit/upernet_deit-s16_mln_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=dict(in_channels=[384, 384, 384, 384], out_channels=384), 8 | auxiliary_head=dict(num_classes=150, in_channels=384)) 9 | -------------------------------------------------------------------------------- /seaformer-seg/demo/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/seaformer-seg/demo/demo.png -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/imgs/speed_iou.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/seaformer-seg/imgs/speed_iou.jpg -------------------------------------------------------------------------------- /seaformer-seg/imgs/speed_iou.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/seaformer-seg/imgs/speed_iou.pdf -------------------------------------------------------------------------------- /seaformer-seg/local_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 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- 1 | # yapf:disable 2 | log_config = dict( 3 | interval=50, 4 | hooks=[ 5 | dict(type='TextLoggerHook', by_epoch=False), 6 | # dict(type='TensorboardLoggerHook') 7 | ]) 8 | # yapf:enable 9 | dist_params = dict(backend='nccl') 10 | log_level = 'INFO' 11 | load_from = None 12 | resume_from = None 13 | workflow = [('train', 1)] 14 | cudnn_benchmark = True 15 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=160000) 8 | checkpoint_config = dict(by_epoch=False, interval=16000) 9 | evaluation = dict(interval=16000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=20000) 8 | checkpoint_config = dict(by_epoch=False, interval=2000) 9 | evaluation = dict(interval=2000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=320000) 8 | checkpoint_config = dict(by_epoch=False, interval=32000) 9 | evaluation = dict(interval=32000, metric='mIoU') 10 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=40000) 8 | checkpoint_config = dict(by_epoch=False, interval=4000) 9 | evaluation = dict(interval=4000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- 1 | # optimizer 2 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0005) 3 | optimizer_config = dict() 4 | # learning policy 5 | lr_config = dict(policy='poly', power=0.9, min_lr=1e-4, by_epoch=False) 6 | # runtime settings 7 | runner = dict(type='IterBasedRunner', max_iters=80000) 8 | checkpoint_config = dict(by_epoch=False, interval=8000) 9 | evaluation = dict(interval=8000, metric='mIoU', pre_eval=True) 10 | -------------------------------------------------------------------------------- /seaformer-seg/local_configs/topformer/topformer_base_512x512_160k_4x8_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './topformer_base_512x512_160k_2x8_ade20k.py' 3 | ] 4 | 5 | data=dict(samples_per_gpu=4) 6 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/core/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .evaluation import * # noqa: F401, F403 3 | from .seg import * # noqa: F401, F403 4 | from .utils import * # noqa: F401, F403 5 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/core/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .misc import add_prefix 3 | 4 | __all__ = ['add_prefix'] 5 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/datasets/dark_zurich.py: -------------------------------------------------------------------------------- 1 | from .builder import DATASETS 2 | from .cityscapes import CityscapesDataset 3 | 4 | 5 | @DATASETS.register_module() 6 | class DarkZurichDataset(CityscapesDataset): 7 | """DarkZurichDataset dataset.""" 8 | 9 | def __init__(self, **kwargs): 10 | super().__init__( 11 | img_suffix='_rgb_anon.png', 12 | seg_map_suffix='_gt_labelTrainIds.png', 13 | **kwargs) 14 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/datasets/night_driving.py: -------------------------------------------------------------------------------- 1 | from .builder import DATASETS 2 | from .cityscapes import CityscapesDataset 3 | 4 | 5 | @DATASETS.register_module() 6 | class NightDrivingDataset(CityscapesDataset): 7 | """NightDrivingDataset dataset.""" 8 | 9 | def __init__(self, **kwargs): 10 | super().__init__( 11 | img_suffix='_leftImg8bit.png', 12 | seg_map_suffix='_gtCoarse_labelTrainIds.png', 13 | **kwargs) 14 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/models/necks/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .fpn import FPN 3 | from .ic_neck import ICNeck 4 | from .jpu import JPU 5 | from .mla_neck import MLANeck 6 | from .multilevel_neck import MultiLevelNeck 7 | 8 | __all__ = ['FPN', 'MultiLevelNeck', 'MLANeck', 'ICNeck', 'JPU'] 9 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/mmseg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .collect_env import collect_env 3 | from .logger import get_root_logger 4 | 5 | __all__ = ['get_root_logger', 'collect_env'] 6 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/optional.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /seaformer-seg/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 | -------------------------------------------------------------------------------- /seaformer-seg/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.1,<=1.4.0 2 | -------------------------------------------------------------------------------- /seaformer-seg/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | -------------------------------------------------------------------------------- /seaformer-seg/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | prettytable 3 | torch 4 | torchvision 5 | -------------------------------------------------------------------------------- /seaformer-seg/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | packaging 4 | prettytable 5 | -------------------------------------------------------------------------------- /seaformer-seg/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | codecov 2 | flake8 3 | interrogate 4 | isort==4.3.21 5 | pytest 6 | xdoctest>=0.10.0 7 | yapf 8 | -------------------------------------------------------------------------------- /seaformer-seg/tools/dist_test.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | CONFIG=$1 4 | CHECKPOINT=$2 5 | GPUS=$3 6 | PORT=${PORT:-29500} 7 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ 8 | python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \ 9 | $(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4} 10 | -------------------------------------------------------------------------------- /seaformer-seg/tools/dist_train.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | CONFIG=$1 4 | GPUS=$2 5 | PORT=${PORT:-29500} 6 | 7 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ 8 | python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \ 9 | $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3} 10 | -------------------------------------------------------------------------------- /seaformer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fudan-zvg/SeaFormer/9718d05b6e70d0561dda7c052b8457feed74d4d9/seaformer.png --------------------------------------------------------------------------------