├── .gitignore ├── .pre-commit-config.yaml ├── CITATION.cff ├── LICENSE ├── README.md ├── configs ├── det │ ├── _base_ │ │ ├── datasets │ │ │ ├── coco_instance.py │ │ │ └── coco_panoptic.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── knet_s3_r50_fpn.py │ │ │ └── knet_s3_r50_fpn_panoptic.py │ │ └── schedules │ │ │ └── schedule_1x.py │ ├── common │ │ ├── mstrain_3x_coco_instance.py │ │ └── mstrain_3x_coco_panoptic.py │ └── knet │ │ ├── knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco-panoptic.py │ │ ├── knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco.py │ │ ├── knet_s3_r101_fpn_ms-3x_coco-panoptic.py │ │ ├── knet_s3_r101_fpn_ms-3x_coco.py │ │ ├── knet_s3_r50_fpn_1x_coco-panoptic.py │ │ ├── knet_s3_r50_fpn_1x_coco.py │ │ ├── knet_s3_r50_fpn_ms-3x_coco-panoptic.py │ │ ├── knet_s3_r50_fpn_ms-3x_coco.py │ │ └── knet_s3_swin-l_fpn_ms-3x_16x2_coco-panoptic.py └── seg │ ├── _base_ │ ├── custom_import.py │ ├── datasets │ │ └── ade20k.py │ ├── default_runtime.py │ ├── models │ │ ├── deeplabv3_r50-d8.py │ │ ├── fcn_r50-d8.py │ │ ├── pspnet_r50-d8.py │ │ └── upernet_r50.py │ └── schedules │ │ ├── schedule_160k.py │ │ └── schedule_adamw_80k.py │ ├── baselines │ ├── deeplabv3_r50-d8_512x512_80k_8x2_adamw_ade20k.py │ ├── fcn_r50-d8_512x512_80k_8x2_adamw_ade20k.py │ ├── pspnet_r50-d8_512x512_80k_8x2_adamw_ade20k.py │ └── upernet_r50_512x512_80k_8x2_adamw_ade20k.py │ └── knet │ ├── knet_s3_deeplabv3_r50-d8_80k_adamw_ade20k.py │ ├── knet_s3_fcn_r50-d8_80k_adamw_ade20k.py │ ├── knet_s3_pspnet_r50-d8_80k_adamw_ade20k.py │ ├── knet_s3_upernet_r50-d8_80k_adamw_ade20k.py │ ├── knet_s3_upernet_swin-l_80k_adamw_640x640_ade20k.py │ ├── knet_s3_upernet_swin-l_80k_adamw_ade20k.py │ └── knet_s3_upernet_swin-t_80k_adamw_ade20k.py ├── knet ├── det │ ├── kernel_head.py │ ├── kernel_iter_head.py │ ├── kernel_update_head.py │ ├── knet.py │ ├── mask_hungarian_assigner.py │ ├── mask_pseudo_sampler.py │ ├── semantic_fpn_wrapper.py │ └── utils.py ├── kernel_updator.py └── seg │ ├── iter_decode_head.py │ ├── kernel_update_head.py │ └── seg_kernel_head.py ├── setup.cfg └── tools ├── gen_panoptic_test_info.py ├── mim_slurm_test.sh └── mim_slurm_train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/README.md -------------------------------------------------------------------------------- /configs/det/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/det/_base_/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_s3_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/models/knet_s3_r50_fpn.py -------------------------------------------------------------------------------- /configs/det/_base_/models/knet_s3_r50_fpn_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/models/knet_s3_r50_fpn_panoptic.py -------------------------------------------------------------------------------- /configs/det/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_3x_coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/common/mstrain_3x_coco_instance.py -------------------------------------------------------------------------------- /configs/det/common/mstrain_3x_coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/common/mstrain_3x_coco_panoptic.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco-panoptic.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r101_dcn-c3-c5_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r101_fpn_ms-3x_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r101_fpn_ms-3x_coco-panoptic.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r101_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r101_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r50_fpn_1x_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r50_fpn_1x_coco-panoptic.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r50_fpn_1x_coco.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r50_fpn_ms-3x_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r50_fpn_ms-3x_coco-panoptic.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_r50_fpn_ms-3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_r50_fpn_ms-3x_coco.py -------------------------------------------------------------------------------- /configs/det/knet/knet_s3_swin-l_fpn_ms-3x_16x2_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/det/knet/knet_s3_swin-l_fpn_ms-3x_16x2_coco-panoptic.py -------------------------------------------------------------------------------- /configs/seg/_base_/custom_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/custom_import.py -------------------------------------------------------------------------------- /configs/seg/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /configs/seg/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/seg/_base_/models/deeplabv3_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/models/deeplabv3_r50-d8.py -------------------------------------------------------------------------------- /configs/seg/_base_/models/fcn_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/models/fcn_r50-d8.py -------------------------------------------------------------------------------- /configs/seg/_base_/models/pspnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/models/pspnet_r50-d8.py -------------------------------------------------------------------------------- /configs/seg/_base_/models/upernet_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/models/upernet_r50.py -------------------------------------------------------------------------------- /configs/seg/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /configs/seg/_base_/schedules/schedule_adamw_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/_base_/schedules/schedule_adamw_80k.py -------------------------------------------------------------------------------- /configs/seg/baselines/deeplabv3_r50-d8_512x512_80k_8x2_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/baselines/deeplabv3_r50-d8_512x512_80k_8x2_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/baselines/fcn_r50-d8_512x512_80k_8x2_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/baselines/fcn_r50-d8_512x512_80k_8x2_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/baselines/pspnet_r50-d8_512x512_80k_8x2_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/baselines/pspnet_r50-d8_512x512_80k_8x2_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/baselines/upernet_r50_512x512_80k_8x2_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/baselines/upernet_r50_512x512_80k_8x2_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_deeplabv3_r50-d8_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_deeplabv3_r50-d8_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_fcn_r50-d8_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_fcn_r50-d8_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_pspnet_r50-d8_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_pspnet_r50-d8_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_upernet_r50-d8_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_upernet_r50-d8_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_upernet_swin-l_80k_adamw_640x640_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_upernet_swin-l_80k_adamw_640x640_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_upernet_swin-l_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_upernet_swin-l_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /configs/seg/knet/knet_s3_upernet_swin-t_80k_adamw_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/configs/seg/knet/knet_s3_upernet_swin-t_80k_adamw_ade20k.py -------------------------------------------------------------------------------- /knet/det/kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/kernel_head.py -------------------------------------------------------------------------------- /knet/det/kernel_iter_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/kernel_iter_head.py -------------------------------------------------------------------------------- /knet/det/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/kernel_update_head.py -------------------------------------------------------------------------------- /knet/det/knet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/knet.py -------------------------------------------------------------------------------- /knet/det/mask_hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/mask_hungarian_assigner.py -------------------------------------------------------------------------------- /knet/det/mask_pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/mask_pseudo_sampler.py -------------------------------------------------------------------------------- /knet/det/semantic_fpn_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/semantic_fpn_wrapper.py -------------------------------------------------------------------------------- /knet/det/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/det/utils.py -------------------------------------------------------------------------------- /knet/kernel_updator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/kernel_updator.py -------------------------------------------------------------------------------- /knet/seg/iter_decode_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/seg/iter_decode_head.py -------------------------------------------------------------------------------- /knet/seg/kernel_update_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/seg/kernel_update_head.py -------------------------------------------------------------------------------- /knet/seg/seg_kernel_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/knet/seg/seg_kernel_head.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/setup.cfg -------------------------------------------------------------------------------- /tools/gen_panoptic_test_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/tools/gen_panoptic_test_info.py -------------------------------------------------------------------------------- /tools/mim_slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/tools/mim_slurm_test.sh -------------------------------------------------------------------------------- /tools/mim_slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZwwWayne/K-Net/HEAD/tools/mim_slurm_train.sh --------------------------------------------------------------------------------