├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── ade20k.py │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── cityscapes_769x769.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── pascal_context.py │ │ ├── pascal_context_59.py │ │ ├── pascal_voc12.py │ │ ├── pascal_voc12_aug.py │ │ └── stare.py │ ├── default_runtime.py │ ├── models │ │ ├── condnet_r50-d8.py │ │ └── condnet_upernet-swin.py │ └── schedules │ │ ├── schedule_160k.py │ │ ├── schedule_20k.py │ │ ├── schedule_40k.py │ │ └── schedule_80k.py └── condnet │ ├── condnet_r101-d8_512x512_160k_ade20k.py │ ├── condnet_r101-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py │ ├── condnet_r50-d8_512x512_160k_ade20k.py │ ├── condnet_r50-d8_512x512_80k_ade20k.py │ ├── condnet_r50-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py │ └── condnet_upernet_swin_base_patch4_window12_512x512_160k_ade20k.py ├── models ├── __init__.py ├── builder.py └── decode_heads │ ├── __init__.py │ ├── cond_aspp_head.py │ ├── cond_sep_aspp_head.py │ ├── cond_uper_head.py │ └── conditional_decode_head.py ├── requirements.txt ├── requirements ├── docs.txt ├── mminstall.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── resources └── model.png └── tools ├── analyze_logs.py ├── benchmark.py ├── convert_datasets ├── chase_db1.py ├── cityscapes.py ├── drive.py ├── hrf.py ├── pascal_context.py ├── stare.py └── voc_aug.py ├── deploy_test.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── mmseg2torchserve.py ├── mmseg_handler.py ├── model_converters ├── mit2mmseg.py ├── swin2mmseg.py └── vit2mmseg.py ├── onnx2tensorrt.py ├── print_config.py ├── publish_model.py ├── pytorch2onnx.py ├── pytorch2torchscript.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /configs/_base_/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/chase_db1.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/cityscapes.py -------------------------------------------------------------------------------- /configs/_base_/datasets/cityscapes_769x769.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/cityscapes_769x769.py -------------------------------------------------------------------------------- /configs/_base_/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/drive.py -------------------------------------------------------------------------------- /configs/_base_/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/hrf.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/pascal_context.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/pascal_context_59.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/pascal_voc12.py -------------------------------------------------------------------------------- /configs/_base_/datasets/pascal_voc12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/pascal_voc12_aug.py -------------------------------------------------------------------------------- /configs/_base_/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/datasets/stare.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/condnet_r50-d8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/models/condnet_r50-d8.py -------------------------------------------------------------------------------- /configs/_base_/models/condnet_upernet-swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/models/condnet_upernet-swin.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /configs/condnet/condnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_r101-d8_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /configs/condnet/condnet_r101-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_r101-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py -------------------------------------------------------------------------------- /configs/condnet/condnet_r50-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_r50-d8_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /configs/condnet/condnet_r50-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_r50-d8_512x512_80k_ade20k.py -------------------------------------------------------------------------------- /configs/condnet/condnet_r50-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_r50-d8_headlrx10_lr1e-3_480x480_80k_pascal_context_59.py -------------------------------------------------------------------------------- /configs/condnet/condnet_upernet_swin_base_patch4_window12_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/configs/condnet/condnet_upernet_swin_base_patch4_window12_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/builder.py -------------------------------------------------------------------------------- /models/decode_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/decode_heads/__init__.py -------------------------------------------------------------------------------- /models/decode_heads/cond_aspp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/decode_heads/cond_aspp_head.py -------------------------------------------------------------------------------- /models/decode_heads/cond_sep_aspp_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/decode_heads/cond_sep_aspp_head.py -------------------------------------------------------------------------------- /models/decode_heads/cond_uper_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/decode_heads/cond_uper_head.py -------------------------------------------------------------------------------- /models/decode_heads/conditional_decode_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/models/decode_heads/conditional_decode_head.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.1,<=1.4.0 2 | -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /resources/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/resources/model.png -------------------------------------------------------------------------------- /tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/convert_datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/chase_db1.py -------------------------------------------------------------------------------- /tools/convert_datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/cityscapes.py -------------------------------------------------------------------------------- /tools/convert_datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/drive.py -------------------------------------------------------------------------------- /tools/convert_datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/hrf.py -------------------------------------------------------------------------------- /tools/convert_datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/pascal_context.py -------------------------------------------------------------------------------- /tools/convert_datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/stare.py -------------------------------------------------------------------------------- /tools/convert_datasets/voc_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/convert_datasets/voc_aug.py -------------------------------------------------------------------------------- /tools/deploy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/deploy_test.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/mmseg2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/mmseg2torchserve.py -------------------------------------------------------------------------------- /tools/mmseg_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/mmseg_handler.py -------------------------------------------------------------------------------- /tools/model_converters/mit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/model_converters/mit2mmseg.py -------------------------------------------------------------------------------- /tools/model_converters/swin2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/model_converters/swin2mmseg.py -------------------------------------------------------------------------------- /tools/model_converters/vit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/model_converters/vit2mmseg.py -------------------------------------------------------------------------------- /tools/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/onnx2tensorrt.py -------------------------------------------------------------------------------- /tools/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/print_config.py -------------------------------------------------------------------------------- /tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/publish_model.py -------------------------------------------------------------------------------- /tools/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/pytorch2onnx.py -------------------------------------------------------------------------------- /tools/pytorch2torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/pytorch2torchscript.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yu-changqian/CondNet/HEAD/tools/train.py --------------------------------------------------------------------------------