MCPcopy Create free account

hub / github.com/SJTU-Thinklab-Det/r3det-pytorch / functions

Functions313 in github.com/SJTU-Thinklab-Det/r3det-pytorch

Method__repr__
str: a string describing the module
r3det/core/bbox/iou_calculators/rotate_iou2d_calculator.py:196
Method__repr__
(self)
r3det/core/anchor/ranchor_generator.py:59
Method__repr__
(self)
r3det/datasets/pipelines/rtransforms.py:228
Method_filter_imgs
Filter images without ground truths.
r3det/datasets/dota1.py:147
Method_get_targets_single
Compute regression and classification targets for anchors in a single image. Args: flat_anchors (torch.Tensor): Multi-lev
r3det/models/dense_heads/rotate_anchor_head.py:172
Method_init_layers
Initialize layers of the head.
r3det/models/dense_heads/rotate_retina_head.py:51
Function_load_dota_single
(imgfile, img_dir, ann_dir)
tools/split/img_split.py:373
Function_merge_func
Merging patch bboxes into full image. Params: CLASSES (list): Label category. iou_thr (float): Threshold of IoU. version
r3det/datasets/dota1.py:632
Method_resize_bboxes
Resize bounding boxes with ``results['scale_factor']``.
r3det/datasets/pipelines/rtransforms.py:30
Method_set_group_flag
Set flag according to image aspect ratio. All set to 0.
r3det/datasets/dota1.py:155
Functionadd_mim_extension
Add extra files that are required to support MIM into the package. These files will be added by creating a symlink to the originals if the pa
setup.py:128
Methodaug_test
Test function with test time augmentation.
r3det/models/detectors/r3det.py:145
Methodbackward
Backward function.
r3det/ops/convex/convex_wrapper.py:17
Methodbackward
Backward function.
r3det/ops/fr/feature_refine_module.py:28
Methodbbox_flip
Flip bboxes horizontally or vertically. Args: bboxes(ndarray): shape (..., 5*k) img_shape(tuple): (height, width)
r3det/datasets/pipelines/rtransforms.py:59
Functionbbox_map_eval
Evaluate mAP of single image det result. Args: det_result (list[list]): [[cls1_det, cls2_det, ...], ...]. The outer list indi
tools/analysis_tools/analyze_results.py:13
Functionbox_iou_rotated_cpu_kernel
r3det/ops/box_iou_rotated/src/box_iou_rotated_cpu.cpp:7
Functioncal_train_time
(log_dicts, args)
tools/analysis_tools/analyze_logs.py:15
Functionconvex_hull_graham
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:160
Functionconvex_hull_graham
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:158
Functionconvex_hull_graham
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:159
Functionconvex_sort
r3det/ops/convex/src/convex_ext.cpp:13
Functionconvex_sort_cpu_kernel
r3det/ops/convex/src/convex_cpu.cpp:9
Functioncross_2d
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:52
Functioncross_2d
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:50
Functioncross_2d
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:51
Methoddecode
Apply transformation `pred_bboxes` to `boxes`. Args: bboxes (torch.Tensor): Basic boxes. Shape (B, N, 4) or (N, 4) pre
r3det/core/bbox/coder/delta_xywha_hbbox_coder.py:69
Functiondot_2d
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:46
Functiondot_2d
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:44
Functiondot_2d
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:46
Methodencode
Get box regression transformation deltas that can be used to transform the ``bboxes`` into the ``gt_bboxes``. Args: bboxes
r3det/core/bbox/coder/delta_xywha_hbbox_coder.py:46
Functionfeature_refine_backward
r3det/ops/fr/src/feature_refine_cuda.cpp:44
Functionfeature_refine_forward
r3det/ops/fr/src/feature_refine_cuda.cpp:24
Methodforward
Forward function.
r3det/ops/convex/convex_wrapper.py:10
Methodforward
Forward function.
r3det/ops/fr/feature_refine_module.py:14
Methodforward
Args: x (list[Tensor]): feature maps of multiple scales best_rbboxes (list[list[Tensor]]):
r3det/ops/fr/feature_refine_module.py:108
Methodforward
Forward features from the upstream network. Args: feats (tuple[Tensor]): Features from the upstream network, each is
r3det/models/dense_heads/rotate_anchor_head.py:123
Methodforward_dummy
Used for computing network flops. See `mmedetection/tools/get_flops.py`
r3det/models/detectors/r3det.py:59
Methodforward_single
Forward feature of a single scale level. Args: x (torch.Tensor): Features of a single scale level. Returns:
r3det/models/dense_heads/rotate_retina_head.py:94
Methodforward_single
Forward feature of a single scale level. Args: x (torch.Tensor): Features of a single scale level. Returns:
r3det/models/dense_heads/rotate_anchor_head.py:106
Methodforward_train
Forward function.
r3det/models/detectors/r3det.py:75
Methodget_anchors
Get anchors according to feature map sizes. Args: featmap_sizes (list[tuple]): Multi-level feature map sizes. img_met
r3det/models/dense_heads/rotate_retina_refine_head.py:99
Methodget_bboxes
Transform network output for a batch into labeled boxes.s. Args: cls_scores (list[Tensor]): Box scores for each scale level
r3det/models/dense_heads/rotate_retina_refine_head.py:148
Functionget_best_begin_point_single
Get the best begin point of the single polygon. Args: coordinate (List): [x1, y1, x2, y2, x3, y3, x4, y4, score] Returns: reo
r3det/core/bbox/rtransforms.py:742
Functionget_intersection_points
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:79
Functionget_intersection_points
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:77
Functionget_intersection_points
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:78
Functionget_rotated_vertices
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:58
Functionget_rotated_vertices
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:56
Functionget_rotated_vertices
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:56
Functionhbb2obb
Convert horizontal bounding boxes to oriented bounding boxes. Args: hbbs (torch.Tensor): [x_lt,y_lt,x_rb,y_rb] version (Str): ang
r3det/core/bbox/rtransforms.py:170
Methodinit_weights
Initialize weights of the head.
r3det/models/dense_heads/rotate_anchor_head.py:101
Methodis_rotate
Randomly decide whether to rotate.
r3det/datasets/pipelines/rtransforms.py:117
Methodload_annotations
Params: ann_folder: folder that contains DOTA v1 annotations txt files
r3det/datasets/dota1.py:56
Methodloss
Loss function of RRetinaRefineHead.
r3det/models/dense_heads/rotate_retina_refine_head.py:128
Methodloss_single
Compute loss of a single scale level. Args: cls_score (torch.Tensor): Box scores for each scale level Has shape (
r3det/models/dense_heads/rotate_anchor_head.py:381
Functionmat_iou_iof
r3det/ops/rbbox_geo/src/rbbox_geo_cuda.cpp:13
Functionnms_poly
r3det/ops/nms_rotated/src/nms_rotated_ext.cpp:38
Functionnms_rotated
r3det/ops/nms_rotated/src/nms_rotated_ext.cpp:24
Functionnms_rotated_cpu_kernel
r3det/ops/nms_rotated/src/nms_rotated_cpu.cpp:9
Functionnms_rotated_cpu_kernel
r3det/ops/ml_nms_rotated/src/nms_rotated_cpu.cpp:7
Functionnmsr_cpu_kernel
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:224
Methodnum_base_anchors
list[int]: total number of base anchors in a feature grid
r3det/core/anchor/ranchor_generator.py:51
Functionobb2hbb_v2_alpha
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] Returns: hbb
r3det/core/bbox/rtransforms.py:492
Functionobb2poly
Convert oriented bounding boxes to polygons. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] version (Str): angle representati
r3det/core/bbox/rtransforms.py:110
Functionobb2xyxy
Convert oriented bounding boxes to horizontal bounding boxes. Args: obbs (torch.Tensor): [x_ctr,y_ctr,w,h,angle] version (Str): a
r3det/core/bbox/rtransforms.py:150
Functionoperator*
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:59
Functionoperator*
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:58
Methodoperator*
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:40
Methodoperator*
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:38
Methodoperator*
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:40
Methodoperator+
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:29
Methodoperator+
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:27
Methodoperator+
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:37
Methodoperator+
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:29
Methodoperator+
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:36
Methodoperator+=
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:42
Methodoperator+=
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:41
Methodoperator-
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:37
Methodoperator-
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:35
Methodoperator-
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:25
Methodoperator-
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:37
Methodoperator-
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:24
Methodoperator-=
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:30
Methodoperator-=
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:29
Methodoperator<
r3det/ops/rnms/src/rcpu/rnms_cpu.cpp:49
Methodoperator<
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:48
Functionplot_curve
(log_dicts, args)
tools/analysis_tools/analyze_logs.py:38
Functionpoly2obb
Convert polygons to oriented bounding boxes. (r3det version) Args: polys (torch.Tensor): [x0,y0,x1,y1,x2,y2,x3,y3] version (S
r3det/core/bbox/rtransforms.py:49
Functionpoly_nms_cpu_kernel
r3det/ops/nms_rotated/src/poly_nms_cpu.cpp:4
Functionpolygon_area
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:294
Functionpolygon_area
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:292
Functionpolygon_area
r3det/ops/ml_nms_rotated/src/box_iou_rotated_utils.h:274
Functionpolygon_iou
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:272
Functionpolygon_iou_kernel
r3det/ops/polygon_geo/src/polygon_geo_cpu.cpp:228
Functionrbbox2roi
Convert a list of bboxes to roi format. Args: bbox_list (list[Tensor]): a list of bboxes corresponding to a batch of images.
r3det/core/bbox/rtransforms.py:28
Methodrefine_bboxes
Refine predicted bounding boxes at each position of the feature maps. This method will be used in R3Det in refinement stages. Args:
r3det/models/dense_heads/rotate_retina_refine_head.py:57
Functionrnms
r3det/ops/rnms/src/rnms_ext.cpp:11
Functionrotated_boxes_intersection
r3det/ops/nms_rotated/src/box_iou_rotated_utils.h:308
Functionrotated_boxes_intersection
r3det/ops/box_iou_rotated/src/box_iou_rotated_utils.h:306
← previousnext →201–300 of 313, ranked by callers