├── LICENSE ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── ade20k.py │ │ ├── ade20k_640x640.py │ │ ├── ade20k_640x640_joint_train.py │ │ ├── ade20k_joint_train.py │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── cityscapes_1024x1024.py │ │ ├── cityscapes_768x768.py │ │ ├── cityscapes_769x769.py │ │ ├── cityscapes_832x832.py │ │ ├── coco-stuff10k.py │ │ ├── coco-stuff164k.py │ │ ├── coco-stuff164k_joint_train.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── isaid.py │ │ ├── loveda.py │ │ ├── mapillary_v1.py │ │ ├── mapillary_v1_65.py │ │ ├── mapillary_v2.py │ │ ├── pascal_context.py │ │ ├── pascal_context_59.py │ │ ├── pascal_voc12.py │ │ ├── pascal_voc12_aug.py │ │ ├── potsdam.py │ │ ├── refuge.py │ │ ├── stare.py │ │ ├── synapse.py │ │ └── vaihingen.py │ ├── default_runtime.py │ ├── models │ │ ├── ann_r50-d8.py │ │ ├── apcnet_r50-d8.py │ │ ├── bisenetv1_r18-d32.py │ │ ├── bisenetv2.py │ │ ├── ccnet_r50-d8.py │ │ ├── cgnet.py │ │ ├── danet_r50-d8.py │ │ ├── deeplabv3_r50-d8.py │ │ ├── deeplabv3_unet_s5-d16.py │ │ ├── deeplabv3plus_r50-d8.py │ │ ├── dmnet_r50-d8.py │ │ ├── dnl_r50-d8.py │ │ ├── dpt_vit-b16.py │ │ ├── emanet_r50-d8.py │ │ ├── encnet_r50-d8.py │ │ ├── erfnet_fcn.py │ │ ├── fast_scnn.py │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ ├── fcn_hr18.py │ │ ├── fcn_r50-d8.py │ │ ├── fcn_unet_s5-d16.py │ │ ├── fpn_poolformer_s12.py │ │ ├── fpn_r50.py │ │ ├── gcnet_r50-d8.py │ │ ├── icnet_r50-d8.py │ │ ├── isanet_r50-d8.py │ │ ├── lraspp_m-v3-d8.py │ │ ├── nonlocal_r50-d8.py │ │ ├── ocrnet_hr18.py │ │ ├── ocrnet_r50-d8.py │ │ ├── pointrend_r50.py │ │ ├── psanet_r50-d8.py │ │ ├── pspnet_r50-d8.py │ │ ├── pspnet_unet_s5-d16.py │ │ ├── segformer_mit-b0.py │ │ ├── segmenter_vit-b16_mask.py │ │ ├── setr_mla.py │ │ ├── setr_naive.py │ │ ├── setr_pup.py │ │ ├── stdc.py │ │ ├── twins_pcpvt-s_fpn.py │ │ ├── twins_pcpvt-s_upernet.py │ │ ├── upernet_beit.py │ │ ├── upernet_convnext.py │ │ ├── upernet_mae.py │ │ ├── upernet_r50.py │ │ ├── upernet_swin.py │ │ └── upernet_vit-b16_ln_mln.py │ └── schedules │ │ ├── schedule_160k.py │ │ ├── schedule_20k.py │ │ ├── schedule_240k.py │ │ ├── schedule_320k.py │ │ ├── schedule_40k.py │ │ └── schedule_80k.py ├── mask2former │ ├── mask2former_r50_8xb2-160k_ade20k-512x512-joint-train.py │ ├── mask2former_r50_8xb2-160k_cocostuff164k-512x512-joint-train.py │ ├── mask2former_swin-b-in1k-384x384-pre_8xb2-320k_ade20k-640x640-joint-train.py │ ├── mask2former_swin-s_8xb2-160k_cocostuff164k-512x512-joint-train.py │ ├── mask2former_swin-s_8xb2-320k_ade20k-512x512-joint-train.py │ ├── mask2former_swin-t_8xb2-160k_cocostuff164k-512x512-joint-train.py │ └── mask2former_swin-t_8xb2-320k_ade20k-512x512-joint-train.py ├── segformer │ ├── segformer_mit-b0_8xb2-160k_ade20k-512x512-joint-train.py │ ├── segformer_mit-b0_8xb2-80k_cocostuff164k-512x512-joint-train.py │ ├── segformer_mit-b2_8xb2-160k_cocostuff164k-512x512-joint-train.py │ ├── segformer_mit-b2_8xb2-320k_ade20k-512x512-joint-train.py │ ├── segformer_mit-b4_8xb2-160k_cocostuff164k-512x512-joint-train.py │ └── segformer_mit-b4_8xb2-320k_ade20k-512x512-joint-train.py └── segmenter │ ├── segmenter_vit-b_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py │ ├── segmenter_vit-b_mask_8xb1-320k_ade20k-512x512-joint-train.py │ ├── segmenter_vit-s_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py │ └── segmenter_vit-s_mask_8xb1-320k_ade20k-512x512-joint-train.py ├── dist_test.sh ├── dist_train.sh ├── docs ├── pipeline.png └── vis.png ├── preprocess ├── filter.py └── resample.py ├── test.py ├── train.py └── training-logs ├── ade20k_mask2former_swin-b_mIoU-53.7.log ├── ade20k_mask2former_swin-s_mIoU-53.3.log ├── ade20k_mask2former_swin-t_mIoU-52.0.log ├── ade20k_segformer_mit-b2_mIoU-47.9.log ├── ade20k_segformer_mit-b4_mIoU-50.6.log ├── ade20k_segmenter_vit-b_mIoU-51.1.log ├── ade20k_segmenter_vit-s_mIoU-47.9.log ├── coco_mask2former_swin-s_mIoU-47.6.log ├── coco_mask2former_swin-t_mIoU-46.4.log ├── coco_segformer_mit-b2_mIoU-44.2.log ├── coco_segformer_mit-b4_mIoU-46.6.log ├── coco_segmenter_vit-b_mIoU-47.5.log └── coco_segmenter_vit-s_mIoU-44.8.log /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k_640x640.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/ade20k_640x640.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k_640x640_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/ade20k_640x640_joint_train.py -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/ade20k_joint_train.py -------------------------------------------------------------------------------- /configs/_base_/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/chase_db1.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/cityscapes.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_1024x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/cityscapes_1024x1024.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_768x768.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/cityscapes_768x768.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_769x769.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/cityscapes_769x769.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_832x832.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/cityscapes_832x832.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/coco-stuff10k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/coco-stuff164k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco-stuff164k_joint_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/coco-stuff164k_joint_train.py -------------------------------------------------------------------------------- /configs/_base_/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/drive.py -------------------------------------------------------------------------------- /configs/_base_/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/hrf.py -------------------------------------------------------------------------------- /configs/_base_/datasets/isaid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/isaid.py -------------------------------------------------------------------------------- /configs/_base_/datasets/loveda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/loveda.py -------------------------------------------------------------------------------- /configs/_base_/datasets/mapillary_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/mapillary_v1.py -------------------------------------------------------------------------------- /configs/_base_/datasets/mapillary_v1_65.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/mapillary_v1_65.py -------------------------------------------------------------------------------- /configs/_base_/datasets/mapillary_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/mapillary_v2.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/pascal_context.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/pascal_context_59.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/pascal_voc12.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/pascal_voc12_aug.py -------------------------------------------------------------------------------- /configs/_base_/datasets/potsdam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/potsdam.py -------------------------------------------------------------------------------- /configs/_base_/datasets/refuge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/refuge.py -------------------------------------------------------------------------------- /configs/_base_/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/stare.py -------------------------------------------------------------------------------- /configs/_base_/datasets/synapse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/synapse.py -------------------------------------------------------------------------------- /configs/_base_/datasets/vaihingen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/datasets/vaihingen.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/ann_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/ann_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/apcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/apcnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/bisenetv1_r18-d32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/bisenetv1_r18-d32.py -------------------------------------------------------------------------------- /configs/_base_/models/bisenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/bisenetv2.py -------------------------------------------------------------------------------- /configs/_base_/models/ccnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/ccnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/cgnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/cgnet.py -------------------------------------------------------------------------------- /configs/_base_/models/danet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/danet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/deeplabv3_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/deeplabv3_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/deeplabv3plus_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/deeplabv3plus_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dmnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/dmnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dnl_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/dnl_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/dpt_vit-b16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/dpt_vit-b16.py -------------------------------------------------------------------------------- /configs/_base_/models/emanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/emanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/encnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/encnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/erfnet_fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/erfnet_fcn.py -------------------------------------------------------------------------------- /configs/_base_/models/fast_scnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fast_scnn.py -------------------------------------------------------------------------------- /configs/_base_/models/fastfcn_r50-d32_jpu_psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fastfcn_r50-d32_jpu_psp.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fcn_hr18.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fcn_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/fcn_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fcn_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/fpn_poolformer_s12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fpn_poolformer_s12.py -------------------------------------------------------------------------------- /configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/gcnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/gcnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/icnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/icnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/isanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/isanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/lraspp_m-v3-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/lraspp_m-v3-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/nonlocal_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/nonlocal_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/ocrnet_hr18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/ocrnet_hr18.py -------------------------------------------------------------------------------- /configs/_base_/models/ocrnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/ocrnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pointrend_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/pointrend_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/psanet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/psanet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pspnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/pspnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/pspnet_unet_s5-d16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/pspnet_unet_s5-d16.py -------------------------------------------------------------------------------- /configs/_base_/models/segformer_mit-b0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/segformer_mit-b0.py -------------------------------------------------------------------------------- /configs/_base_/models/segmenter_vit-b16_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/segmenter_vit-b16_mask.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_mla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/setr_mla.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/setr_naive.py -------------------------------------------------------------------------------- /configs/_base_/models/setr_pup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/setr_pup.py -------------------------------------------------------------------------------- /configs/_base_/models/stdc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/stdc.py -------------------------------------------------------------------------------- /configs/_base_/models/twins_pcpvt-s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/twins_pcpvt-s_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/twins_pcpvt-s_upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/twins_pcpvt-s_upernet.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_beit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_beit.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_convnext.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_mae.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_r50.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_swin.py -------------------------------------------------------------------------------- /configs/_base_/models/upernet_vit-b16_ln_mln.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/models/upernet_vit-b16_ln_mln.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_240k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_240k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_r50_8xb2-160k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_r50_8xb2-160k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_r50_8xb2-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_r50_8xb2-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_swin-b-in1k-384x384-pre_8xb2-320k_ade20k-640x640-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_swin-b-in1k-384x384-pre_8xb2-320k_ade20k-640x640-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_swin-s_8xb2-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_swin-s_8xb2-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_swin-s_8xb2-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_swin-s_8xb2-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_swin-t_8xb2-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_swin-t_8xb2-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/mask2former/mask2former_swin-t_8xb2-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/mask2former/mask2former_swin-t_8xb2-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b0_8xb2-160k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b0_8xb2-160k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b0_8xb2-80k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b0_8xb2-80k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b2_8xb2-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b2_8xb2-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b2_8xb2-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b2_8xb2-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b4_8xb2-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b4_8xb2-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segformer/segformer_mit-b4_8xb2-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segformer/segformer_mit-b4_8xb2-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segmenter/segmenter_vit-b_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segmenter/segmenter_vit-b_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segmenter/segmenter_vit-b_mask_8xb1-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segmenter/segmenter_vit-b_mask_8xb1-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segmenter/segmenter_vit-s_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segmenter/segmenter_vit-s_mask_8xb1-160k_cocostuff164k-512x512-joint-train.py -------------------------------------------------------------------------------- /configs/segmenter/segmenter_vit-s_mask_8xb1-320k_ade20k-512x512-joint-train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/configs/segmenter/segmenter_vit-s_mask_8xb1-320k_ade20k-512x512-joint-train.py -------------------------------------------------------------------------------- /dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/dist_test.sh -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/dist_train.sh -------------------------------------------------------------------------------- /docs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/docs/pipeline.png -------------------------------------------------------------------------------- /docs/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/docs/vis.png -------------------------------------------------------------------------------- /preprocess/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/preprocess/filter.py -------------------------------------------------------------------------------- /preprocess/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/preprocess/resample.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/train.py -------------------------------------------------------------------------------- /training-logs/ade20k_mask2former_swin-b_mIoU-53.7.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_mask2former_swin-b_mIoU-53.7.log -------------------------------------------------------------------------------- /training-logs/ade20k_mask2former_swin-s_mIoU-53.3.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_mask2former_swin-s_mIoU-53.3.log -------------------------------------------------------------------------------- /training-logs/ade20k_mask2former_swin-t_mIoU-52.0.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_mask2former_swin-t_mIoU-52.0.log -------------------------------------------------------------------------------- /training-logs/ade20k_segformer_mit-b2_mIoU-47.9.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_segformer_mit-b2_mIoU-47.9.log -------------------------------------------------------------------------------- /training-logs/ade20k_segformer_mit-b4_mIoU-50.6.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_segformer_mit-b4_mIoU-50.6.log -------------------------------------------------------------------------------- /training-logs/ade20k_segmenter_vit-b_mIoU-51.1.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_segmenter_vit-b_mIoU-51.1.log -------------------------------------------------------------------------------- /training-logs/ade20k_segmenter_vit-s_mIoU-47.9.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/ade20k_segmenter_vit-s_mIoU-47.9.log -------------------------------------------------------------------------------- /training-logs/coco_mask2former_swin-s_mIoU-47.6.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_mask2former_swin-s_mIoU-47.6.log -------------------------------------------------------------------------------- /training-logs/coco_mask2former_swin-t_mIoU-46.4.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_mask2former_swin-t_mIoU-46.4.log -------------------------------------------------------------------------------- /training-logs/coco_segformer_mit-b2_mIoU-44.2.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_segformer_mit-b2_mIoU-44.2.log -------------------------------------------------------------------------------- /training-logs/coco_segformer_mit-b4_mIoU-46.6.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_segformer_mit-b4_mIoU-46.6.log -------------------------------------------------------------------------------- /training-logs/coco_segmenter_vit-b_mIoU-47.5.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_segmenter_vit-b_mIoU-47.5.log -------------------------------------------------------------------------------- /training-logs/coco_segmenter_vit-s_mIoU-44.8.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiheYoung/FreeMask/HEAD/training-logs/coco_segmenter_vit-s_mIoU-44.8.log --------------------------------------------------------------------------------