├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── INSTALL.md ├── LICENSE ├── README.md ├── configs ├── FCOS │ ├── Base-FCOS.yaml │ ├── R_50_1x.yaml │ └── coco-standard │ │ ├── fcos_R_50_ut2_sup10_run0.yaml │ │ ├── fcos_R_50_ut2_sup1_run0.yaml │ │ ├── fcos_R_50_ut2_sup2_run0.yaml │ │ └── fcos_R_50_ut2_sup5_run0.yaml └── Faster-RCNN │ ├── Base-RCNN-FPN.yaml │ └── coco-standard │ ├── faster_rcnn_R_50_FPN_ut2_sup10_run0.yaml │ ├── faster_rcnn_R_50_FPN_ut2_sup1_run0.yaml │ ├── faster_rcnn_R_50_FPN_ut2_sup2_run0.yaml │ └── faster_rcnn_R_50_FPN_ut2_sup5_run0.yaml ├── dataseed └── COCO_supervision.txt ├── teaser ├── pytorch-logo-dark.png └── teaser_utv2.png ├── train_net.py └── ubteacher ├── __init__.py ├── checkpoint └── detection_checkpoint.py ├── config.py ├── data ├── __init__.py ├── build.py ├── common.py ├── dataset_mapper.py ├── datasets │ └── builtin.py ├── detection_utils.py └── transforms │ └── augmentation_impl.py ├── engine ├── __init__.py └── trainer.py ├── evaluation ├── coco_evaluation.py └── evaluator.py ├── layers ├── __init__.py ├── iou_loss.py ├── kl_loss.py ├── ml_nms.py └── naive_group_norm.py ├── modeling ├── __init__.py ├── backbone │ └── fpn.py ├── box_regression.py ├── fcos │ ├── __init__.py │ ├── fcos.py │ ├── fcos_outputs.py │ ├── loss.py │ └── utils.py ├── meta_arch │ ├── rcnn.py │ └── ts_ensemble.py ├── one_stage_detector.py ├── proposal_generator │ └── rpn.py ├── pseudo_generator.py └── roi_heads │ ├── fast_rcnn.py │ └── roi_heads.py ├── solver ├── build.py └── lr_scheduler.py └── utils └── comm.py /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/README.md -------------------------------------------------------------------------------- /configs/FCOS/Base-FCOS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/Base-FCOS.yaml -------------------------------------------------------------------------------- /configs/FCOS/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/FCOS/coco-standard/fcos_R_50_ut2_sup10_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/coco-standard/fcos_R_50_ut2_sup10_run0.yaml -------------------------------------------------------------------------------- /configs/FCOS/coco-standard/fcos_R_50_ut2_sup1_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/coco-standard/fcos_R_50_ut2_sup1_run0.yaml -------------------------------------------------------------------------------- /configs/FCOS/coco-standard/fcos_R_50_ut2_sup2_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/coco-standard/fcos_R_50_ut2_sup2_run0.yaml -------------------------------------------------------------------------------- /configs/FCOS/coco-standard/fcos_R_50_ut2_sup5_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/FCOS/coco-standard/fcos_R_50_ut2_sup5_run0.yaml -------------------------------------------------------------------------------- /configs/Faster-RCNN/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/Faster-RCNN/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup10_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup10_run0.yaml -------------------------------------------------------------------------------- /configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup1_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup1_run0.yaml -------------------------------------------------------------------------------- /configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup2_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup2_run0.yaml -------------------------------------------------------------------------------- /configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup5_run0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/configs/Faster-RCNN/coco-standard/faster_rcnn_R_50_FPN_ut2_sup5_run0.yaml -------------------------------------------------------------------------------- /dataseed/COCO_supervision.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/dataseed/COCO_supervision.txt -------------------------------------------------------------------------------- /teaser/pytorch-logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/teaser/pytorch-logo-dark.png -------------------------------------------------------------------------------- /teaser/teaser_utv2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/teaser/teaser_utv2.png -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/train_net.py -------------------------------------------------------------------------------- /ubteacher/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/__init__.py -------------------------------------------------------------------------------- /ubteacher/checkpoint/detection_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/checkpoint/detection_checkpoint.py -------------------------------------------------------------------------------- /ubteacher/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/config.py -------------------------------------------------------------------------------- /ubteacher/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/__init__.py -------------------------------------------------------------------------------- /ubteacher/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/build.py -------------------------------------------------------------------------------- /ubteacher/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/common.py -------------------------------------------------------------------------------- /ubteacher/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/dataset_mapper.py -------------------------------------------------------------------------------- /ubteacher/data/datasets/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/datasets/builtin.py -------------------------------------------------------------------------------- /ubteacher/data/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/detection_utils.py -------------------------------------------------------------------------------- /ubteacher/data/transforms/augmentation_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/data/transforms/augmentation_impl.py -------------------------------------------------------------------------------- /ubteacher/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/engine/__init__.py -------------------------------------------------------------------------------- /ubteacher/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/engine/trainer.py -------------------------------------------------------------------------------- /ubteacher/evaluation/coco_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/evaluation/coco_evaluation.py -------------------------------------------------------------------------------- /ubteacher/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/evaluation/evaluator.py -------------------------------------------------------------------------------- /ubteacher/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/layers/__init__.py -------------------------------------------------------------------------------- /ubteacher/layers/iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/layers/iou_loss.py -------------------------------------------------------------------------------- /ubteacher/layers/kl_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/layers/kl_loss.py -------------------------------------------------------------------------------- /ubteacher/layers/ml_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/layers/ml_nms.py -------------------------------------------------------------------------------- /ubteacher/layers/naive_group_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/layers/naive_group_norm.py -------------------------------------------------------------------------------- /ubteacher/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/__init__.py -------------------------------------------------------------------------------- /ubteacher/modeling/backbone/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/backbone/fpn.py -------------------------------------------------------------------------------- /ubteacher/modeling/box_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/box_regression.py -------------------------------------------------------------------------------- /ubteacher/modeling/fcos/__init__.py: -------------------------------------------------------------------------------- 1 | from .fcos import FCOS # noqa 2 | -------------------------------------------------------------------------------- /ubteacher/modeling/fcos/fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/fcos/fcos.py -------------------------------------------------------------------------------- /ubteacher/modeling/fcos/fcos_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/fcos/fcos_outputs.py -------------------------------------------------------------------------------- /ubteacher/modeling/fcos/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/fcos/loss.py -------------------------------------------------------------------------------- /ubteacher/modeling/fcos/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/fcos/utils.py -------------------------------------------------------------------------------- /ubteacher/modeling/meta_arch/rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/meta_arch/rcnn.py -------------------------------------------------------------------------------- /ubteacher/modeling/meta_arch/ts_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/meta_arch/ts_ensemble.py -------------------------------------------------------------------------------- /ubteacher/modeling/one_stage_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/one_stage_detector.py -------------------------------------------------------------------------------- /ubteacher/modeling/proposal_generator/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/proposal_generator/rpn.py -------------------------------------------------------------------------------- /ubteacher/modeling/pseudo_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/pseudo_generator.py -------------------------------------------------------------------------------- /ubteacher/modeling/roi_heads/fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/roi_heads/fast_rcnn.py -------------------------------------------------------------------------------- /ubteacher/modeling/roi_heads/roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/modeling/roi_heads/roi_heads.py -------------------------------------------------------------------------------- /ubteacher/solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/solver/build.py -------------------------------------------------------------------------------- /ubteacher/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/solver/lr_scheduler.py -------------------------------------------------------------------------------- /ubteacher/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/unbiased-teacher-v2/HEAD/ubteacher/utils/comm.py --------------------------------------------------------------------------------