MCPcopy Index your code

hub / github.com/YXB-NKU/Strip-R-CNN / functions

Functions746 in github.com/YXB-NKU/Strip-R-CNN

↓ 1 callersFunctiondraw_rbboxes
Draw oriented bounding boxes on the axes. Args: ax (matplotlib.Axes): The input axes. bboxes (ndarray): The input bounding boxes
mmrotate/core/visualization/image.py:40
↓ 1 callersMethodem_runner
Performs one iteration of the expectation-maximization algorithm by calling the respective subroutines. Args: x (torch.Te
mmrotate/core/bbox/utils/gmm.py:276
↓ 1 callersFunctionennAvgPool
enn Average Pooling. Args: inplanes (int): The number of input channel. kernel_size (int, optional): The size of kernel.
mmrotate/models/utils/enn.py:122
↓ 1 callersFunctionennInterpolate
enn Interpolate.
mmrotate/models/utils/enn.py:154
↓ 1 callersFunctionennTrivialConv
enn convolution with trivial input featurn. Args: in_channels (List[int]): Number of input channels per scale. out_channels (int)
mmrotate/models/utils/enn.py:76
↓ 1 callersMethodevaluate_output_shape
Evaluate output shape.
mmrotate/models/backbones/re_resnet.py:129
↓ 1 callersMethodfeature_cosine_similarity
Compute the points features similarity for points-wise correlation. Args: points_features (torch.tensor): sampling point feature
mmrotate/models/dense_heads/oriented_reppoints_head.py:404
↓ 1 callersMethodfilter_border
Filter the box whose center point is outside or whose side length is less than 5.
mmrotate/datasets/pipelines/transforms.py:193
↓ 1 callersMethodforward
Forward features from the upstream network. Args: feats (tuple[Tensor]): Features from the upstream network, each is
mmrotate/models/dense_heads/rotated_anchor_head.py:131
↓ 1 callersMethodforward_features
(self, x)
mmrotate/models/backbones/stripnet.py:191
↓ 1 callersFunctiongen_packages_items
()
setup.py:84
↓ 1 callersMethodget_anchors
Get anchors according to feature map sizes. Args: featmap_sizes (list[tuple]): Multi-level feature map sizes. img_met
mmrotate/models/dense_heads/rotated_anchor_head.py:150
↓ 1 callersMethodget_bboxes
Transform network output for a batch into bbox predictions. Args: cls_scores (list[Tensor]): Box scores for each scale level
mmrotate/models/dense_heads/rotated_anchor_head.py:515
↓ 1 callersMethodget_cfa_targets
Compute corresponding GT box and classification targets for proposals. Args: proposals_list (list[list]): Multi level poi
mmrotate/models/dense_heads/rotated_reppoints_head.py:462
↓ 1 callersFunctionget_cls_results
Get det results and gt information of a certain class. Args: det_results (list[list]): Same as `eval_map()`. annotations (list[di
mmrotate/core/evaluation/eval_map.py:97
↓ 1 callersFunctionget_horizontal_bboxes
Get horizontal bboxes from polygons. Args: gt_rbboxes (torch.Tensor): Groundtruth polygons, shape (k, 8). Returns: gt_rect_b
mmrotate/core/bbox/assigners/sas_assigner.py:27
↓ 1 callersMethodget_horizontal_bboxes
get_horizontal_bboxes from polygons. Args: gt_rbboxes (torch.Tensor): Groundtruth polygons, shape (k, 8). Returns:
mmrotate/core/bbox/assigners/convex_assigner.py:28
↓ 1 callersMethodget_horizontal_bboxes
get_horizontal_bboxes from polygons. Args: gt_rbboxes (torch.Tensor): Groundtruth polygons, shape (k, 8). Returns:
mmrotate/core/bbox/assigners/atss_kld_assigner.py:238
↓ 1 callersMethodget_indices
Get the indices of ORConv2d.
mmrotate/models/utils/orconv.py:68
↓ 1 callersFunctionget_multiscale_patch
Get multiscale patch sizes and steps. Args: sizes (list): A list of patch sizes. steps (list): A list of steps to slide patches.
mmrotate/core/patch/split.py:8
↓ 1 callersMethodget_offset
Get the offset of AlignConv.
mmrotate/models/detectors/utils.py:41
↓ 1 callersMethodget_score
Computes the log-likelihood of the data under the model. Args: x (torch.Tensor): (T, n, 1, d) sum_data (bool,optional
mmrotate/core/bbox/utils/gmm.py:290
↓ 1 callersFunctionget_sliding_window
Get sliding windows. Args: info (dict): Dict of image's width and height. sizes (list): List of window's sizes. gaps (lis
tools/data/dota/split/img_split.py:137
↓ 1 callersFunctionget_sliding_window
Get sliding windows. Args: info (dict): Dict of image's width and height. sizes (list): List of window's sizes. gaps (lis
tools/data/fair/img_split.py:139
↓ 1 callersMethodget_targets
Compute regression, classification and centerness targets for points in multiple images. Args: points (list[Tensor]): Poi
mmrotate/models/dense_heads/rotated_fcos_head.py:318
↓ 1 callersMethodget_targets
Compute regression and classification targets for anchors in multiple images. Args: anchor_list (list[list[Tensor]]): Mul
mmrotate/models/dense_heads/rotated_anchor_head.py:290
↓ 1 callersFunctionget_version
Get version of mmrotate.
setup.py:16
↓ 1 callersFunctionget_window_obj
Args: info (dict): Dict of bbox annotations. windows (np.array): information of sliding windows. iof_thr (float): Thresh
tools/data/dota/split/img_split.py:247
↓ 1 callersFunctionget_window_obj
Args: info (dict): Dict of bbox annotations. windows (np.array): information of sliding windows. iof_thr (float): Thresh
tools/data/fair/img_split.py:249
↓ 1 callersFunctionhbb2obb
Convert horizontal bounding boxes to oriented bounding boxes. Args: hbbs (torch.Tensor): [x_lt,y_lt,x_rb,y_rb] version (Str): ang
mmrotate/core/bbox/transforms.py:221
↓ 1 callersFunctionhbb2obb_le135
Convert horizontal bounding boxes to oriented bounding boxes. Args: hbbs (torch.Tensor): [x_lt,y_lt,x_rb,y_rb] Returns: obbs
mmrotate/core/bbox/transforms.py:597
↓ 1 callersFunctionhbb2obb_le90
Convert horizontal bounding boxes to oriented bounding boxes. Args: hbbs (torch.Tensor): [x_lt,y_lt,x_rb,y_rb] Returns: obbs
mmrotate/core/bbox/transforms.py:617
↓ 1 callersFunctionhbb2obb_oc
Convert horizontal bounding boxes to oriented bounding boxes. Args: hbbs (torch.Tensor): [x_lt,y_lt,x_rb,y_rb] Returns: obbs
mmrotate/core/bbox/transforms.py:579
↓ 1 callersFunctioninference_detector_by_patches
inference patches with the detector. Split huge image(s) into patches and inference them with the detector. Finally, merge patch results on o
mmrotate/apis/inference.py:13
↓ 1 callersMethodinit_assigner_sampler
Initialize assigner and sampler.
mmrotate/models/roi_heads/rotate_standard_roi_head.py:55
↓ 1 callersMethodinit_assigner_sampler
Initialize assigner and sampler for each stage.
mmrotate/models/roi_heads/roi_trans_roi_head.py:81
↓ 1 callersMethodinit_bbox_head
Initialize ``bbox_head``. Args: bbox_roi_extractor (dict): Config of ``bbox_roi_extractor``. bbox_head (dict): Config
mmrotate/models/roi_heads/rotate_standard_roi_head.py:64
↓ 1 callersMethodinit_bbox_head
Initialize box head and box roi extractor. Args: bbox_roi_extractor (dict): Config of box roi extractor. bbox_head (d
mmrotate/models/roi_heads/roi_trans_roi_head.py:61
↓ 1 callersMethodinit_weights
(self)
mmrotate/models/roi_heads/bbox_heads/utils_basic.py:32
↓ 1 callersMethodinit_weights
(self)
mmrotate/models/roi_heads/bbox_heads/plugins.py:27
↓ 1 callersFunctionkfiou_loss
Kalman filter IoU loss. Args: pred (torch.Tensor): Predicted bboxes. target (torch.Tensor): Corresponding gt bboxes. pred
mmrotate/models/losses/kf_iou_loss.py:38
↓ 1 callersFunctionkld_loss
Kullback-Leibler Divergence loss. Args: pred (torch.Tensor): Convexes with shape (N, 9, 2). target (torch.Tensor): Polygons with
mmrotate/models/losses/kld_reppoints_loss.py:40
↓ 1 callersMethodkld_mixture2single
Compute Kullback-Leibler Divergence between two Gaussian distribution. Args: g1 (dict[str, torch.Tensor]): Gaussian distr
mmrotate/core/bbox/assigners/atss_kld_assigner.py:202
↓ 1 callersMethodkld_overlaps
Compute overlaps between polygons and points by Kullback-Leibler Divergence loss. Args: gt_rbboxes (torch.Tensor): Ground
mmrotate/core/bbox/assigners/atss_kld_assigner.py:180
↓ 1 callersFunctionkld_single2single
Compute Kullback-Leibler Divergence. Args: g1 (dict[str, torch.Tensor]): Gaussian distribution 1. g2 (torch.Tensor): Gaussian dis
mmrotate/models/losses/kld_reppoints_loss.py:10
↓ 1 callersFunctionload_dota
Load DOTA dataset. Args: img_dir (str): Path of images. ann_dir (str): Path of annotations. nproc (int): number of proces
tools/data/dota/split/img_split.py:438
↓ 1 callersFunctionload_dota
Load DOTA dataset. Args: img_dir (str): Path of images. ann_dir (str): Path of annotations. nproc (int): number of proces
tools/data/fair/img_split.py:440
↓ 1 callersFunctionload_json_logs
Load and convert json_logs to log_dict, key is epoch, value is a sub dict keys of sub dict is different metrics, e.g. memory, bbox_mAP value of
tools/analysis_tools/analyze_logs.py:159
↓ 1 callersMethodlog_resp_step
Computes log-responses that indicate the (logarithmic) posterior belief (sometimes called responsibilities) that a data point was gene
mmrotate/core/bbox/utils/gmm.py:216
↓ 1 callersFunctionmain
()
tools/train.py:82
↓ 1 callersFunctionmain
()
tools/test.py:101
↓ 1 callersFunctionmain
Main function of publish model.
tools/model_converters/publish_model.py:44
↓ 1 callersFunctionmain
()
tools/analysis_tools/benchmark.py:205
↓ 1 callersFunctionmain
()
tools/analysis_tools/confusion_matrix.py:233
↓ 1 callersFunctionmain
Main function of analyze logs.
tools/analysis_tools/analyze_logs.py:186
↓ 1 callersFunctionmain
()
tools/analysis_tools/get_flops.py:45
↓ 1 callersFunctionmain
Main function of image split.
tools/data/dota/split/img_split.py:534
↓ 1 callersFunctionmain
Main function of image split.
tools/data/fair/img_split.py:536
↓ 1 callersFunctionmain
()
tools/misc/browse_dataset.py:78
↓ 1 callersFunctionmain
Print config.
tools/misc/print_config.py:42
↓ 1 callersFunctionmain
(args)
demo/huge_image_demo.py:62
↓ 1 callersFunctionmain
(args)
demo/image_demo.py:28
↓ 1 callersMethodmake_res_layer
Build Reslayer.
mmrotate/models/backbones/re_resnet.py:552
↓ 1 callersFunctionmap_masks
Map masks to the huge image. Args: masks (list[np.ndarray]): masks need to be mapped. offset (np.ndarray): The offset to translat
mmrotate/core/patch/merge_results.py:32
↓ 1 callersMethodmap_roi_levels
Map rois to corresponding feature levels by scales. - scale < finest_scale * 2: level 0 - finest_scale * 2 <= scale < finest_scale *
mmrotate/models/roi_heads/roi_extractors/rotate_single_level_roi_extractor.py:68
↓ 1 callersFunctionmeasure_inference_speed
Inference speed statistics. Args: cfg (object): Test config object. checkpoint (str): Checkpoint file path. max_iter (int
tools/analysis_tools/benchmark.py:62
↓ 1 callersMethodmerge_aug_bboxes
Merge augmented detection bboxes and scores. Args: aug_bboxes (list[Tensor]): shape (n, 4*#class) aug_scores (list[Te
mmrotate/models/dense_heads/rotated_anchor_head.py:760
↓ 1 callersMethodmerge_det
Merging patch bboxes into full image. Args: results (list): Testing results of the dataset. nproc (int): number of pr
mmrotate/datasets/fair.py:245
↓ 1 callersMethodmerge_det
Merging patch bboxes into full image. Args: results (list): Testing results of the dataset. nproc (int): number of pr
mmrotate/datasets/dota.py:215
↓ 1 callersMethodmerge_det
Merging patch bboxes into full image. Args: results (list): Testing results of the dataset. nproc (int): number of pr
mmrotate/datasets/dota_1_5.py:215
↓ 1 callersFunctionmerge_results
Merge patch results via nms. Args: results (list[np.ndarray] | list[tuple]): A list of patches results. offsets (np.ndarray): Pos
mmrotate/core/patch/merge_results.py:69
↓ 1 callersFunctionmmrotate2torchserve
Converts MMRotate model (config + checkpoint) to TorchServe `.mar`. Args: config_file: In MMRotate config format.
tools/deployment/mmrotate2torchserve.py:15
↓ 1 callersMethodnorm1
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:91
↓ 1 callersMethodnorm1
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:229
↓ 1 callersMethodnorm1
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:557
↓ 1 callersMethodnorm1
nn.Module: normalization layer after the first convolution layer
mmrotate/models/backbones/resnet.py:250
↓ 1 callersMethodnorm2
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:96
↓ 1 callersMethodnorm2
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:234
↓ 1 callersMethodnorm2
nn.Module: normalization layer after the second convolution layer
mmrotate/models/backbones/resnet.py:255
↓ 1 callersMethodnorm3
Get normalizion layer's name.
mmrotate/models/backbones/re_resnet.py:239
↓ 1 callersMethodnorm3
nn.Module: normalization layer after the third convolution layer
mmrotate/models/backbones/resnet.py:260
↓ 1 callersMethodobb2distance
(self, points, distance, max_dis=None, eps=None)
mmrotate/core/bbox/coder/distance_angle_point_coder.py:69
↓ 1 callersFunctionobb2hbb_le135
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:525
↓ 1 callersFunctionobb2hbb_le90
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:553
↓ 1 callersFunctionobb2hbb_oc
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:502
↓ 1 callersFunctionobb2poly_le90
Convert oriented bounding boxes to polygons. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: polys (torch.Tensor
mmrotate/core/bbox/transforms.py:474
↓ 1 callersFunctionobb2poly_np_le135
Convert oriented bounding boxes to polygons. Args: obbs (ndarray): [x_ctr,y_ctr,w,h,angle,score] Returns: polys (ndarray): [
mmrotate/core/bbox/transforms.py:733
↓ 1 callersFunctionobb2poly_np_le90
Convert oriented bounding boxes to polygons. Args: obbs (ndarray): [x_ctr,y_ctr,w,h,angle,score] Returns: polys (ndarray): [
mmrotate/core/bbox/transforms.py:760
↓ 1 callersFunctionobb2poly_np_oc
Convert oriented bounding boxes to polygons. Args: obbs (ndarray): [x_ctr,y_ctr,w,h,angle,score] Returns: polys (ndarray): [
mmrotate/core/bbox/transforms.py:705
↓ 1 callersFunctionobb2poly_oc
Convert oriented bounding boxes to polygons. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: polys (torch.Tensor
mmrotate/core/bbox/transforms.py:421
↓ 1 callersFunctionobb2xyxy_le135
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:665
↓ 1 callersFunctionobb2xyxy_le90
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:685
↓ 1 callersFunctionobb2xyxy_oc
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hb
mmrotate/core/bbox/transforms.py:637
↓ 1 callersFunctionparse_args
()
tools/train.py:24
↓ 1 callersFunctionparse_args
Parse parameters.
tools/test.py:23
↓ 1 callersFunctionparse_args
Parse parameters.
tools/model_converters/publish_model.py:8
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/benchmark.py:18
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/confusion_matrix.py:17
↓ 1 callersFunctionparse_args
Parse parameters.
tools/analysis_tools/analyze_logs.py:148
↓ 1 callersFunctionparse_args
()
tools/analysis_tools/get_flops.py:16
← previousnext →201–300 of 746, ranked by callers