├── .gitignore ├── .isort.cfg ├── .pre-commit-config.yaml ├── LICENSE ├── Makefile ├── README.md ├── SECURITY.md ├── configs ├── baseline │ ├── base.py │ ├── faster_rcnn_r101_caffe_fpn_coco_full_720k.py │ ├── faster_rcnn_r50_caffe_fpn_coco_full_720k.py │ └── faster_rcnn_r50_caffe_fpn_coco_partial_180k.py ├── soft_teacher │ ├── base.py │ ├── soft_teacher_faster_rcnn_r101_caffe_fpn_coco_full_1080k.py │ ├── soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py │ ├── soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_1440k.py │ └── soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py ├── soft_teacher_msi │ ├── base.py │ ├── soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py │ └── soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py ├── soft_teacher_voc │ ├── base.py │ ├── soft_teacher_voc_4x8_40k.py │ └── soft_teacher_voc_coco_4x8_40k.py └── soft_teacher_voc_msi │ ├── base.py │ ├── soft_teacher_voc_4x8_40k.py │ └── soft_teacher_voc_coco_4x8_40k.py ├── pipeline.png ├── requirements.txt ├── setup.py ├── ssod ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ └── masks │ │ ├── __init__.py │ │ └── structures.py ├── datasets │ ├── __init__.py │ ├── builder.py │ ├── dataset_wrappers.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── formatting.py │ │ ├── geo_utils.py │ │ ├── rand_aug.py │ │ └── semi_augment.py │ ├── pseudo_coco.py │ └── samplers │ │ ├── __init__.py │ │ └── semi_sampler.py ├── models │ ├── __init__.py │ ├── faster_rcnn_msi.py │ ├── multi_stream_detector.py │ ├── soft_teacher.py │ ├── soft_teacher_msi.py │ ├── soft_teacher_simple_msi.py │ └── utils │ │ ├── __init__.py │ │ └── bbox_utils.py ├── utils │ ├── __init__.py │ ├── exts │ │ ├── __init__.py │ │ └── optimizer_constructor.py │ ├── hooks │ │ ├── __init__.py │ │ ├── evaluation.py │ │ ├── mean_teacher.py │ │ ├── msi_evaluation.py │ │ ├── submodules_evaluation.py │ │ ├── weight_adjust.py │ │ └── weights_summary.py │ ├── logger.py │ ├── patch.py │ ├── signature.py │ ├── structure_utils.py │ └── vars.py └── version.py └── tools ├── dataset ├── prepare_coco_data.sh ├── semi_coco.py └── semi_coco.sh ├── dist_test.sh ├── dist_train.sh ├── dist_train_partially.sh ├── misc └── browse_dataset.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/SECURITY.md -------------------------------------------------------------------------------- /configs/baseline/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/baseline/base.py -------------------------------------------------------------------------------- /configs/baseline/faster_rcnn_r101_caffe_fpn_coco_full_720k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/baseline/faster_rcnn_r101_caffe_fpn_coco_full_720k.py -------------------------------------------------------------------------------- /configs/baseline/faster_rcnn_r50_caffe_fpn_coco_full_720k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/baseline/faster_rcnn_r50_caffe_fpn_coco_full_720k.py -------------------------------------------------------------------------------- /configs/baseline/faster_rcnn_r50_caffe_fpn_coco_partial_180k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/baseline/faster_rcnn_r50_caffe_fpn_coco_partial_180k.py -------------------------------------------------------------------------------- /configs/soft_teacher/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher/base.py -------------------------------------------------------------------------------- /configs/soft_teacher/soft_teacher_faster_rcnn_r101_caffe_fpn_coco_full_1080k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher/soft_teacher_faster_rcnn_r101_caffe_fpn_coco_full_1080k.py -------------------------------------------------------------------------------- /configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py -------------------------------------------------------------------------------- /configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_1440k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_1440k.py -------------------------------------------------------------------------------- /configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py -------------------------------------------------------------------------------- /configs/soft_teacher_msi/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_msi/base.py -------------------------------------------------------------------------------- /configs/soft_teacher_msi/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_msi/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_180k.py -------------------------------------------------------------------------------- /configs/soft_teacher_msi/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_msi/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc/base.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc/soft_teacher_voc_4x8_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc/soft_teacher_voc_4x8_40k.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc/soft_teacher_voc_coco_4x8_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc/soft_teacher_voc_coco_4x8_40k.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc_msi/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc_msi/base.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc_msi/soft_teacher_voc_4x8_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc_msi/soft_teacher_voc_4x8_40k.py -------------------------------------------------------------------------------- /configs/soft_teacher_voc_msi/soft_teacher_voc_coco_4x8_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/configs/soft_teacher_voc_msi/soft_teacher_voc_coco_4x8_40k.py -------------------------------------------------------------------------------- /pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/pipeline.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/setup.py -------------------------------------------------------------------------------- /ssod/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import * 2 | -------------------------------------------------------------------------------- /ssod/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/apis/__init__.py -------------------------------------------------------------------------------- /ssod/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/apis/inference.py -------------------------------------------------------------------------------- /ssod/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/apis/test.py -------------------------------------------------------------------------------- /ssod/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/apis/train.py -------------------------------------------------------------------------------- /ssod/core/__init__.py: -------------------------------------------------------------------------------- 1 | from .masks import TrimapMasks 2 | -------------------------------------------------------------------------------- /ssod/core/masks/__init__.py: -------------------------------------------------------------------------------- 1 | from .structures import TrimapMasks 2 | -------------------------------------------------------------------------------- /ssod/core/masks/structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/core/masks/structures.py -------------------------------------------------------------------------------- /ssod/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/__init__.py -------------------------------------------------------------------------------- /ssod/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/builder.py -------------------------------------------------------------------------------- /ssod/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /ssod/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /ssod/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /ssod/datasets/pipelines/geo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pipelines/geo_utils.py -------------------------------------------------------------------------------- /ssod/datasets/pipelines/rand_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pipelines/rand_aug.py -------------------------------------------------------------------------------- /ssod/datasets/pipelines/semi_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pipelines/semi_augment.py -------------------------------------------------------------------------------- /ssod/datasets/pseudo_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/pseudo_coco.py -------------------------------------------------------------------------------- /ssod/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /ssod/datasets/samplers/semi_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/datasets/samplers/semi_sampler.py -------------------------------------------------------------------------------- /ssod/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/__init__.py -------------------------------------------------------------------------------- /ssod/models/faster_rcnn_msi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/faster_rcnn_msi.py -------------------------------------------------------------------------------- /ssod/models/multi_stream_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/multi_stream_detector.py -------------------------------------------------------------------------------- /ssod/models/soft_teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/soft_teacher.py -------------------------------------------------------------------------------- /ssod/models/soft_teacher_msi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/soft_teacher_msi.py -------------------------------------------------------------------------------- /ssod/models/soft_teacher_simple_msi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/soft_teacher_simple_msi.py -------------------------------------------------------------------------------- /ssod/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .bbox_utils import Transform2D, filter_invalid 2 | -------------------------------------------------------------------------------- /ssod/models/utils/bbox_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/models/utils/bbox_utils.py -------------------------------------------------------------------------------- /ssod/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/__init__.py -------------------------------------------------------------------------------- /ssod/utils/exts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/exts/__init__.py -------------------------------------------------------------------------------- /ssod/utils/exts/optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/exts/optimizer_constructor.py -------------------------------------------------------------------------------- /ssod/utils/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/__init__.py -------------------------------------------------------------------------------- /ssod/utils/hooks/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/evaluation.py -------------------------------------------------------------------------------- /ssod/utils/hooks/mean_teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/mean_teacher.py -------------------------------------------------------------------------------- /ssod/utils/hooks/msi_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/msi_evaluation.py -------------------------------------------------------------------------------- /ssod/utils/hooks/submodules_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/submodules_evaluation.py -------------------------------------------------------------------------------- /ssod/utils/hooks/weight_adjust.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/weight_adjust.py -------------------------------------------------------------------------------- /ssod/utils/hooks/weights_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/hooks/weights_summary.py -------------------------------------------------------------------------------- /ssod/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/logger.py -------------------------------------------------------------------------------- /ssod/utils/patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/patch.py -------------------------------------------------------------------------------- /ssod/utils/signature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/signature.py -------------------------------------------------------------------------------- /ssod/utils/structure_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/structure_utils.py -------------------------------------------------------------------------------- /ssod/utils/vars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/utils/vars.py -------------------------------------------------------------------------------- /ssod/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/ssod/version.py -------------------------------------------------------------------------------- /tools/dataset/prepare_coco_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dataset/prepare_coco_data.sh -------------------------------------------------------------------------------- /tools/dataset/semi_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dataset/semi_coco.py -------------------------------------------------------------------------------- /tools/dataset/semi_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dataset/semi_coco.sh -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/dist_train_partially.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/dist_train_partially.sh -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/MixTeacher/HEAD/tools/train.py --------------------------------------------------------------------------------