python source code of pseudo

AerialDetection-master
- compile.sh
- demo
- DOTA_devkit
  - DOTA.py
  - prepare_dota1.py
  - SplitOnlyImage_multi_process.py
  - ImgSplit.py
  - prepare_dota1_5_aug.py
  - polyiou.i
  - setup.py
  - ResultMerge_multi_process.py
  - SplitOnlyImage.py
  - prepare_dota1_aug.py
  - poly_overlaps_test.py
  - rotate_test.py
  - polyiou.h
  - DOTA2COCO.py
  - ResultMerge.py
  - utils.py
  - dota_utils.py
  - results_obb2hbb.py
  - readme.md
  - nms.py
  - .gitignore
  - ImgSplit_multi_process.py
  - prepare_dota1_5.py
  - polyiou.cpp
  - HRSC2COCO.py
  - polyiou.py
  - rotate.py
- demo_large_image.py
- LICENSE
- setup.py
- .style.yapf
- TECHNICAL_DETAILS.md
- GETTING_STARTED.md
- configs
  - DOTA1_5
    - faster_rcnn_RoITrans_r50_fpn_1x_dota1_5_gap512_msTrainTest_rotationTrainTest.py
    - faster_rcnn_r50_fpn_1x_dota1_5.py
    - retinanet_obb_r50_fpn_2x_dota1_5.py
    - faster_rcnn_obb_r50_fpn_1x_dota1_5.py
    - faster_rcnn_RoITrans_r50_fpn_1x_dota1_5.py
    - faster_rcnn_obb_dpool_r50_fpn_1x_dota1_5.py
    - retinanet_r50_fpn_2x_dota1_5.py
    - cascade_mask_rcnn_r50_fpn_1x_dota1_5.py
    - faster_rcnn_h-obb_r50_fpn_1x_dota1_5.py
    - mask_rcnn_r50_fpn_1x_dota1_5.py
    - htc_without_semantic_r50_fpn_1x_dota1_5.py
  - mask_rcnn_r50_caffe_c4_1x.py
  - pascal_voc
    - faster_rcnn_r50_fpn_1x_voc0712.py
    - ssd512_voc.py
    - ssd300_voc.py
  - cascade_mask_rcnn_x101_64x4d_fpn_1x.py
  - gn+ws
    - faster_rcnn_r50_fpn_gn_ws_1x.py
    - mask_rcnn_x101_32x4d_fpn_gn_ws_2x.py
    - mask_rcnn_r50_fpn_gn_ws_2x.py
    - mask_rcnn_r50_fpn_gn_ws_20_23_24e.py
    - README.md
  - hrnet
    - mask_rcnn_hrnetv2p_w18_1x.py
    - cascade_rcnn_hrnetv2p_w32_20e.py
    - faster_rcnn_hrnetv2p_w40_1x.py
    - mask_rcnn_hrnetv2p_w32_1x.py
    - README.md
    - faster_rcnn_hrnetv2p_w18_1x.py
    - faster_rcnn_hrnetv2p_w32_1x.py
  - fcos
    - fcos_r50_caffe_fpn_gn_1x_4gpu.py
    - fcos_mstrain_640_800_x101_64x4d_fpn_gn_2x.py
    - README.md
    - fcos_mstrain_640_800_r101_caffe_fpn_gn_2x_4gpu.py
  - DOTA
    - faster_rcnn_r50_fpn_1x_dota.py
    - retinanet_obb_r50_fpn_2x_dota.py
    - cascade_mask_rcnn_r50_fpn_1x_dota.py
    - retinanet_r50_fpn_2x_dota.py
    - faster_rcnn_h-obb_r50_fpn_1x_dota.py
    - faster_rcnn_obb_dpool_r50_fpn_1x_dota.py
    - faster_rcnn_obb_r50_fpn_1x_dota.py
    - faster_rcnn_RoITrans_r50_fpn_1x_dota.py
    - htc_without_semantic_r50_fpn_1x_dota.py
    - mask_rcnn_r50_fpn_1x_dota.py
    - faster_rcnn_RoITrans_r50_fpn_1x_dota_gap512_msTrainTest_rotationTrainTest.py
  - rpn_r101_fpn_1x.py
  - faster_rcnn_x101_64x4d_fpn_1x.py
  - cascade_rcnn_r50_caffe_c4_1x.py
  - gn
    - mask_rcnn_r50_fpn_gn_2x.py
    - mask_rcnn_r50_fpn_gn_contrib_2x.py
    - README.md
    - mask_rcnn_r101_fpn_gn_2x.py
  - cascade_mask_rcnn_r50_fpn_1x.py
  - fast_mask_rcnn_r50_caffe_c4_1x.py
  - cascade_rcnn_x101_32x4d_fpn_1x.py
  - ghm
    - README.md
    - retinanet_ghm_r50_fpn_1x.py
  - cascade_rcnn_r50_fpn_1x.py
  - ssd512_coco.py
  - rpn_r50_caffe_c4_1x.py
  - mask_rcnn_r101_fpn_1x.py
  - cascade_rcnn_x101_64x4d_fpn_1x.py
  - scratch
    - scratch_faster_rcnn_r50_fpn_gn_6x.py
    - scratch_mask_rcnn_r50_fpn_gn_6x.py
    - README.md
  - rpn_x101_32x4d_fpn_1x.py
  - fast_mask_rcnn_r50_fpn_1x.py
  - retinanet_r50_fpn_1x.py
  - faster_rcnn_r101_fpn_1x.py
  - dcn
    - cascade_mask_rcnn_dconv_c3-c5_r50_fpn_1x.py
    - faster_rcnn_dconv_c3-c5_r50_fpn_1x.py
    - faster_rcnn_mdconv_c3-c5_r50_fpn_1x.py
    - cascade_rcnn_dconv_c3-c5_r50_fpn_1x.py
    - mask_rcnn_dconv_c3-c5_r50_fpn_1x.py
    - faster_rcnn_mdpool_r50_fpn_1x.py
    - faster_rcnn_dpool_r50_fpn_1x.py
    - README.md
    - faster_rcnn_dconv_c3-c5_x101_32x4d_fpn_1x.py
  - ssd300_coco.py
  - retinanet_r101_fpn_1x.py
  - cascade_rcnn_r101_fpn_1x.py
  - rpn_x101_64x4d_fpn_1x.py
  - fast_rcnn_r101_fpn_1x.py
  - mask_rcnn_x101_32x4d_fpn_1x.py
  - faster_rcnn_r50_fpn_1x.py
  - retinanet_x101_64x4d_fpn_1x.py
  - fast_rcnn_r50_fpn_1x.py
  - retinanet_x101_32x4d_fpn_1x.py
  - mask_rcnn_x101_64x4d_fpn_1x.py
  - gcnet
    - mask_rcnn_r4_gcb_c3-c5_r50_fpn_syncbn_1x.py
    - mask_rcnn_r16_gcb_c3-c5_r50_fpn_1x.py
    - mask_rcnn_r4_gcb_c3-c5_r50_fpn_1x.py
    - mask_rcnn_r50_fpn_sbn_1x.py
    - README.md
    - mask_rcnn_r16_gcb_c3-c5_r50_fpn_syncbn_1x.py
  - cascade_mask_rcnn_x101_32x4d_fpn_1x.py
  - libra_rcnn
    - libra_faster_rcnn_x101_64x4d_fpn_1x.py
    - libra_fast_rcnn_r50_fpn_1x.py
    - libra_faster_rcnn_r101_fpn_1x.py
    - libra_retinanet_r50_fpn_1x.py
    - README.md
    - libra_faster_rcnn_r50_fpn_1x.py
  - fast_mask_rcnn_r101_fpn_1x.py
  - cascade_mask_rcnn_r50_caffe_c4_1x.py
  - mask_rcnn_r50_fpn_1x.py
  - faster_rcnn_x101_32x4d_fpn_1x.py
  - htc
    - htc_without_semantic_r50_fpn_1x.py
    - htc_r50_fpn_20e.py
    - htc_r50_fpn_1x.py
    - README.md
    - htc_x101_32x4d_fpn_20e_16gpu.py
    - htc_r101_fpn_20e.py
    - htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn_20e.py
    - htc_x101_64x4d_fpn_20e_16gpu.py
  - faster_rcnn_r50_caffe_c4_1x.py
  - fast_rcnn_r50_caffe_c4_1x.py
  - faster_rcnn_ohem_r50_fpn_1x.py
  - rpn_r50_fpn_1x.py
  - guided_anchoring
    - ga_rpn_r101_caffe_rpn_1x.py
    - ga_fast_r50_caffe_fpn_1x.py
    - ga_rpn_x101_32x4d_fpn_1x.py
    - ga_retinanet_r50_caffe_fpn_1x.py
    - ga_rpn_r50_caffe_fpn_1x.py
    - README.md
    - ga_faster_x101_32x4d_fpn_1x.py
    - ga_faster_r50_caffe_fpn_1x.py
    - ga_retinanet_x101_32x4d_fpn_1x.py
  - cascade_mask_rcnn_r101_fpn_1x.py
