├── .gitignore ├── LICENSE ├── README.md ├── configs ├── cat_base.yaml ├── cat_small.yaml └── cat_tiny.yaml ├── detection ├── cat.py ├── configs │ ├── _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 │ │ │ └── mask_rcnn_cat_fpn.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ ├── atss_cat_base_fpn_1x_coco.py │ ├── atss_cat_small_fpn_1x_coco.py │ ├── cascade_rcnn_cat_base_fpn_1x_coco.py │ ├── cascade_rcnn_cat_base_fpn_1x_coco_ms.py │ ├── cascade_rcnn_cat_small_fpn_1x_coco.py │ ├── fcos_cat_base_1x_coco.py │ ├── fcos_cat_small_1x_coco.py │ ├── mask_rcnn_cat_base_1x_coco_ms.py │ ├── mask_rcnn_cat_small_1x_coco_ms.py │ ├── retinanet_cat_base_fpn_1x_coco.py │ └── retinanet_cat_small_fpn_1x_coco.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── test.py └── train.py ├── figures └── architecture.jpg ├── get_started.md ├── main.py ├── segmentation ├── cat.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── cityscapes_769x769.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── pascal_context.py │ │ │ ├── pascal_voc12.py │ │ │ ├── pascal_voc12_aug.py │ │ │ └── stare.py │ │ ├── default_runtime.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ ├── semantic_fpn_cat_base_512x512_160k_ade20k.py │ ├── semantic_fpn_cat_base_512x512_80k_ade20k.py │ ├── semantic_fpn_cat_small_512x512_160k_ade20k.py │ └── semantic_fpn_cat_small_512x512_80k_ade20k.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── test.py └── train.py └── src ├── __init__.py ├── cat.py ├── config.py ├── dataset.py ├── image_folder.py ├── logger.py ├── lr_scheduler.py ├── optimizer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/README.md -------------------------------------------------------------------------------- /configs/cat_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/configs/cat_base.yaml -------------------------------------------------------------------------------- /configs/cat_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/configs/cat_small.yaml -------------------------------------------------------------------------------- /configs/cat_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/configs/cat_tiny.yaml -------------------------------------------------------------------------------- /detection/cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/cat.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_cat_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/models/mask_rcnn_cat_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/configs/atss_cat_base_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/atss_cat_base_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/atss_cat_small_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/atss_cat_small_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/cascade_rcnn_cat_base_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/cascade_rcnn_cat_base_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/cascade_rcnn_cat_base_fpn_1x_coco_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/cascade_rcnn_cat_base_fpn_1x_coco_ms.py -------------------------------------------------------------------------------- /detection/configs/cascade_rcnn_cat_small_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/cascade_rcnn_cat_small_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/fcos_cat_base_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/fcos_cat_base_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/fcos_cat_small_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/fcos_cat_small_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_cat_base_1x_coco_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/mask_rcnn_cat_base_1x_coco_ms.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_cat_small_1x_coco_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/mask_rcnn_cat_small_1x_coco_ms.py -------------------------------------------------------------------------------- /detection/configs/retinanet_cat_base_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/retinanet_cat_base_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/retinanet_cat_small_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/configs/retinanet_cat_small_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/dist_test.sh -------------------------------------------------------------------------------- /detection/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/dist_train.sh -------------------------------------------------------------------------------- /detection/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/get_flops.py -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/detection/train.py -------------------------------------------------------------------------------- /figures/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/figures/architecture.jpg -------------------------------------------------------------------------------- /get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/get_started.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/main.py -------------------------------------------------------------------------------- /segmentation/cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/cat.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/chase_db1.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/cityscapes.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/cityscapes_769x769.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/cityscapes_769x769.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/drive.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/hrf.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/pascal_context.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/pascal_voc12.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/pascal_voc12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/pascal_voc12_aug.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/datasets/stare.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /segmentation/configs/semantic_fpn_cat_base_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/semantic_fpn_cat_base_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/semantic_fpn_cat_base_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/semantic_fpn_cat_base_512x512_80k_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/semantic_fpn_cat_small_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/semantic_fpn_cat_small_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/semantic_fpn_cat_small_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/configs/semantic_fpn_cat_small_512x512_80k_ade20k.py -------------------------------------------------------------------------------- /segmentation/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/dist_test.sh -------------------------------------------------------------------------------- /segmentation/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/dist_train.sh -------------------------------------------------------------------------------- /segmentation/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/get_flops.py -------------------------------------------------------------------------------- /segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/test.py -------------------------------------------------------------------------------- /segmentation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/segmentation/train.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/cat.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/config.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/image_folder.py -------------------------------------------------------------------------------- /src/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/logger.py -------------------------------------------------------------------------------- /src/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/lr_scheduler.py -------------------------------------------------------------------------------- /src/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/optimizer.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/linhezheng19/CAT/HEAD/src/utils.py --------------------------------------------------------------------------------