├── .gitignore ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── ade20k.py │ │ ├── ade20k151.py │ │ ├── ade20k_640x640.py │ │ ├── ade20kmetric.py │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── cityscapes20.py │ │ ├── cityscapes20_1024x1024.py │ │ ├── cityscapes_1024x1024.py │ │ ├── cityscapes_2048x1024.py │ │ ├── cityscapes_768x768.py │ │ ├── cityscapes_769x769.py │ │ ├── cityscapes_832x832.py │ │ ├── cityscapesmetric.py │ │ ├── cityscapesmetric_1024x1024.py │ │ ├── coco-stuff10k.py │ │ ├── coco-stuff10k172.py │ │ ├── coco-stuff164k.py │ │ ├── coco-stuff164k172.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── imagenets.py │ │ ├── isaid.py │ │ ├── loveda.py │ │ ├── occlude_face.py │ │ ├── pascal_context.py │ │ ├── pascal_context_59.py │ │ ├── pascal_voc12.py │ │ ├── pascal_voc12_aug.py │ │ ├── potsdam.py │ │ ├── stare.py │ │ └── vaihingen.py │ ├── default_runtime.py │ ├── models │ │ ├── ann_r50-d8.py │ │ ├── apcnet_r50-d8.py │ │ ├── bisenetv1_r18-d32.py │ │ ├── bisenetv2.py │ │ ├── ccnet_r50-d8.py │ │ ├── cgnet.py │ │ ├── danet_r50-d8.py │ │ ├── deeplabv3_r50-d8.py │ │ ├── deeplabv3_unet_s5-d16.py │ │ ├── deeplabv3plus_m-v2-d8.py │ │ ├── deeplabv3plus_r50-d8.py │ │ ├── dmnet_r50-d8.py │ │ ├── dnl_r50-d8.py │ │ ├── dpt_vit-b16.py │ │ ├── emanet_r50-d8.py │ │ ├── encnet_r50-d8.py │ │ ├── erfnet_fcn.py │ │ ├── fast_scnn.py │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ ├── fcn_hr18.py │ │ ├── fcn_r50-d8.py │ │ ├── fcn_unet_s5-d16.py │ │ ├── fpn_poolformer_s12.py │ │ ├── fpn_r50.py │ │ ├── gcnet_r50-d8.py │ │ ├── icnet_r50-d8.py │ │ ├── isanet_r50-d8.py │ │ ├── lraspp_m-v3-d8.py │ │ ├── mask2former_beit.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 │ │ ├── segformer_mit-b2_segformer_head_unet_fc.py │ │ ├── segmenter_vit-b16_mask.py │ │ ├── setr_mla.py │ │ ├── setr_naive.py │ │ ├── setr_pup.py │ │ ├── stdc.py │ │ ├── twins_pcpvt-s_fpn.py │ │ ├── twins_pcpvt-s_upernet.py │ │ ├── upernet_beit.py │ │ ├── upernet_beit_adapter.py │ │ ├── upernet_convnext.py │ │ ├── upernet_mae.py │ │ ├── upernet_r50.py │ │ ├── upernet_swin.py │ │ └── upernet_vit-b16_ln_mln.py │ └── schedules │ │ ├── schedule_160k.py │ │ ├── schedule_20k.py │ │ ├── schedule_320k.py │ │ ├── schedule_400k.py │ │ ├── schedule_40k.py │ │ └── schedule_80k.py └── ade20k │ ├── segformer_b2_ade20k_multistep.py │ ├── segformer_b2_ade20k_multistep_inference.py │ └── segformer_b2_ade20k_singlestep.py ├── environment.yaml ├── mmseg ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── builder.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── class_names.py │ │ ├── eval_hooks.py │ │ └── metrics.py │ ├── hook │ │ ├── __init__.py │ │ └── wandblogger_hook.py │ ├── optimizers │ │ ├── __init__.py │ │ └── layer_decay_optimizer_constructor.py │ ├── seg │ │ ├── __init__.py │ │ ├── builder.py │ │ └── sampler │ │ │ ├── __init__.py │ │ │ ├── base_pixel_sampler.py │ │ │ └── ohem_pixel_sampler.py │ └── utils │ │ ├── __init__.py │ │ ├── dist_util.py │ │ └── misc.py ├── datasets │ ├── __init__.py │ ├── ade.py │ ├── builder.py │ ├── chase_db1.py │ ├── cityscapes.py │ ├── coco_stuff.py │ ├── custom.py │ ├── dark_zurich.py │ ├── dataset_wrappers.py │ ├── drive.py │ ├── face.py │ ├── hrf.py │ ├── imagenets.py │ ├── isaid.py │ ├── isprs.py │ ├── loveda.py │ ├── night_driving.py │ ├── pascal_context.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── compose.py │ │ ├── formating.py │ │ ├── formatting.py │ │ ├── loading.py │ │ ├── test_time_aug.py │ │ └── transforms.py │ ├── potsdam.py │ ├── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py │ ├── stare.py │ └── voc.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── beit.py │ │ ├── bisenetv1.py │ │ ├── bisenetv2.py │ │ ├── cgnet.py │ │ ├── erfnet.py │ │ ├── fast_scnn.py │ │ ├── hrnet.py │ │ ├── icnet.py │ │ ├── mae.py │ │ ├── mit.py │ │ ├── mobilenet_v2.py │ │ ├── mobilenet_v3.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ ├── resnext.py │ │ ├── stdc.py │ │ ├── swin.py │ │ ├── timm_backbone.py │ │ ├── twins.py │ │ ├── unet.py │ │ └── vit.py │ ├── builder.py │ ├── decode_heads │ │ ├── __init__.py │ │ ├── ann_head.py │ │ ├── apc_head.py │ │ ├── aspp_head.py │ │ ├── cascade_decode_head.py │ │ ├── cc_head.py │ │ ├── da_head.py │ │ ├── decode_head.py │ │ ├── dm_head.py │ │ ├── dnl_head.py │ │ ├── dpt_head.py │ │ ├── ema_head.py │ │ ├── enc_head.py │ │ ├── fcn_head.py │ │ ├── fpn_head.py │ │ ├── gc_head.py │ │ ├── isa_head.py │ │ ├── knet_head.py │ │ ├── lraspp_head.py │ │ ├── nl_head.py │ │ ├── ocr_head.py │ │ ├── point_head.py │ │ ├── psa_head.py │ │ ├── psp_head.py │ │ ├── segformer_head.py │ │ ├── segmenter_mask_head.py │ │ ├── sep_aspp_head.py │ │ ├── sep_fcn_head.py │ │ ├── setr_mla_head.py │ │ ├── setr_up_head.py │ │ ├── stdc_head.py │ │ └── uper_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── cross_entropy_loss.py │ │ ├── dice_loss.py │ │ ├── focal_loss.py │ │ ├── lovasz_loss.py │ │ ├── tversky_loss.py │ │ └── utils.py │ ├── necks │ │ ├── __init__.py │ │ ├── featurepyramid.py │ │ ├── fpn.py │ │ ├── ic_neck.py │ │ ├── jpu.py │ │ ├── mla_neck.py │ │ └── multilevel_neck.py │ ├── segmentors │ │ ├── __init__.py │ │ ├── base.py │ │ ├── cascade_encoder_decoder.py │ │ └── encoder_decoder.py │ └── utils │ │ ├── __init__.py │ │ ├── embed.py │ │ ├── inverted_residual.py │ │ ├── make_divisible.py │ │ ├── res_layer.py │ │ ├── se_layer.py │ │ ├── self_attention_block.py │ │ ├── shape_convert.py │ │ └── up_conv_block.py ├── ops │ ├── __init__.py │ ├── encoding.py │ └── wrappers.py ├── utils │ ├── __init__.py │ ├── collect_env.py │ ├── logger.py │ ├── misc.py │ ├── set_env.py │ └── util_distribution.py └── version.py ├── mmseg_custom ├── __init__.py ├── apis │ ├── __init__.py │ ├── test_multi_steps.py │ └── train_multi_steps.py ├── core │ ├── __init__.py │ ├── anchor │ │ ├── __init__.py │ │ ├── builder.py │ │ └── point_generator.py │ ├── box │ │ ├── __init__.py │ │ ├── builder.py │ │ └── samplers │ │ │ ├── __init__.py │ │ │ ├── base_sampler.py │ │ │ ├── mask_pseudo_sampler.py │ │ │ ├── mask_sampling_result.py │ │ │ └── sampling_result.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks_multi_steps.py │ │ └── metrics.py │ ├── hook │ │ ├── __init__.py │ │ └── ema.py │ ├── mask │ │ ├── __init__.py │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── dist_utils.py │ │ └── misc.py ├── datasets │ ├── __init__.py │ ├── ade20k_151.py │ ├── ade20k_metric.py │ ├── cityscapes_20.py │ ├── cityscapes_metric.py │ ├── coco_stuff_172.py │ └── pipelines │ │ ├── __init__.py │ │ ├── formatting.py │ │ └── transform.py └── models │ ├── __init__.py │ ├── backbones │ ├── __init__.py │ ├── mit_custom_init_weights.py │ ├── mobilenet_v2_custom_init_weights.py │ └── resnet_custom_init_weights.py │ ├── builder.py │ ├── decode_heads │ ├── __init__.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── misc.py │ │ └── schedule.py │ ├── segformer_head_unet_fc_head_multi_step.py │ ├── segformer_head_unet_fc_head_single_step.py │ └── unet │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── norm.py │ │ ├── pos_emb.py │ │ ├── unet.py │ │ └── unet_time_embed.py │ ├── losses │ ├── __init__.py │ ├── cross_entropy_loss.py │ ├── dice_loss.py │ ├── focal_loss.py │ ├── match_costs.py │ └── match_loss.py │ ├── plugins │ ├── __init__.py │ ├── msdeformattn_pixel_decoder.py │ └── pixel_decoder.py │ ├── segmentors │ ├── __init__.py │ ├── encoder_decoder_diffusion.py │ ├── encoder_decoder_diffusion_ensemble.py │ ├── encoder_decoder_ensemble.py │ └── encoder_decoder_freeze.py │ └── utils │ ├── __init__.py │ ├── assigner.py │ ├── point_sample.py │ ├── positional_encoding.py │ └── transformer.py └── tools ├── benchmark.py ├── convert_ema_model.py ├── diffusion_gt_infer.py ├── dist_test.sh ├── dist_test_diffusion.sh ├── dist_test_diffusion_origin.sh ├── dist_train.sh ├── dist_train_diffusion.sh ├── get_flops.py ├── get_params.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py ├── test_diffusion.py ├── test_diffusion_origin.py ├── train.py └── train_diffusion.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k151.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/ade20k151.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k_640x640.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/ade20k_640x640.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20kmetric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/ade20kmetric.py -------------------------------------------------------------------------------- /configs/_base_/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/chase_db1.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes20.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes20_1024x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes20_1024x1024.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_1024x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes_1024x1024.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_2048x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes_2048x1024.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_768x768.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes_768x768.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_769x769.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes_769x769.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_832x832.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapes_832x832.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapesmetric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapesmetric.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapesmetric_1024x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/cityscapesmetric_1024x1024.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/coco-stuff10k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff10k172.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/coco-stuff10k172.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/coco-stuff164k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff164k172.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/coco-stuff164k172.py -------------------------------------------------------------------------------- /configs/_base_/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/drive.py -------------------------------------------------------------------------------- /configs/_base_/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/hrf.py -------------------------------------------------------------------------------- /configs/_base_/datasets/imagenets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/imagenets.py -------------------------------------------------------------------------------- /configs/_base_/datasets/isaid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/isaid.py -------------------------------------------------------------------------------- /configs/_base_/datasets/loveda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/loveda.py -------------------------------------------------------------------------------- /configs/_base_/datasets/occlude_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/occlude_face.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/pascal_context.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/pascal_context_59.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/pascal_voc12.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/pascal_voc12_aug.py -------------------------------------------------------------------------------- /configs/_base_/datasets/potsdam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/potsdam.py -------------------------------------------------------------------------------- /configs/_base_/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/stare.py -------------------------------------------------------------------------------- /configs/_base_/datasets/vaihingen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/datasets/vaihingen.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/ann_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/ann_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/apcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/apcnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/bisenetv1_r18-d32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/bisenetv1_r18-d32.py -------------------------------------------------------------------------------- /configs/_base_/models/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/bisenetv2.py -------------------------------------------------------------------------------- /configs/_base_/models/ccnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/ccnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/cgnet.py -------------------------------------------------------------------------------- /configs/_base_/models/danet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/danet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/deeplabv3_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/deeplabv3_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3plus_m-v2-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/deeplabv3plus_m-v2-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3plus_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/deeplabv3plus_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dmnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/dmnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dnl_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/dnl_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dpt_vit-b16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/dpt_vit-b16.py -------------------------------------------------------------------------------- /configs/_base_/models/emanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/emanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/encnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/encnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/erfnet_fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/erfnet_fcn.py -------------------------------------------------------------------------------- /configs/_base_/models/fast_scnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fast_scnn.py -------------------------------------------------------------------------------- /configs/_base_/models/fastfcn_r50-d32_jpu_psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fcn_hr18.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fcn_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fcn_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/fpn_poolformer_s12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fpn_poolformer_s12.py -------------------------------------------------------------------------------- /configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/gcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/gcnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/icnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/icnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/isanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/isanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/lraspp_m-v3-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/lraspp_m-v3-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/mask2former_beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/mask2former_beit.py -------------------------------------------------------------------------------- /configs/_base_/models/nonlocal_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/nonlocal_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/ocrnet_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/ocrnet_hr18.py -------------------------------------------------------------------------------- /configs/_base_/models/ocrnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/ocrnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pointrend_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/pointrend_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/psanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/psanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pspnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/pspnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pspnet_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/pspnet_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/segformer_mit-b0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/segformer_mit-b0.py -------------------------------------------------------------------------------- /configs/_base_/models/segformer_mit-b2_segformer_head_unet_fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/segformer_mit-b2_segformer_head_unet_fc.py -------------------------------------------------------------------------------- /configs/_base_/models/segmenter_vit-b16_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/segmenter_vit-b16_mask.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_mla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/setr_mla.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/setr_naive.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_pup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/setr_pup.py -------------------------------------------------------------------------------- /configs/_base_/models/stdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/stdc.py -------------------------------------------------------------------------------- /configs/_base_/models/twins_pcpvt-s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/twins_pcpvt-s_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/twins_pcpvt-s_upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/twins_pcpvt-s_upernet.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_beit.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_beit_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_beit_adapter.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_convnext.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_mae.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_swin.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_vit-b16_ln_mln.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/models/upernet_vit-b16_ln_mln.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_400k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_400k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /configs/ade20k/segformer_b2_ade20k_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/ade20k/segformer_b2_ade20k_multistep.py -------------------------------------------------------------------------------- /configs/ade20k/segformer_b2_ade20k_multistep_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/ade20k/segformer_b2_ade20k_multistep_inference.py -------------------------------------------------------------------------------- /configs/ade20k/segformer_b2_ade20k_singlestep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/configs/ade20k/segformer_b2_ade20k_singlestep.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/environment.yaml -------------------------------------------------------------------------------- /mmseg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/__init__.py -------------------------------------------------------------------------------- /mmseg/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/apis/__init__.py -------------------------------------------------------------------------------- /mmseg/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/apis/inference.py -------------------------------------------------------------------------------- /mmseg/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/apis/test.py -------------------------------------------------------------------------------- /mmseg/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/apis/train.py -------------------------------------------------------------------------------- /mmseg/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/__init__.py -------------------------------------------------------------------------------- /mmseg/core/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/builder.py -------------------------------------------------------------------------------- /mmseg/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmseg/core/evaluation/class_names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/evaluation/class_names.py -------------------------------------------------------------------------------- /mmseg/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mmseg/core/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/evaluation/metrics.py -------------------------------------------------------------------------------- /mmseg/core/hook/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/hook/__init__.py -------------------------------------------------------------------------------- /mmseg/core/hook/wandblogger_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/hook/wandblogger_hook.py -------------------------------------------------------------------------------- /mmseg/core/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/optimizers/__init__.py -------------------------------------------------------------------------------- /mmseg/core/optimizers/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/optimizers/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /mmseg/core/seg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/seg/__init__.py -------------------------------------------------------------------------------- /mmseg/core/seg/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/seg/builder.py -------------------------------------------------------------------------------- /mmseg/core/seg/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/seg/sampler/__init__.py -------------------------------------------------------------------------------- /mmseg/core/seg/sampler/base_pixel_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/seg/sampler/base_pixel_sampler.py -------------------------------------------------------------------------------- /mmseg/core/seg/sampler/ohem_pixel_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/seg/sampler/ohem_pixel_sampler.py -------------------------------------------------------------------------------- /mmseg/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/utils/__init__.py -------------------------------------------------------------------------------- /mmseg/core/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/utils/dist_util.py -------------------------------------------------------------------------------- /mmseg/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/core/utils/misc.py -------------------------------------------------------------------------------- /mmseg/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/__init__.py -------------------------------------------------------------------------------- /mmseg/datasets/ade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/ade.py -------------------------------------------------------------------------------- /mmseg/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/builder.py -------------------------------------------------------------------------------- /mmseg/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/chase_db1.py -------------------------------------------------------------------------------- /mmseg/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/cityscapes.py -------------------------------------------------------------------------------- /mmseg/datasets/coco_stuff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/coco_stuff.py -------------------------------------------------------------------------------- /mmseg/datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/custom.py -------------------------------------------------------------------------------- /mmseg/datasets/dark_zurich.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/dark_zurich.py -------------------------------------------------------------------------------- /mmseg/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmseg/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/drive.py -------------------------------------------------------------------------------- /mmseg/datasets/face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/face.py -------------------------------------------------------------------------------- /mmseg/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/hrf.py -------------------------------------------------------------------------------- /mmseg/datasets/imagenets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/imagenets.py -------------------------------------------------------------------------------- /mmseg/datasets/isaid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/isaid.py -------------------------------------------------------------------------------- /mmseg/datasets/isprs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/isprs.py -------------------------------------------------------------------------------- /mmseg/datasets/loveda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/loveda.py -------------------------------------------------------------------------------- /mmseg/datasets/night_driving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/night_driving.py -------------------------------------------------------------------------------- /mmseg/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pascal_context.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/test_time_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/test_time_aug.py -------------------------------------------------------------------------------- /mmseg/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /mmseg/datasets/potsdam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/potsdam.py -------------------------------------------------------------------------------- /mmseg/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /mmseg/datasets/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /mmseg/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/stare.py -------------------------------------------------------------------------------- /mmseg/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/datasets/voc.py -------------------------------------------------------------------------------- /mmseg/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/__init__.py -------------------------------------------------------------------------------- /mmseg/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmseg/models/backbones/beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/beit.py -------------------------------------------------------------------------------- /mmseg/models/backbones/bisenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/bisenetv1.py -------------------------------------------------------------------------------- /mmseg/models/backbones/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/bisenetv2.py -------------------------------------------------------------------------------- /mmseg/models/backbones/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/cgnet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/erfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/erfnet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/fast_scnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/fast_scnn.py -------------------------------------------------------------------------------- /mmseg/models/backbones/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/hrnet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/icnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/icnet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/mae.py -------------------------------------------------------------------------------- /mmseg/models/backbones/mit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/mit.py -------------------------------------------------------------------------------- /mmseg/models/backbones/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/mobilenet_v2.py -------------------------------------------------------------------------------- /mmseg/models/backbones/mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/mobilenet_v3.py -------------------------------------------------------------------------------- /mmseg/models/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/resnest.py -------------------------------------------------------------------------------- /mmseg/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/resnet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/resnext.py -------------------------------------------------------------------------------- /mmseg/models/backbones/stdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/stdc.py -------------------------------------------------------------------------------- /mmseg/models/backbones/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/swin.py -------------------------------------------------------------------------------- /mmseg/models/backbones/timm_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/timm_backbone.py -------------------------------------------------------------------------------- /mmseg/models/backbones/twins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/twins.py -------------------------------------------------------------------------------- /mmseg/models/backbones/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/unet.py -------------------------------------------------------------------------------- /mmseg/models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/backbones/vit.py -------------------------------------------------------------------------------- /mmseg/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/builder.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/__init__.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/ann_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/ann_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/apc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/apc_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/aspp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/aspp_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/cascade_decode_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/cascade_decode_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/cc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/cc_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/da_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/da_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/decode_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/decode_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/dm_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/dm_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/dnl_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/dnl_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/dpt_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/dpt_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/ema_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/ema_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/enc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/enc_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/fcn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/fcn_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/fpn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/fpn_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/gc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/gc_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/isa_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/isa_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/knet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/knet_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/lraspp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/lraspp_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/nl_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/nl_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/ocr_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/ocr_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/point_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/point_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/psa_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/psa_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/psp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/psp_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/segformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/segformer_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/segmenter_mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/segmenter_mask_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/sep_aspp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/sep_aspp_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/sep_fcn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/sep_fcn_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/setr_mla_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/setr_mla_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/setr_up_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/setr_up_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/stdc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/stdc_head.py -------------------------------------------------------------------------------- /mmseg/models/decode_heads/uper_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/decode_heads/uper_head.py -------------------------------------------------------------------------------- /mmseg/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/__init__.py -------------------------------------------------------------------------------- /mmseg/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/accuracy.py -------------------------------------------------------------------------------- /mmseg/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /mmseg/models/losses/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/dice_loss.py -------------------------------------------------------------------------------- /mmseg/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/focal_loss.py -------------------------------------------------------------------------------- /mmseg/models/losses/lovasz_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/lovasz_loss.py -------------------------------------------------------------------------------- /mmseg/models/losses/tversky_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/tversky_loss.py -------------------------------------------------------------------------------- /mmseg/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/losses/utils.py -------------------------------------------------------------------------------- /mmseg/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/__init__.py -------------------------------------------------------------------------------- /mmseg/models/necks/featurepyramid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/featurepyramid.py -------------------------------------------------------------------------------- /mmseg/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/fpn.py -------------------------------------------------------------------------------- /mmseg/models/necks/ic_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/ic_neck.py -------------------------------------------------------------------------------- /mmseg/models/necks/jpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/jpu.py -------------------------------------------------------------------------------- /mmseg/models/necks/mla_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/mla_neck.py -------------------------------------------------------------------------------- /mmseg/models/necks/multilevel_neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/necks/multilevel_neck.py -------------------------------------------------------------------------------- /mmseg/models/segmentors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/segmentors/__init__.py -------------------------------------------------------------------------------- /mmseg/models/segmentors/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/segmentors/base.py -------------------------------------------------------------------------------- /mmseg/models/segmentors/cascade_encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/segmentors/cascade_encoder_decoder.py -------------------------------------------------------------------------------- /mmseg/models/segmentors/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/segmentors/encoder_decoder.py -------------------------------------------------------------------------------- /mmseg/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/__init__.py -------------------------------------------------------------------------------- /mmseg/models/utils/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/embed.py -------------------------------------------------------------------------------- /mmseg/models/utils/inverted_residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/inverted_residual.py -------------------------------------------------------------------------------- /mmseg/models/utils/make_divisible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/make_divisible.py -------------------------------------------------------------------------------- /mmseg/models/utils/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/res_layer.py -------------------------------------------------------------------------------- /mmseg/models/utils/se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/se_layer.py -------------------------------------------------------------------------------- /mmseg/models/utils/self_attention_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/self_attention_block.py -------------------------------------------------------------------------------- /mmseg/models/utils/shape_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/shape_convert.py -------------------------------------------------------------------------------- /mmseg/models/utils/up_conv_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/models/utils/up_conv_block.py -------------------------------------------------------------------------------- /mmseg/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/ops/__init__.py -------------------------------------------------------------------------------- /mmseg/ops/encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/ops/encoding.py -------------------------------------------------------------------------------- /mmseg/ops/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/ops/wrappers.py -------------------------------------------------------------------------------- /mmseg/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/__init__.py -------------------------------------------------------------------------------- /mmseg/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/collect_env.py -------------------------------------------------------------------------------- /mmseg/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/logger.py -------------------------------------------------------------------------------- /mmseg/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/misc.py -------------------------------------------------------------------------------- /mmseg/utils/set_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/set_env.py -------------------------------------------------------------------------------- /mmseg/utils/util_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/utils/util_distribution.py -------------------------------------------------------------------------------- /mmseg/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg/version.py -------------------------------------------------------------------------------- /mmseg_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/apis/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/apis/test_multi_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/apis/test_multi_steps.py -------------------------------------------------------------------------------- /mmseg_custom/apis/train_multi_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/apis/train_multi_steps.py -------------------------------------------------------------------------------- /mmseg_custom/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/anchor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/anchor/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/anchor/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/anchor/builder.py -------------------------------------------------------------------------------- /mmseg_custom/core/anchor/point_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/anchor/point_generator.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/builder.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/samplers/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/samplers/base_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/samplers/base_sampler.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/samplers/mask_pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/samplers/mask_pseudo_sampler.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/samplers/mask_sampling_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/samplers/mask_sampling_result.py -------------------------------------------------------------------------------- /mmseg_custom/core/box/samplers/sampling_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/box/samplers/sampling_result.py -------------------------------------------------------------------------------- /mmseg_custom/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/evaluation/eval_hooks_multi_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/evaluation/eval_hooks_multi_steps.py -------------------------------------------------------------------------------- /mmseg_custom/core/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/evaluation/metrics.py -------------------------------------------------------------------------------- /mmseg_custom/core/hook/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/hook/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/hook/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/hook/ema.py -------------------------------------------------------------------------------- /mmseg_custom/core/mask/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/mask/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/mask/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/mask/utils.py -------------------------------------------------------------------------------- /mmseg_custom/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/utils/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/core/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/utils/dist_utils.py -------------------------------------------------------------------------------- /mmseg_custom/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/core/utils/misc.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/ade20k_151.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/ade20k_151.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/ade20k_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/ade20k_metric.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/cityscapes_20.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/cityscapes_20.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/cityscapes_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/cityscapes_metric.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/coco_stuff_172.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/coco_stuff_172.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /mmseg_custom/datasets/pipelines/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/datasets/pipelines/transform.py -------------------------------------------------------------------------------- /mmseg_custom/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/backbones/mit_custom_init_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/backbones/mit_custom_init_weights.py -------------------------------------------------------------------------------- /mmseg_custom/models/backbones/mobilenet_v2_custom_init_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/backbones/mobilenet_v2_custom_init_weights.py -------------------------------------------------------------------------------- /mmseg_custom/models/backbones/resnet_custom_init_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/backbones/resnet_custom_init_weights.py -------------------------------------------------------------------------------- /mmseg_custom/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/builder.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/diffusion/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/diffusion/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/diffusion/misc.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/diffusion/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/diffusion/schedule.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/segformer_head_unet_fc_head_multi_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/segformer_head_unet_fc_head_multi_step.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/segformer_head_unet_fc_head_single_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/segformer_head_unet_fc_head_single_step.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/attention.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/norm.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/pos_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/pos_emb.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/unet.py -------------------------------------------------------------------------------- /mmseg_custom/models/decode_heads/unet/unet_time_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/decode_heads/unet/unet_time_embed.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/dice_loss.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/focal_loss.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/match_costs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/match_costs.py -------------------------------------------------------------------------------- /mmseg_custom/models/losses/match_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/losses/match_loss.py -------------------------------------------------------------------------------- /mmseg_custom/models/plugins/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/plugins/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/plugins/msdeformattn_pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/plugins/msdeformattn_pixel_decoder.py -------------------------------------------------------------------------------- /mmseg_custom/models/plugins/pixel_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/plugins/pixel_decoder.py -------------------------------------------------------------------------------- /mmseg_custom/models/segmentors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/segmentors/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/segmentors/encoder_decoder_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/segmentors/encoder_decoder_diffusion.py -------------------------------------------------------------------------------- /mmseg_custom/models/segmentors/encoder_decoder_diffusion_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/segmentors/encoder_decoder_diffusion_ensemble.py -------------------------------------------------------------------------------- /mmseg_custom/models/segmentors/encoder_decoder_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/segmentors/encoder_decoder_ensemble.py -------------------------------------------------------------------------------- /mmseg_custom/models/segmentors/encoder_decoder_freeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/segmentors/encoder_decoder_freeze.py -------------------------------------------------------------------------------- /mmseg_custom/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/utils/__init__.py -------------------------------------------------------------------------------- /mmseg_custom/models/utils/assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/utils/assigner.py -------------------------------------------------------------------------------- /mmseg_custom/models/utils/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/utils/point_sample.py -------------------------------------------------------------------------------- /mmseg_custom/models/utils/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/utils/positional_encoding.py -------------------------------------------------------------------------------- /mmseg_custom/models/utils/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/mmseg_custom/models/utils/transformer.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/convert_ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/convert_ema_model.py -------------------------------------------------------------------------------- /tools/diffusion_gt_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/diffusion_gt_infer.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_test_diffusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/dist_test_diffusion.sh -------------------------------------------------------------------------------- /tools/dist_test_diffusion_origin.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/dist_test_diffusion_origin.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/dist_train_diffusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/dist_train_diffusion.sh -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/get_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/get_params.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/test_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/test_diffusion.py -------------------------------------------------------------------------------- /tools/test_diffusion_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/test_diffusion_origin.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/DDPS/HEAD/tools/train_diffusion.py --------------------------------------------------------------------------------