- mmdet
  - datasets
    - coco.py
    - concat_dataset.py
    - repeat_dataset.py
    - extra_aug.py
    - DOTA.py
    - xml_style.py
    - DOTA1_5.py
    - rotate_aug.py
    - loader
      - sampler.py
      - build_loader.py
      - __init__.py
    - __init__.py
    - utils.py
    - HRSC.py
    - voc.py
    - DOTA2.py
    - transforms.py
    - custom.py
  - utils
    - flops_counter.py
    - registry.py
    - __init__.py
  - apis
    - env.py
    - train.py
    - __init__.py
    - inference.py
  - models
    - mask_heads
      - htc_mask_head.py
      - fcn_mask_head.py
      - __init__.py
      - fused_semantic_head.py
    - utils
      - conv_ws.py
      - scale.py
      - conv_module.py
      - norm.py
      - __init__.py
      - weight_init.py
    - anchor_heads
      - fcos_head.py
      - retina_head.py
      - guided_anchor_head.py
      - ga_rpn_head.py
      - ga_retina_head.py
      - anchor_head.py
      - rpn_head.py
      - __init__.py
      - retina_head_rbbox.py
      - anchor_head_rbbox.py
      - ssd_head.py
    - rroi_extractors
      - rbox_single_level.py
      - __init__.py
    - bbox_heads
      - convfc_bbox_head.py
      - __init__.py
      - bbox_head.py
    - losses
      - focal_loss.py
      - cross_entropy_loss.py
      - ghm_loss.py
      - smooth_l1_loss.py
      - __init__.py
      - balanced_l1_loss.py
      - iou_loss.py
    - plugins
      - __init__.py
      - non_local.py
    - detectors
      - test_mixins.py
      - single_stage.py
      - two_stage.py
      - retinanet_obb.py
      - fast_rcnn.py
      - rpn.py
      - two_stage_rbbox.py
      - faster_rcnn.py
      - cascade_rcnn.py
      - single_stage_rbbox.py
      - RoITransformer.py
      - retinanet.py
      - faster_rcnn_obb.py
      - __init__.py
      - fcos.py
      - base_new.py
      - htc.py
      - mask_rcnn.py
      - faster_rcnn_hbb_obb.py
      - base.py
    - registry.py
    - roi_extractors
      - __init__.py
      - single_level.py
    - __init__.py
    - backbones
      - ssd_vgg.py
      - hrnet.py
      - resnet.py
      - __init__.py
      - resnext.py
    - builder.py
    - rbbox_heads
      - convfc_rbbox_head.py
      - rbbox_head.py
      - __init__.py
    - necks
      - fpn.py
      - __init__.py
      - hrfpn.py
      - bfp.py
    - shared_heads
      - res_layer.py
      - __init__.py
  - __init__.py
  - ops
    - roi_align
      - src
        roi_align_cuda.cpp
        roi_align_kernel.cu
      - functions
        roi_align.py
        __init__.py
      - gradcheck.py
      - setup.py
      - __init__.py
      - modules
        roi_align.py
        __init__.py
    - nms
      - src
        nms_kernel.cu
        nms_cuda.cpp
        soft_nms_cpu.pyx
        nms_cpu.cpp
      - rnms_wrapper.py
      - setup.py
      - __init__.py
      - nms_wrapper.py
    - psroi_align_rotated
      - src
        psroi_align_rotated_kernel.cu
        psroi_align_rotated_cuda.cpp
      - functions
        psroi_align_rotated.py
        __init__.py
      - gradcheck.py
      - setup.py
      - __init__.py
      - modules
        psroi_align_rotated.py
        __init__.py
    - sigmoid_focal_loss
      - src
        sigmoid_focal_loss_cuda.cu
        sigmoid_focal_loss.cpp
      - functions
        __init__.py
        sigmoid_focal_loss.py
      - setup.py
      - __init__.py
      - modules
        __init__.py
        sigmoid_focal_loss.py
    - poly_nms
      - src
        poly_nms_kernel.cu
        poly_nms_cuda.cpp
      - poly_nms_wrapper.py
      - setup.py
      - __init__.py
    - dcn
      - src
        deform_pool_cuda_kernel.cu
        deform_conv_cuda.cpp
        deform_conv_cuda_kernel.cu
        deform_pool_cuda.cpp
      - functions
        deform_conv.py
        deform_pool.py
        __init__.py
      - setup.py
      - __init__.py
      - modules
        deform_conv.py
        deform_pool.py
        __init__.py
    - masked_conv
      - src
        masked_conv2d_cuda.cpp
        masked_conv2d_kernel.cu
      - functions
        masked_conv.py
        __init__.py
      - setup.py
      - __init__.py
      - modules
        masked_conv.py
        __init__.py
    - roi_align_rotated
      - src
        roi_align_rotated_kernel.cu
        roi_align_rotated_cuda.cpp
      - functions
        roi_align_rotated.py
        __init__.py
      - gradcheck.py
      - setup.py
      - __init__.py
      - modules
        roi_align_rotated.py
        __init__.py
    - gcb
      - __init__.py
      - context_block.py
    - __init__.py
    - roi_pool
      - src
        roi_pool_kernel.cu
        roi_pool_cuda.cpp
      - functions
        roi_pool.py
        __init__.py
      - gradcheck.py
      - setup.py
      - __init__.py
      - modules
        roi_pool.py
        __init__.py
  - core
    - utils
      - misc.py
      - dist_utils.py
      - __init__.py
    - anchor
      - anchor_target.py
      - anchor_target_rbbox.py
      - guided_anchor_target.py
      - __init__.py
      - anchor_generator.py
    - bbox
      - transforms_rbbox.py
      - bbox_target_rbbox.py
      - geometry_test.py
      - geometry.py
      - bbox_target.py
      - setup_linux.py
      - transforms_rbbox_test.py
      - __init__.py
      - transforms.py
      - bbox.pyx
      - samplers
        rbbox_random_sampler.py
        rbbox_base_sampler.py
        sampling_result.py
        pseudo_sampler.py
        __init__.py
        instance_balanced_pos_sampler.py
        random_sampler.py
        iou_balanced_neg_sampler.py
        ohem_sampler.py
        base_sampler.py
        combined_sampler.py
      - assigners
        approx_max_iou_assigner.py
        base_assigner.py
        assign_result.py
        max_iou_assigner.py
        max_iou_assigner_hbb_cy.py
        __init__.py
        max_iou_assigner_rbbox.py
      - assign_sampling.py
    - evaluation
      - recall.py
      - mean_ap.py
      - eval_hooks.py
      - coco_utils.py
      - class_names.py
      - bbox_overlaps.py
      - __init__.py
      - dota_utils.py
    - mask
      - mask_target.py
      - __init__.py
      - utils.py
    - __init__.py
    - loss
      - losses.py
      - __init__.py
    - post_processing
      - merge_augs_rotate.py
      - rbbox_nms.py
      - merge_augs.py
      - __init__.py
      - bbox_nms.py
