├── .gitignore ├── LICENSE ├── README.md ├── config.py ├── configs ├── crossformer │ ├── base_patch4_group7_224.yaml │ ├── large_patch4_group7_224.yaml │ ├── small_patch4_group7_224.yaml │ └── tiny_patch4_group7_224.yaml └── crossformer_pp │ ├── base_patch4_group_const_224.yaml │ ├── huge_patch4_group_const_224.yaml │ ├── large_patch4_group_const_224.yaml │ └── small_patch4_group_const_224.yaml ├── data ├── __init__.py ├── build.py ├── cached_image_folder.py ├── samplers.py ├── tfrecord_torch_loader.py ├── yt_example_pb2.py ├── yt_feature_pb2.py └── zipreader.py ├── detection ├── README.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── coco_detection.py │ │ │ └── coco_instance.py │ │ ├── default_runtime.py │ │ └── models │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ └── retinanet_r50_fpn.py │ ├── crossformer │ │ ├── mask_rcnn_crossformer_b_fpn_1x_coco.py │ │ ├── mask_rcnn_crossformer_s_fpn_1x_coco.py │ │ ├── retinanet_crossformer_b_fpn_1x_coco.py │ │ └── retinanet_crossformer_s_fpn_1x_coco.py │ └── crossformer_pp │ │ ├── mask_rcnn_crossformerpp_s_fpn_1x_coco.py │ │ └── retinanet_crossformerpp_s_fpn_1x_coco.py ├── crossformer_factory.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── preprocess.py ├── test.py └── train.py ├── figures ├── github_pic.png └── github_pic_2.png ├── logger.py ├── lr_scheduler.py ├── main.py ├── models ├── __init__.py ├── build.py ├── crossformer.py ├── crossformer_backbone_det.py └── crossformer_backbone_seg.py ├── optimizer.py ├── segmentation ├── README.md ├── align_resize.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ └── ade20k_swin.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── fpn_r50.py │ │ │ └── upernet_crossformer.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ ├── crossformer │ │ ├── fpn_crossformer_b_ade20k_40k.py │ │ ├── fpn_crossformer_l_ade20k_40k.py │ │ ├── fpn_crossformer_s_ade20k_40k.py │ │ ├── upernet_crossformer_b_ade20k.py │ │ ├── upernet_crossformer_l_ade20k.py │ │ └── upernet_crossformer_s_ade20k.py │ └── crossformer_pp │ │ ├── fpn_crossformerpp_s_ade20k_40k.py │ │ └── upernet_crossformerpp_s_ade20k.py ├── crossformer_factory.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── preprocess.py ├── test.py └── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/config.py -------------------------------------------------------------------------------- /configs/crossformer/base_patch4_group7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer/base_patch4_group7_224.yaml -------------------------------------------------------------------------------- /configs/crossformer/large_patch4_group7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer/large_patch4_group7_224.yaml -------------------------------------------------------------------------------- /configs/crossformer/small_patch4_group7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer/small_patch4_group7_224.yaml -------------------------------------------------------------------------------- /configs/crossformer/tiny_patch4_group7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer/tiny_patch4_group7_224.yaml -------------------------------------------------------------------------------- /configs/crossformer_pp/base_patch4_group_const_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer_pp/base_patch4_group_const_224.yaml -------------------------------------------------------------------------------- /configs/crossformer_pp/huge_patch4_group_const_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer_pp/huge_patch4_group_const_224.yaml -------------------------------------------------------------------------------- /configs/crossformer_pp/large_patch4_group_const_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer_pp/large_patch4_group_const_224.yaml -------------------------------------------------------------------------------- /configs/crossformer_pp/small_patch4_group_const_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/configs/crossformer_pp/small_patch4_group_const_224.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/build.py -------------------------------------------------------------------------------- /data/cached_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/cached_image_folder.py -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/samplers.py -------------------------------------------------------------------------------- /data/tfrecord_torch_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/tfrecord_torch_loader.py -------------------------------------------------------------------------------- /data/yt_example_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/yt_example_pb2.py -------------------------------------------------------------------------------- /data/yt_feature_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/yt_feature_pb2.py -------------------------------------------------------------------------------- /data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/data/zipreader.py -------------------------------------------------------------------------------- /detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/README.md -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/crossformer/mask_rcnn_crossformer_b_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer/mask_rcnn_crossformer_b_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/crossformer/mask_rcnn_crossformer_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer/mask_rcnn_crossformer_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/crossformer/retinanet_crossformer_b_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer/retinanet_crossformer_b_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/crossformer/retinanet_crossformer_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer/retinanet_crossformer_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/crossformer_pp/mask_rcnn_crossformerpp_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer_pp/mask_rcnn_crossformerpp_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/crossformer_pp/retinanet_crossformerpp_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/configs/crossformer_pp/retinanet_crossformerpp_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/crossformer_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/crossformer_factory.py -------------------------------------------------------------------------------- /detection/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/dist_test.sh -------------------------------------------------------------------------------- /detection/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/dist_train.sh -------------------------------------------------------------------------------- /detection/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/get_flops.py -------------------------------------------------------------------------------- /detection/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/preprocess.py -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/detection/train.py -------------------------------------------------------------------------------- /figures/github_pic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/figures/github_pic.png -------------------------------------------------------------------------------- /figures/github_pic_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/figures/github_pic_2.png -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/logger.py -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/models/build.py -------------------------------------------------------------------------------- /models/crossformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/models/crossformer.py -------------------------------------------------------------------------------- /models/crossformer_backbone_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/models/crossformer_backbone_det.py -------------------------------------------------------------------------------- /models/crossformer_backbone_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/models/crossformer_backbone_seg.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/optimizer.py -------------------------------------------------------------------------------- /segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/README.md -------------------------------------------------------------------------------- /segmentation/align_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/align_resize.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/datasets/ade20k_swin.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/upernet_crossformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/models/upernet_crossformer.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/fpn_crossformer_b_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/fpn_crossformer_b_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/fpn_crossformer_l_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/fpn_crossformer_l_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/fpn_crossformer_s_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/fpn_crossformer_s_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/upernet_crossformer_b_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/upernet_crossformer_b_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/upernet_crossformer_l_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/upernet_crossformer_l_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer/upernet_crossformer_s_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer/upernet_crossformer_s_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer_pp/fpn_crossformerpp_s_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer_pp/fpn_crossformerpp_s_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/crossformer_pp/upernet_crossformerpp_s_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/configs/crossformer_pp/upernet_crossformerpp_s_ade20k.py -------------------------------------------------------------------------------- /segmentation/crossformer_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/crossformer_factory.py -------------------------------------------------------------------------------- /segmentation/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/dist_test.sh -------------------------------------------------------------------------------- /segmentation/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/dist_train.sh -------------------------------------------------------------------------------- /segmentation/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/get_flops.py -------------------------------------------------------------------------------- /segmentation/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/preprocess.py -------------------------------------------------------------------------------- /segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/test.py -------------------------------------------------------------------------------- /segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/segmentation/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cheerss/CrossFormer/HEAD/utils.py --------------------------------------------------------------------------------