├── .gitignore ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── assets ├── arch_large.png └── tokenmixer.png ├── classification ├── data │ ├── __init__.py │ ├── datasets.py │ ├── imagenet1k │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt │ ├── samplers.py │ └── threeaugment.py ├── engine.py ├── losses.py ├── main.py ├── model │ ├── __init__.py │ └── rcvit.py ├── optim_factory.py └── utils.py ├── detection ├── configs │ ├── RCViT │ │ ├── mask_rcnn_rcvit_m_fpn_1x_coco_bs4.py │ │ ├── mask_rcnn_rcvit_s_fpn_1x_coco_bs4.py │ │ ├── mask_rcnn_rcvit_t_fpn_1x_coco_bs4.py │ │ ├── mask_rcnn_rcvit_xs_fpn_1x_coco_bs4.py │ │ ├── retinanet_rcvit_m_fpn_1x_coco_bs4.py │ │ ├── retinanet_rcvit_s_fpn_1x_coco_bs4.py │ │ ├── retinanet_rcvit_t_fpn_1x_coco_bs4.py │ │ └── retinanet_rcvit_xs_fpn_1x_coco_bs4.py │ └── _base_ │ │ ├── datasets │ │ ├── cityscapes_detection.py │ │ ├── cityscapes_instance.py │ │ ├── coco_detection.py │ │ ├── coco_instance.py │ │ ├── coco_instance_semantic.py │ │ ├── deepfashion.py │ │ ├── lvis_v0.5_instance.py │ │ ├── lvis_v1_instance.py │ │ ├── voc0712.py │ │ └── wider_face.py │ │ ├── default_runtime.py │ │ ├── models │ │ ├── cascade_mask_rcnn_pvtv2_b2_fpn.py │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ ├── cascade_rcnn_r50_fpn.py │ │ ├── fast_rcnn_r50_fpn.py │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ ├── faster_rcnn_r50_caffe_dc5.py │ │ ├── faster_rcnn_r50_fpn.py │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ ├── mask_rcnn_r50_fpn.py │ │ ├── retinanet_r50_fpn.py │ │ ├── rpn_r50_caffe_c4.py │ │ ├── rpn_r50_fpn.py │ │ └── ssd300.py │ │ └── schedules │ │ ├── schedule_1x.py │ │ ├── schedule_20e.py │ │ └── schedule_2x.py ├── mmcv_custom │ ├── __init__.py │ ├── checkpoint.py │ └── runner │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── epoch_based_runner.py │ │ └── optimizer.py ├── mmdet_custom │ └── apis │ │ └── train.py ├── model │ ├── __init__.py │ └── rcvit.py ├── test.py └── train.py └── segmentation ├── align_resize.py ├── configs ├── RCViT │ ├── fpn_rcvit_m_512x512_40k_ade20k_bs4.py │ ├── fpn_rcvit_s_512x512_40k_ade20k_bs4.py │ ├── fpn_rcvit_t_512x512_40k_ade20k_bs4.py │ └── fpn_rcvit_xs_512x512_40k_ade20k_bs4.py └── _base_ │ ├── datasets │ ├── ade20k.py │ └── ade20k_upernet.py │ ├── default_runtime.py │ ├── models │ ├── fpn_r50.py │ └── upernet_r50.py │ └── schedules │ ├── schedule_160k.py │ ├── schedule_20k.py │ ├── schedule_40k.py │ └── schedule_80k.py ├── model ├── __init__.py └── rcvit.py └── tools ├── analyze_logs.py ├── benchmark.py ├── browse_dataset.py ├── convert_datasets ├── chase_db1.py ├── cityscapes.py ├── coco_stuff10k.py ├── coco_stuff164k.py ├── drive.py ├── hrf.py ├── pascal_context.py ├── stare.py └── voc_aug.py ├── deploy_test.py ├── dist_test.sh ├── dist_train.sh ├── example.py ├── get_flops.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 ├── torchserve ├── mmseg2torchserve.py ├── mmseg_handler.py └── test_torchserve.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/README.md -------------------------------------------------------------------------------- /assets/arch_large.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/assets/arch_large.png -------------------------------------------------------------------------------- /assets/tokenmixer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/assets/tokenmixer.png -------------------------------------------------------------------------------- /classification/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /classification/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/datasets.py -------------------------------------------------------------------------------- /classification/data/imagenet1k/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/imagenet1k/test.txt -------------------------------------------------------------------------------- /classification/data/imagenet1k/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/imagenet1k/train.txt -------------------------------------------------------------------------------- /classification/data/imagenet1k/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/imagenet1k/val.txt -------------------------------------------------------------------------------- /classification/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/samplers.py -------------------------------------------------------------------------------- /classification/data/threeaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/data/threeaugment.py -------------------------------------------------------------------------------- /classification/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/engine.py -------------------------------------------------------------------------------- /classification/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/losses.py -------------------------------------------------------------------------------- /classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/main.py -------------------------------------------------------------------------------- /classification/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .rcvit import * 2 | 3 | 4 | -------------------------------------------------------------------------------- /classification/model/rcvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/model/rcvit.py -------------------------------------------------------------------------------- /classification/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/optim_factory.py -------------------------------------------------------------------------------- /classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/classification/utils.py -------------------------------------------------------------------------------- /detection/configs/RCViT/mask_rcnn_rcvit_m_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/mask_rcnn_rcvit_m_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/mask_rcnn_rcvit_s_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/mask_rcnn_rcvit_s_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/mask_rcnn_rcvit_t_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/mask_rcnn_rcvit_t_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/mask_rcnn_rcvit_xs_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/mask_rcnn_rcvit_xs_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/retinanet_rcvit_m_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/retinanet_rcvit_m_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/retinanet_rcvit_s_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/retinanet_rcvit_s_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/retinanet_rcvit_t_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/retinanet_rcvit_t_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/RCViT/retinanet_rcvit_xs_fpn_1x_coco_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/RCViT/retinanet_rcvit_xs_fpn_1x_coco_bs4.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_mask_rcnn_pvtv2_b2_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/cascade_mask_rcnn_pvtv2_b2_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/fast_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/fast_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/rpn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/models/ssd300.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /detection/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/runner/__init__.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/runner/checkpoint.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmcv_custom/runner/optimizer.py -------------------------------------------------------------------------------- /detection/mmdet_custom/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/mmdet_custom/apis/train.py -------------------------------------------------------------------------------- /detection/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .rcvit import * -------------------------------------------------------------------------------- /detection/model/rcvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/model/rcvit.py -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/detection/train.py -------------------------------------------------------------------------------- /segmentation/align_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/align_resize.py -------------------------------------------------------------------------------- /segmentation/configs/RCViT/fpn_rcvit_m_512x512_40k_ade20k_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/RCViT/fpn_rcvit_m_512x512_40k_ade20k_bs4.py -------------------------------------------------------------------------------- /segmentation/configs/RCViT/fpn_rcvit_s_512x512_40k_ade20k_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/RCViT/fpn_rcvit_s_512x512_40k_ade20k_bs4.py -------------------------------------------------------------------------------- /segmentation/configs/RCViT/fpn_rcvit_t_512x512_40k_ade20k_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/RCViT/fpn_rcvit_t_512x512_40k_ade20k_bs4.py -------------------------------------------------------------------------------- /segmentation/configs/RCViT/fpn_rcvit_xs_512x512_40k_ade20k_bs4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/RCViT/fpn_rcvit_xs_512x512_40k_ade20k_bs4.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k_upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/datasets/ade20k_upernet.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/models/upernet_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /segmentation/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .rcvit import * -------------------------------------------------------------------------------- /segmentation/model/rcvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/model/rcvit.py -------------------------------------------------------------------------------- /segmentation/tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/analyze_logs.py -------------------------------------------------------------------------------- /segmentation/tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/benchmark.py -------------------------------------------------------------------------------- /segmentation/tools/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/browse_dataset.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/chase_db1.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/cityscapes.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/coco_stuff10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/coco_stuff10k.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/coco_stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/coco_stuff164k.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/drive.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/hrf.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/pascal_context.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/stare.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/voc_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/convert_datasets/voc_aug.py -------------------------------------------------------------------------------- /segmentation/tools/deploy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/deploy_test.py -------------------------------------------------------------------------------- /segmentation/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/dist_test.sh -------------------------------------------------------------------------------- /segmentation/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/dist_train.sh -------------------------------------------------------------------------------- /segmentation/tools/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/example.py -------------------------------------------------------------------------------- /segmentation/tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/get_flops.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/mit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/model_converters/mit2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/swin2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/model_converters/swin2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/vit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/model_converters/vit2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/onnx2tensorrt.py -------------------------------------------------------------------------------- /segmentation/tools/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/print_config.py -------------------------------------------------------------------------------- /segmentation/tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/publish_model.py -------------------------------------------------------------------------------- /segmentation/tools/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/pytorch2onnx.py -------------------------------------------------------------------------------- /segmentation/tools/pytorch2torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/pytorch2torchscript.py -------------------------------------------------------------------------------- /segmentation/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/slurm_test.sh -------------------------------------------------------------------------------- /segmentation/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/slurm_train.sh -------------------------------------------------------------------------------- /segmentation/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/test.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/mmseg2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/torchserve/mmseg2torchserve.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/mmseg_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/torchserve/mmseg_handler.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/test_torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/torchserve/test_torchserve.py -------------------------------------------------------------------------------- /segmentation/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tianfang-Zhang/CAS-ViT/HEAD/segmentation/tools/train.py --------------------------------------------------------------------------------