- .travis.yml
- README.md
- MODEL_ZOO.md
- tools
  - coco_eval.py
  - get_flops.py
  - slurm_train.sh
  - slurm_test.sh
  - voc_eval.py
  - robustness_eval.py
  - dist_test.sh
  - test_robustness.py
  - detectron2pytorch.py
  - parse_results.py
  - upgrade_model_version.py
  - train.py
  - dist_train.sh
  - test.py
  - config_trans_map.py
  - analyze_logs.py
  - copy_models.py
  - convert_datasets
    - pascal_voc.py
  - publish_model.py
- requirements.txt
- .gitignore
- INSTALL.md

import torch

from .base_sampler import BaseSampler
from .sampling_result import SamplingResult


class PseudoSampler(BaseSampler):

    def __init__(self, **kwargs):
        pass

    def _sample_pos(self, **kwargs):
        raise NotImplementedError

    def _sample_neg(self, **kwargs):
        raise NotImplementedError

    def sample(self, assign_result, bboxes, gt_bboxes, **kwargs):
        pos_inds = torch.nonzero(
            assign_result.gt_inds > 0).squeeze(-1).unique()
        neg_inds = torch.nonzero(
            assign_result.gt_inds == 0).squeeze(-1).unique()
        gt_flags = bboxes.new_zeros(bboxes.shape[0], dtype=torch.uint8)
        sampling_result = SamplingResult(pos_inds, neg_inds, bboxes, gt_bboxes,
                                         assign_result, gt_flags)
        return sampling_result