├── .gitignore ├── .isort.cfg ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── figs └── grouprcnn.png ├── projects ├── api │ └── test.py ├── configs │ ├── 10_coco │ │ ├── 01_student_fcos.py │ │ ├── base.py │ │ ├── group_rcnn_24e_10_percent_coco_detr_augmentation.py │ │ └── group_rcnn_50e_10_percent_coco_detr_augmentation.py │ └── _base_ │ │ ├── datasets │ │ └── coco_detection.py │ │ ├── default_runtime.py │ │ └── schedules │ │ └── schedule_1x.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── point_coco.py │ └── transform.py └── models │ ├── __init__.py │ ├── group_rcnn.py │ └── group_roi_head.py └── tools ├── dist_test.sh ├── dist_train.sh ├── generate_anns.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/README.md -------------------------------------------------------------------------------- /figs/grouprcnn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/figs/grouprcnn.png -------------------------------------------------------------------------------- /projects/api/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/api/test.py -------------------------------------------------------------------------------- /projects/configs/10_coco/01_student_fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/10_coco/01_student_fcos.py -------------------------------------------------------------------------------- /projects/configs/10_coco/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/10_coco/base.py -------------------------------------------------------------------------------- /projects/configs/10_coco/group_rcnn_24e_10_percent_coco_detr_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/10_coco/group_rcnn_24e_10_percent_coco_detr_augmentation.py -------------------------------------------------------------------------------- /projects/configs/10_coco/group_rcnn_50e_10_percent_coco_detr_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/10_coco/group_rcnn_50e_10_percent_coco_detr_augmentation.py -------------------------------------------------------------------------------- /projects/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /projects/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /projects/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /projects/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/datasets/__init__.py -------------------------------------------------------------------------------- /projects/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/datasets/builder.py -------------------------------------------------------------------------------- /projects/datasets/point_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/datasets/point_coco.py -------------------------------------------------------------------------------- /projects/datasets/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/datasets/transform.py -------------------------------------------------------------------------------- /projects/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/models/__init__.py -------------------------------------------------------------------------------- /projects/models/group_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/models/group_rcnn.py -------------------------------------------------------------------------------- /projects/models/group_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/projects/models/group_roi_head.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/generate_anns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/generate_anns.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jshilong/GroupRCNN/HEAD/tools/train.py --------------------------------------------------------------------------------