Get matrix of the corresponding relationship between the parent class and the child class. Args: hierarchy_map (dict): In

CopyPaste transform function. Args: dst_results (dict): Result dict of the destination image. src_results (dict): Res

Function to randomly crop images, bounding boxes, masks, semantic segmentation maps. Args: results (dict): Result dict fr

delete plugins for block if exist. Args: plugin_names (list[str]): List of plugins name to delete.

Create random test data for the ``mmdet.models.bbox_heads.bbox_head.BBoxHead.refine_boxes`` method.

Convert detection results to COCO json style.

Evaluation in Cityscapes protocol. Args: results (list): Testing results of the dataset. txtfile_prefix (str | None):

Expand onehot labels to match the size of prediction.

Filter out bboxes too small after Mosaic.

Compute candidate boxes which include following 5 things: bbox1 before augment, bbox2 after augment, min_bbox_size (pixels), min_area

Filter images too small.

Filter bboxes and corresponding masks too small after shear augmentation.

Filter bboxes and corresponding masks too small after rotate augmentation.

Filter bboxes and masks too small or translated out of image.

Args: cls_pred (Tensor): Predicted classification logits, shape (num_query, num_class). gt_labels (Te

Forward function for classification and regression parts.

Forward function for shared part.

Transform outputs of a single image into bbox predictions. Args: cls_score_list (list[Tensor]): Box scores from all scale

Transform outputs of a single image into bbox results. Args: center_heatmap_pred (Tensor): Center heatmap for current level with

Similar to func:``AnchorHead._get_bboxes_single``, but additionally processes coeff_preds_list and uses fast NMS instead of traditional

Transform outputs of a single image into bbox predictions. Args: cls_scores (list[Tensor]): Box scores from all scale

Transform outputs from the last decoder layer into bbox predictions for each image. Args: cls_score (Tensor): Box score l

Transform outputs from the last decoder layer into bbox predictions for each image. Args: cls_score (Tensor): Box score l

Get different colors for each masks. Get different colors for each masks by adding a bias color to the base category color. Args:

Find the predefined detector config directory.

Find the predefined detector config directory.

Find the predefined detector config directory.

Load a configuration as a python module.

Randomly generates the absolute crop size based on `crop_type` and `image_size`. Args: image_size (tuple): (h, w).

Get guided anchors and loc masks for a single level. Args: square (tensor): Squares of a single level. shape_pred (te

Convert gt bboxes to center offset and log width height.

Get points of a single scale level. This function will be deprecated soon.

Get repeat factor for each images in the dataset. Args: dataset (:obj:`CustomDataset`): The dataset repeat_thr (float

Get processed mask related results of single image. Args: kernel_preds (Tensor): Dynamic kernel prediction of all points

Get processed mask related results of single image. Args: cls_scores (Tensor): Classification score of all points

Get processed mask related results of single image. Args: cls_scores (Tensor): Classification score of all points

Compute targets for predictions of single image. Args: gt_bboxes (Tensor): Ground truth bbox of each instance, sh

Equalizes the histogram of one image.

Slice the infinite indices by rank.

Slice the infinite indices by rank.

Infinitely yield a sequence of indices.

Infinitely yield a sequence of indices.

Initialize centripetal layers. Including feature adaption deform convs (feat_adaption), deform offset prediction convs (dcn_off), gui

Initialize corner embedding layers. Only include corner embedding branch with two parts: prefix `tl_` for top-left and `br_` for bott

Initialize corner keypoint layers. Including corner heatmap branch and corner offset branch. Each branch has two parts: prefix `tl_`

Initialize the W&B Tables for validation data.

Initialize layers of the head.

Initialize a sparse set of proposal boxes and proposal features.

Initialize layers of the head.

Initialize layers of the head.

A helper function to take a config setting and turn it into a network.

Initialize layers of the head.

Initialize layers for CornerHead. Including two parts: corner keypoint layers and corner embedding layers

Initialize layers of the transformer head.

Initialize layers of the DeformableDetrTransformer.

Initialize predictor layers of the head.

Lay instance masks to a result map. Args: bboxes: The bboxes results, (K, 4). labels: The labels of bboxes, (K, ).

load checkpoints.

Private function to load mask annotations. Args: results (dict): Result dict from :obj:`mmdet.CustomDataset`. Returns: