gitextract_wj4p2u9n/ ├── .gitignore ├── .gitignore.swp ├── README.md ├── dall_e/ │ ├── __init__.py │ ├── decoder.py │ ├── encoder.py │ └── utils.py ├── downstream_tasks/ │ ├── detection/ │ │ ├── README.md │ │ ├── evaluation/ │ │ │ └── object_detection/ │ │ │ ├── configs/ │ │ │ │ ├── _base_/ │ │ │ │ │ ├── datasets/ │ │ │ │ │ │ └── coco_instance.py │ │ │ │ │ ├── default_runtime.py │ │ │ │ │ ├── models/ │ │ │ │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ │ │ │ ├── cascade_mask_rcnn_swin_fpn.py │ │ │ │ │ │ ├── cascade_mask_rcnn_vit_fpn.py │ │ │ │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ │ │ │ └── mask_rcnn_vit_fpn.py │ │ │ │ │ └── schedules/ │ │ │ │ │ └── schedule_1x.py │ │ │ │ └── mask_rcnn/ │ │ │ │ ├── vit_base_giou_4conv1f_coco_maskrcnn_1x_cae_sincos_init0.1_lr00003.py │ │ │ │ └── vit_large_giou_4conv1f_coco_maskrcnn_1x_cae_sincos_init0.1_lr00002_lrdr0.85_dp0.2.py │ │ │ ├── mmcv_custom/ │ │ │ │ ├── __init__.py │ │ │ │ ├── checkpoint.py │ │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ │ ├── prepare_rpe.py │ │ │ │ ├── register_backbone.py │ │ │ │ └── runner/ │ │ │ │ ├── __init__.py │ │ │ │ ├── checkpoint.py │ │ │ │ └── epoch_based_runner.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── loader.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── head.py │ │ │ ├── swin_transformer.py │ │ │ └── vision_transformer.py │ │ ├── scripts/ │ │ │ ├── run_eval.sh │ │ │ ├── run_train_maskrcnn_vit_base.sh │ │ │ └── run_train_maskrcnn_vit_large.sh │ │ └── utils.py │ └── semantic_segmentation/ │ ├── README.md │ ├── backbone/ │ │ ├── beit.py │ │ ├── beit_fapn.py │ │ ├── cae.py │ │ ├── fapn.py │ │ └── mae.py │ ├── configs_local/ │ │ ├── _base_/ │ │ │ ├── datasets/ │ │ │ │ ├── ade20k.py │ │ │ │ ├── ade20k_640x640.py │ │ │ │ ├── chase_db1.py │ │ │ │ ├── cityscapes.py │ │ │ │ ├── cityscapes_769x769.py │ │ │ │ ├── coco-stuff10k.py │ │ │ │ ├── drive.py │ │ │ │ ├── hrf.py │ │ │ │ ├── pascal_context.py │ │ │ │ ├── pascal_voc12.py │ │ │ │ ├── pascal_voc12_aug.py │ │ │ │ └── stare.py │ │ │ ├── default_runtime.py │ │ │ ├── models/ │ │ │ │ ├── ann_r50-d8.py │ │ │ │ ├── apcnet_r50-d8.py │ │ │ │ ├── ccnet_r50-d8.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── danet_r50-d8.py │ │ │ │ ├── deeplabv3_r50-d8.py │ │ │ │ ├── deeplabv3_unet_s5-d16.py │ │ │ │ ├── deeplabv3plus_r50-d8.py │ │ │ │ ├── dmnet_r50-d8.py │ │ │ │ ├── dnl_r50-d8.py │ │ │ │ ├── emanet_r50-d8.py │ │ │ │ ├── encnet_r50-d8.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── fcn_hr18.py │ │ │ │ ├── fcn_r50-d8.py │ │ │ │ ├── fcn_unet_s5-d16.py │ │ │ │ ├── fpn_r50.py │ │ │ │ ├── gcnet_r50-d8.py │ │ │ │ ├── lraspp_m-v3-d8.py │ │ │ │ ├── nonlocal_r50-d8.py │ │ │ │ ├── ocrnet_hr18.py │ │ │ │ ├── ocrnet_r50-d8.py │ │ │ │ ├── pointrend_r50.py │ │ │ │ ├── psanet_r50-d8.py │ │ │ │ ├── pspnet_r50-d8.py │ │ │ │ ├── pspnet_unet_s5-d16.py │ │ │ │ ├── upernet_cae.py │ │ │ │ └── upernet_r50.py │ │ │ └── schedules/ │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ │ ├── beit/ │ │ │ └── upernet_beit_base_12_512_slide_160k_ade20k_pt_4e-4.py │ │ ├── cae/ │ │ │ └── upernet/ │ │ │ ├── upernet_cae_base_12_512_slide_160k_ade20k_pt_1e-4.py │ │ │ ├── upernet_cae_base_12_512_slide_160k_ade20k_pt_2e-4.py │ │ │ ├── upernet_cae_base_12_512_slide_160k_ade20k_pt_3e-4.py │ │ │ └── upernet_cae_large_24_512_slide_160k_ade20k_pt_decay095_4e-5_dp015.py │ │ └── mae/ │ │ └── upernet_mae_large_12_512_slide_160k_ade20k_pt_4e-4.py │ ├── mmcv_custom/ │ │ ├── __init__.py │ │ ├── apex_runner/ │ │ │ ├── __init__.py │ │ │ ├── apex_iter_based_runner.py │ │ │ ├── checkpoint.py │ │ │ └── optimizer.py │ │ ├── checkpoint.py │ │ ├── checkpoint_beit.py │ │ ├── layer_decay_optimizer_constructor.py │ │ ├── resize_transform.py │ │ └── train_api.py │ ├── mmseg/ │ │ ├── __init__.py │ │ ├── apis/ │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── core/ │ │ │ ├── __init__.py │ │ │ ├── evaluation/ │ │ │ │ ├── __init__.py │ │ │ │ ├── class_names.py │ │ │ │ ├── eval_hooks.py │ │ │ │ └── metrics.py │ │ │ ├── seg/ │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── sampler/ │ │ │ │ ├── __init__.py │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ └── ohem_pixel_sampler.py │ │ │ └── utils/ │ │ │ ├── __init__.py │ │ │ └── misc.py │ │ ├── datasets/ │ │ │ ├── __init__.py │ │ │ ├── ade.py │ │ │ ├── builder.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── coco_stuff.py │ │ │ ├── custom.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── pascal_context.py │ │ │ ├── pipelines/ │ │ │ │ ├── __init__.py │ │ │ │ ├── compose.py │ │ │ │ ├── formating.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms.py │ │ │ ├── stare.py │ │ │ └── voc.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── backbones/ │ │ │ │ ├── __init__.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ └── unet.py │ │ │ ├── builder.py │ │ │ ├── decode_heads/ │ │ │ │ ├── __init__.py │ │ │ │ ├── ann_head.py │ │ │ │ ├── apc_head.py │ │ │ │ ├── aspp_head.py │ │ │ │ ├── cascade_decode_head.py │ │ │ │ ├── cc_head.py │ │ │ │ ├── da_head.py │ │ │ │ ├── decode_head.py │ │ │ │ ├── dm_head.py │ │ │ │ ├── dnl_head.py │ │ │ │ ├── ema_head.py │ │ │ │ ├── enc_head.py │ │ │ │ ├── fcn_head.py │ │ │ │ ├── fpn_head.py │ │ │ │ ├── gc_head.py │ │ │ │ ├── lraspp_head.py │ │ │ │ ├── nl_head.py │ │ │ │ ├── ocr_head.py │ │ │ │ ├── point_head.py │ │ │ │ ├── psa_head.py │ │ │ │ ├── psp_head.py │ │ │ │ ├── sep_aspp_head.py │ │ │ │ ├── sep_fcn_head.py │ │ │ │ └── uper_head.py │ │ │ ├── losses/ │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── lovasz_loss.py │ │ │ │ └── utils.py │ │ │ ├── necks/ │ │ │ │ ├── __init__.py │ │ │ │ └── fpn.py │ │ │ ├── segmentors/ │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── cascade_encoder_decoder.py │ │ │ │ └── encoder_decoder.py │ │ │ └── utils/ │ │ │ ├── __init__.py │ │ │ ├── inverted_residual.py │ │ │ ├── make_divisible.py │ │ │ ├── res_layer.py │ │ │ ├── se_layer.py │ │ │ ├── self_attention_block.py │ │ │ └── up_conv_block.py │ │ ├── ops/ │ │ │ ├── __init__.py │ │ │ ├── encoding.py │ │ │ └── wrappers.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ └── logger.py │ │ └── version.py │ └── tools/ │ ├── dist_test.sh │ ├── dist_train.sh │ ├── test.py │ └── train.py ├── furnace/ │ ├── dataset_folder.py │ ├── datasets.py │ ├── engine_for_finetuning.py │ ├── engine_for_pretraining.py │ ├── masking_generator.py │ ├── optim_factory.py │ ├── transforms.py │ └── utils.py ├── linear_util/ │ ├── crop.py │ ├── datasets.py │ ├── engine_finetune.py │ ├── lars.py │ ├── lr_decay.py │ ├── lr_sched.py │ ├── misc.py │ └── pos_embed.py ├── models/ │ ├── modeling_cae.py │ ├── modeling_cae_helper.py │ ├── modeling_discrete_vae.py │ └── modeling_finetune.py ├── requirements.txt ├── scripts/ │ ├── cae_base_800e.sh │ ├── cae_base_finetune.sh │ ├── cae_large_1600e.sh │ └── cae_large_finetune.sh ├── tokenizer-weights/ │ └── README └── tools/ ├── run_attentive.py ├── run_class_finetuning.py ├── run_linear.py └── run_pretraining.py