├── .gitignore ├── LICENSE ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── coco_detection.py │ │ ├── coco_detection_direct_size448.py │ │ ├── coco_instance.py │ │ └── coco_instance_direct_size448.py │ ├── default_runtime.py │ ├── models │ │ ├── cascade_rcnn_r50_fpn.py │ │ ├── faster_rcnn_r50_fpn.py │ │ ├── mask_rcnn_r50_fpn.py │ │ ├── mask_rcnn_swin_fpn.py │ │ └── retinanet_r50_fpn.py │ └── schedules │ │ ├── schedule_1x.py │ │ ├── schedule_20e.py │ │ └── schedule_2x.py └── direct_pretraining │ ├── cascade_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py │ ├── cascade_rcnn_r50_fpn_direct_pretrain_1x_coco.py │ ├── faster_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py │ ├── faster_rcnn_r50_fpn_direct_pretrain_1x_coco.py │ ├── mask_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py │ ├── mask_rcnn_r50_fpn_direct_pretrain_1x_coco.py │ ├── mask_rcnn_swin_tiny_direct_finetune_p1x_1x_coco.py │ ├── mask_rcnn_swin_tiny_direct_pretrain_1x_coco.py │ ├── retinanet_r50_fpn_direct_finetune_p1x_1x_coco.py │ └── retinanet_r50_fpn_direct_pretrain_1x_coco.py ├── dist_test.sh ├── dist_train.sh ├── figs └── process.png ├── modules ├── __init__.py ├── mmcv_custom │ ├── __init__.py │ ├── checkpoint.py │ └── runner │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── epoch_based_runner.py │ │ └── optimizer.py ├── swin_transformer.py └── train.py ├── slurm.sh ├── test.py ├── train.py └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_detection_direct_size448.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/datasets/coco_detection_direct_size448.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_instance_direct_size448.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/datasets/coco_instance_direct_size448.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/mask_rcnn_swin_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/models/mask_rcnn_swin_fpn.py -------------------------------------------------------------------------------- /configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /configs/direct_pretraining/cascade_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/cascade_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/cascade_rcnn_r50_fpn_direct_pretrain_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/cascade_rcnn_r50_fpn_direct_pretrain_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/faster_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/faster_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/faster_rcnn_r50_fpn_direct_pretrain_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/faster_rcnn_r50_fpn_direct_pretrain_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/mask_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/mask_rcnn_r50_fpn_direct_finetune_p1x_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/mask_rcnn_r50_fpn_direct_pretrain_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/mask_rcnn_r50_fpn_direct_pretrain_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/mask_rcnn_swin_tiny_direct_finetune_p1x_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/mask_rcnn_swin_tiny_direct_finetune_p1x_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/mask_rcnn_swin_tiny_direct_pretrain_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/mask_rcnn_swin_tiny_direct_pretrain_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/retinanet_r50_fpn_direct_finetune_p1x_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/retinanet_r50_fpn_direct_finetune_p1x_1x_coco.py -------------------------------------------------------------------------------- /configs/direct_pretraining/retinanet_r50_fpn_direct_pretrain_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/configs/direct_pretraining/retinanet_r50_fpn_direct_pretrain_1x_coco.py -------------------------------------------------------------------------------- /dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/dist_test.sh -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/dist_train.sh -------------------------------------------------------------------------------- /figs/process.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/figs/process.png -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /modules/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /modules/mmcv_custom/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/runner/__init__.py -------------------------------------------------------------------------------- /modules/mmcv_custom/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/runner/checkpoint.py -------------------------------------------------------------------------------- /modules/mmcv_custom/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /modules/mmcv_custom/runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/mmcv_custom/runner/optimizer.py -------------------------------------------------------------------------------- /modules/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/swin_transformer.py -------------------------------------------------------------------------------- /modules/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/modules/train.py -------------------------------------------------------------------------------- /slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/slurm.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxzs5/direct-pretraining/HEAD/train.sh --------------------------------------------------------------------------------