├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── LICENSE ├── MANIFEST.in ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── coco_instance.py │ │ ├── lvis_v0.5_instance.py │ │ └── lvis_v1_instance.py │ ├── default_runtime.py │ ├── models │ │ └── mask_rcnn_r50_fpn.py │ └── schedules │ │ ├── schedule_1x.py │ │ ├── schedule_20e.py │ │ └── schedule_2x.py └── pcb │ ├── mask_rcnn_r101_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py │ ├── mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v0.5_fp16.py │ ├── mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v1_fp16.py │ ├── mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v0.5_fp16.py │ ├── mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v1_fp16.py │ ├── mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v0.5_fp16.py │ ├── mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v1_fp16.py │ ├── mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v0.5_fp16.py │ └── mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py ├── mmdet ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── anchor │ │ ├── __init__.py │ │ ├── anchor_generator.py │ │ ├── builder.py │ │ └── utils.py │ ├── bbox │ │ ├── __init__.py │ │ ├── assigners │ │ │ ├── __init__.py │ │ │ ├── approx_max_iou_assigner.py │ │ │ ├── assign_result.py │ │ │ ├── atss_assigner.py │ │ │ ├── base_assigner.py │ │ │ ├── center_region_assigner.py │ │ │ ├── grid_assigner.py │ │ │ ├── hungarian_assigner.py │ │ │ ├── max_iou_assigner.py │ │ │ ├── point_assigner.py │ │ │ ├── region_assigner.py │ │ │ ├── sim_ota_assigner.py │ │ │ └── uniform_assigner.py │ │ ├── builder.py │ │ ├── coder │ │ │ ├── __init__.py │ │ │ ├── base_bbox_coder.py │ │ │ ├── bucketing_bbox_coder.py │ │ │ ├── delta_xywh_bbox_coder.py │ │ │ ├── legacy_delta_xywh_bbox_coder.py │ │ │ ├── pseudo_bbox_coder.py │ │ │ ├── tblr_bbox_coder.py │ │ │ └── yolo_bbox_coder.py │ │ ├── demodata.py │ │ ├── iou_calculators │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── iou2d_calculator.py │ │ ├── match_costs │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ └── match_cost.py │ │ ├── samplers │ │ │ ├── __init__.py │ │ │ ├── base_sampler.py │ │ │ ├── combined_sampler.py │ │ │ ├── instance_balanced_pos_sampler.py │ │ │ ├── iou_balanced_neg_sampler.py │ │ │ ├── ohem_sampler.py │ │ │ ├── pseudo_sampler.py │ │ │ ├── random_sampler.py │ │ │ ├── sampling_result.py │ │ │ └── score_hlr_sampler.py │ │ └── transforms.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── bbox_overlaps.py │ │ ├── class_names.py │ │ ├── eval_hooks.py │ │ ├── mean_ap.py │ │ └── recall.py │ ├── export │ │ ├── __init__.py │ │ ├── model_wrappers.py │ │ ├── onnx_helper.py │ │ └── pytorch2onnx.py │ ├── hook │ │ ├── __init__.py │ │ ├── checkloss_hook.py │ │ ├── ema.py │ │ ├── sync_norm_hook.py │ │ ├── sync_random_size_hook.py │ │ ├── yolox_lrupdater_hook.py │ │ └── yolox_mode_switch_hook.py │ ├── mask │ │ ├── __init__.py │ │ ├── mask_target.py │ │ ├── structures.py │ │ └── utils.py │ ├── post_processing │ │ ├── __init__.py │ │ ├── bbox_nms.py │ │ └── merge_augs.py │ ├── utils │ │ ├── __init__.py │ │ ├── dist_utils.py │ │ └── misc.py │ └── visualization │ │ ├── __init__.py │ │ └── image.py ├── datasets │ ├── __init__.py │ ├── api_wrappers │ │ ├── __init__.py │ │ └── coco_api.py │ ├── builder.py │ ├── coco.py │ ├── custom.py │ ├── dataset_wrappers.py │ ├── lvis.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── auto_augment.py │ │ ├── compose.py │ │ ├── formating.py │ │ ├── instaboost.py │ │ ├── loading.py │ │ ├── test_time_aug.py │ │ └── transforms.py │ ├── samplers │ │ ├── __init__.py │ │ ├── distributed_sampler.py │ │ └── group_sampler.py │ ├── utils.py │ └── xml_style.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ └── resnet.py │ ├── builder.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── anchor_head.py │ │ ├── base_dense_head.py │ │ ├── dense_test_mixins.py │ │ └── rpn_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── base.py │ │ ├── mask_rcnn.py │ │ ├── rpn.py │ │ └── two_stage.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── cross_entropy_loss.py │ │ ├── cross_entropy_pcb_loss.py │ │ ├── eqlv2.py │ │ ├── eqlv2_pcb_loss.py │ │ ├── pisa_loss.py │ │ ├── seesaw_loss.py │ │ ├── seesaw_pcb_loss.py │ │ ├── smooth_l1_loss.py │ │ └── utils.py │ ├── necks │ │ ├── __init__.py │ │ └── fpn.py │ ├── roi_heads │ │ ├── __init__.py │ │ ├── base_roi_head.py │ │ ├── bbox_heads │ │ │ ├── __init__.py │ │ │ ├── bbox_head.py │ │ │ └── convfc_bbox_head.py │ │ ├── iterative_roi_head.py │ │ ├── mask_heads │ │ │ ├── __init__.py │ │ │ └── fcn_mask_head.py │ │ ├── roi_extractors │ │ │ ├── __init__.py │ │ │ ├── base_roi_extractor.py │ │ │ ├── generic_roi_extractor.py │ │ │ └── single_level_roi_extractor.py │ │ ├── shared_heads │ │ │ ├── __init__.py │ │ │ └── res_layer.py │ │ ├── standard_roi_head.py │ │ └── test_mixins.py │ └── utils │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── mlp.py │ │ ├── normed_predictor.py │ │ └── res_layer.py ├── utils │ ├── __init__.py │ ├── collect_env.py │ ├── contextmanagers.py │ ├── logger.py │ ├── profiling.py │ ├── util_mixins.py │ └── util_random.py └── version.py ├── model-index.yml ├── pytest.ini ├── requirements.txt ├── requirements ├── build.txt ├── docs.txt ├── mminstall.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── resources └── PCB_framework.png ├── setup.cfg ├── setup.py └── tools ├── analysis_tools ├── analyze_logs.py ├── analyze_results.py ├── benchmark.py ├── coco_error_analysis.py ├── eval_metric.py ├── get_flops.py ├── optimize_anchors.py ├── robustness_eval.py └── test_robustness.py ├── dataset_converters ├── cityscapes.py ├── images2coco.py └── pascal_voc.py ├── deployment ├── mmdet2torchserve.py ├── mmdet_handler.py ├── onnx2tensorrt.py ├── pytorch2onnx.py └── test.py ├── dist_test.sh ├── dist_train.sh ├── misc ├── browse_dataset.py └── print_config.py ├── model_converters ├── detectron2pytorch.py ├── publish_model.py ├── regnet2mmdet.py ├── selfsup2mmdet.py ├── upgrade_model_version.py └── upgrade_ssd_version.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r101_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r101_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v0.5_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v0.5_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v1_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_mstrain_eqlv2_pcb_2x_lvis_v1_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v0.5_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v0.5_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v1_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_mstrain_pcb_2x_lvis_v1_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v0.5_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v0.5_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v1_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_sample1e-3_mstrain_pcb_2x_lvis_v1_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v0.5_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v0.5_fp16.py -------------------------------------------------------------------------------- /configs/pcb/mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/configs/pcb/mask_rcnn_r50_fpn_sample1e-3_seesaw_pcb_normed_mask_mstrain_2x_lvis_v1_fp16.py -------------------------------------------------------------------------------- /mmdet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/__init__.py -------------------------------------------------------------------------------- /mmdet/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/apis/__init__.py -------------------------------------------------------------------------------- /mmdet/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/apis/inference.py -------------------------------------------------------------------------------- /mmdet/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/apis/test.py -------------------------------------------------------------------------------- /mmdet/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/apis/train.py -------------------------------------------------------------------------------- /mmdet/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/__init__.py -------------------------------------------------------------------------------- /mmdet/core/anchor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/anchor/__init__.py -------------------------------------------------------------------------------- /mmdet/core/anchor/anchor_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/anchor/anchor_generator.py -------------------------------------------------------------------------------- /mmdet/core/anchor/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/anchor/builder.py -------------------------------------------------------------------------------- /mmdet/core/anchor/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/anchor/utils.py -------------------------------------------------------------------------------- /mmdet/core/bbox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/approx_max_iou_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/approx_max_iou_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/assign_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/assign_result.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/atss_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/atss_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/base_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/base_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/center_region_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/center_region_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/grid_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/grid_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/hungarian_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/hungarian_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/max_iou_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/max_iou_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/point_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/point_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/region_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/region_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/sim_ota_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/sim_ota_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/assigners/uniform_assigner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/assigners/uniform_assigner.py -------------------------------------------------------------------------------- /mmdet/core/bbox/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/builder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/base_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/base_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/bucketing_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/bucketing_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/delta_xywh_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/delta_xywh_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/legacy_delta_xywh_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/legacy_delta_xywh_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/pseudo_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/pseudo_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/tblr_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/tblr_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/coder/yolo_bbox_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/coder/yolo_bbox_coder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/demodata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/demodata.py -------------------------------------------------------------------------------- /mmdet/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/iou_calculators/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/iou_calculators/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/iou_calculators/builder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/iou_calculators/iou2d_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/iou_calculators/iou2d_calculator.py -------------------------------------------------------------------------------- /mmdet/core/bbox/match_costs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/match_costs/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/match_costs/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/match_costs/builder.py -------------------------------------------------------------------------------- /mmdet/core/bbox/match_costs/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/match_costs/match_cost.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/__init__.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/base_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/base_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/combined_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/combined_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/instance_balanced_pos_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/instance_balanced_pos_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/ohem_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/ohem_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/pseudo_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/pseudo_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/random_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/random_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/sampling_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/sampling_result.py -------------------------------------------------------------------------------- /mmdet/core/bbox/samplers/score_hlr_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/samplers/score_hlr_sampler.py -------------------------------------------------------------------------------- /mmdet/core/bbox/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/bbox/transforms.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/bbox_overlaps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/bbox_overlaps.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/class_names.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/class_names.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/mean_ap.py -------------------------------------------------------------------------------- /mmdet/core/evaluation/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/evaluation/recall.py -------------------------------------------------------------------------------- /mmdet/core/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/export/__init__.py -------------------------------------------------------------------------------- /mmdet/core/export/model_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/export/model_wrappers.py -------------------------------------------------------------------------------- /mmdet/core/export/onnx_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/export/onnx_helper.py -------------------------------------------------------------------------------- /mmdet/core/export/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/export/pytorch2onnx.py -------------------------------------------------------------------------------- /mmdet/core/hook/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/__init__.py -------------------------------------------------------------------------------- /mmdet/core/hook/checkloss_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/checkloss_hook.py -------------------------------------------------------------------------------- /mmdet/core/hook/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/ema.py -------------------------------------------------------------------------------- /mmdet/core/hook/sync_norm_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/sync_norm_hook.py -------------------------------------------------------------------------------- /mmdet/core/hook/sync_random_size_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/sync_random_size_hook.py -------------------------------------------------------------------------------- /mmdet/core/hook/yolox_lrupdater_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/yolox_lrupdater_hook.py -------------------------------------------------------------------------------- /mmdet/core/hook/yolox_mode_switch_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/hook/yolox_mode_switch_hook.py -------------------------------------------------------------------------------- /mmdet/core/mask/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/mask/__init__.py -------------------------------------------------------------------------------- /mmdet/core/mask/mask_target.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/mask/mask_target.py -------------------------------------------------------------------------------- /mmdet/core/mask/structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/mask/structures.py -------------------------------------------------------------------------------- /mmdet/core/mask/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/mask/utils.py -------------------------------------------------------------------------------- /mmdet/core/post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/post_processing/__init__.py -------------------------------------------------------------------------------- /mmdet/core/post_processing/bbox_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/post_processing/bbox_nms.py -------------------------------------------------------------------------------- /mmdet/core/post_processing/merge_augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/post_processing/merge_augs.py -------------------------------------------------------------------------------- /mmdet/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/utils/__init__.py -------------------------------------------------------------------------------- /mmdet/core/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/utils/dist_utils.py -------------------------------------------------------------------------------- /mmdet/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/utils/misc.py -------------------------------------------------------------------------------- /mmdet/core/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/visualization/__init__.py -------------------------------------------------------------------------------- /mmdet/core/visualization/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/core/visualization/image.py -------------------------------------------------------------------------------- /mmdet/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/__init__.py -------------------------------------------------------------------------------- /mmdet/datasets/api_wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/api_wrappers/__init__.py -------------------------------------------------------------------------------- /mmdet/datasets/api_wrappers/coco_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/api_wrappers/coco_api.py -------------------------------------------------------------------------------- /mmdet/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/builder.py -------------------------------------------------------------------------------- /mmdet/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/coco.py -------------------------------------------------------------------------------- /mmdet/datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/custom.py -------------------------------------------------------------------------------- /mmdet/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmdet/datasets/lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/lvis.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/auto_augment.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/instaboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/instaboost.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/test_time_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/test_time_aug.py -------------------------------------------------------------------------------- /mmdet/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /mmdet/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /mmdet/datasets/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /mmdet/datasets/samplers/group_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/samplers/group_sampler.py -------------------------------------------------------------------------------- /mmdet/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/utils.py -------------------------------------------------------------------------------- /mmdet/datasets/xml_style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/datasets/xml_style.py -------------------------------------------------------------------------------- /mmdet/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/__init__.py -------------------------------------------------------------------------------- /mmdet/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmdet/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/backbones/resnet.py -------------------------------------------------------------------------------- /mmdet/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/builder.py -------------------------------------------------------------------------------- /mmdet/models/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/dense_heads/__init__.py -------------------------------------------------------------------------------- /mmdet/models/dense_heads/anchor_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/dense_heads/anchor_head.py -------------------------------------------------------------------------------- /mmdet/models/dense_heads/base_dense_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/dense_heads/base_dense_head.py -------------------------------------------------------------------------------- /mmdet/models/dense_heads/dense_test_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/dense_heads/dense_test_mixins.py -------------------------------------------------------------------------------- /mmdet/models/dense_heads/rpn_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/dense_heads/rpn_head.py -------------------------------------------------------------------------------- /mmdet/models/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/detectors/__init__.py -------------------------------------------------------------------------------- /mmdet/models/detectors/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/detectors/base.py -------------------------------------------------------------------------------- /mmdet/models/detectors/mask_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/detectors/mask_rcnn.py -------------------------------------------------------------------------------- /mmdet/models/detectors/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/detectors/rpn.py -------------------------------------------------------------------------------- /mmdet/models/detectors/two_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/detectors/two_stage.py -------------------------------------------------------------------------------- /mmdet/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/__init__.py -------------------------------------------------------------------------------- /mmdet/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/accuracy.py -------------------------------------------------------------------------------- /mmdet/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/cross_entropy_pcb_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/cross_entropy_pcb_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/eqlv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/eqlv2.py -------------------------------------------------------------------------------- /mmdet/models/losses/eqlv2_pcb_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/eqlv2_pcb_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/pisa_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/pisa_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/seesaw_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/seesaw_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/seesaw_pcb_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/seesaw_pcb_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/smooth_l1_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/smooth_l1_loss.py -------------------------------------------------------------------------------- /mmdet/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/losses/utils.py -------------------------------------------------------------------------------- /mmdet/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/necks/__init__.py -------------------------------------------------------------------------------- /mmdet/models/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/necks/fpn.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/__init__.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/base_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/base_roi_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/bbox_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/bbox_heads/__init__.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/bbox_heads/bbox_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/bbox_heads/bbox_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/iterative_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/iterative_roi_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/mask_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/mask_heads/__init__.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/mask_heads/fcn_mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/roi_extractors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/roi_extractors/__init__.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/roi_extractors/base_roi_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/roi_extractors/base_roi_extractor.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/roi_extractors/generic_roi_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/roi_extractors/generic_roi_extractor.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/roi_extractors/single_level_roi_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/roi_extractors/single_level_roi_extractor.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/shared_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/shared_heads/__init__.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/shared_heads/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/shared_heads/res_layer.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/standard_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/standard_roi_head.py -------------------------------------------------------------------------------- /mmdet/models/roi_heads/test_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/roi_heads/test_mixins.py -------------------------------------------------------------------------------- /mmdet/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/utils/__init__.py -------------------------------------------------------------------------------- /mmdet/models/utils/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/utils/builder.py -------------------------------------------------------------------------------- /mmdet/models/utils/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/utils/mlp.py -------------------------------------------------------------------------------- /mmdet/models/utils/normed_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/utils/normed_predictor.py -------------------------------------------------------------------------------- /mmdet/models/utils/res_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/models/utils/res_layer.py -------------------------------------------------------------------------------- /mmdet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/__init__.py -------------------------------------------------------------------------------- /mmdet/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/collect_env.py -------------------------------------------------------------------------------- /mmdet/utils/contextmanagers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/contextmanagers.py -------------------------------------------------------------------------------- /mmdet/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/logger.py -------------------------------------------------------------------------------- /mmdet/utils/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/profiling.py -------------------------------------------------------------------------------- /mmdet/utils/util_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/util_mixins.py -------------------------------------------------------------------------------- /mmdet/utils/util_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/utils/util_random.py -------------------------------------------------------------------------------- /mmdet/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/mmdet/version.py -------------------------------------------------------------------------------- /model-index.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/model-index.yml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/pytest.ini -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/build.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/build.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.8 2 | -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/optional.txt -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /resources/PCB_framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/resources/PCB_framework.png -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/setup.py -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /tools/analysis_tools/analyze_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/analyze_results.py -------------------------------------------------------------------------------- /tools/analysis_tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/benchmark.py -------------------------------------------------------------------------------- /tools/analysis_tools/coco_error_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/coco_error_analysis.py -------------------------------------------------------------------------------- /tools/analysis_tools/eval_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/eval_metric.py -------------------------------------------------------------------------------- /tools/analysis_tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/get_flops.py -------------------------------------------------------------------------------- /tools/analysis_tools/optimize_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/optimize_anchors.py -------------------------------------------------------------------------------- /tools/analysis_tools/robustness_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/robustness_eval.py -------------------------------------------------------------------------------- /tools/analysis_tools/test_robustness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/analysis_tools/test_robustness.py -------------------------------------------------------------------------------- /tools/dataset_converters/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/dataset_converters/cityscapes.py -------------------------------------------------------------------------------- /tools/dataset_converters/images2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/dataset_converters/images2coco.py -------------------------------------------------------------------------------- /tools/dataset_converters/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/dataset_converters/pascal_voc.py -------------------------------------------------------------------------------- /tools/deployment/mmdet2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/deployment/mmdet2torchserve.py -------------------------------------------------------------------------------- /tools/deployment/mmdet_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/deployment/mmdet_handler.py -------------------------------------------------------------------------------- /tools/deployment/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/deployment/onnx2tensorrt.py -------------------------------------------------------------------------------- /tools/deployment/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/deployment/pytorch2onnx.py -------------------------------------------------------------------------------- /tools/deployment/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/deployment/test.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/misc/print_config.py -------------------------------------------------------------------------------- /tools/model_converters/detectron2pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/detectron2pytorch.py -------------------------------------------------------------------------------- /tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /tools/model_converters/regnet2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/regnet2mmdet.py -------------------------------------------------------------------------------- /tools/model_converters/selfsup2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/selfsup2mmdet.py -------------------------------------------------------------------------------- /tools/model_converters/upgrade_model_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/upgrade_model_version.py -------------------------------------------------------------------------------- /tools/model_converters/upgrade_ssd_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/model_converters/upgrade_ssd_version.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/PCB/HEAD/tools/train.py --------------------------------------------------------------------------------