MCPcopy Create free account

hub / github.com/Sense-X/Co-DETR / functions

Functions2,920 in github.com/Sense-X/Co-DETR

↓ 2 callersFunctionreorder_cls_channel
(val, num_classes=81)
tools/model_converters/upgrade_model_version.py:46
↓ 2 callersMethodrescale
see :func:`BaseInstanceMasks.rescale`
mmdet/core/mask/structures.py:635
↓ 2 callersMethodrfp_forward
The forward function that also takes the RFP features as input.
mmdet/models/backbones/detectors_resnet.py:73
↓ 2 callersMethodroi_rescale
Scale RoI coordinates by scale factor. Args: rois (torch.Tensor): RoI (Region of Interest), shape (n, 5) scale_factor
mmdet/models/roi_heads/roi_extractors/base_roi_extractor.py:62
↓ 2 callersFunctionrotate_half
(x)
mmdet/models/backbones/vit.py:59
↓ 2 callersMethodsanitize_coordinates
Sanitizes the input coordinates so that x1 < x2, x1 != x2, x1 >= 0, and x2 <= image_size. Also converts from relative to absolute coor
mmdet/models/dense_heads/yolact_head.py:910
↓ 2 callersFunctionsave_anns
(name, images, annotations)
tools/misc/split_coco.py:53
↓ 2 callersMethodsave_result
(self, anchors, path=None)
tools/analysis_tools/optimize_anchors.py:140
↓ 2 callersMethodsemantic_postprocess
Semantic segmengation postprocess. Args: mask_cls (Tensor): Classfication outputs of shape (num_queries, cls_out_
mmdet/models/seg_heads/panoptic_fusion_heads/maskformer_fusion_head.py:94
↓ 2 callersFunctionset_random_seed
Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for
mmdet/apis/train.py:53
↓ 2 callersMethodsimple_test_bboxes
Test det bboxes without test-time augmentation. Args: feats (tuple[torch.Tensor]): Multi-level features from the
projects/models/co_deformable_detr_head.py:1079
↓ 2 callersMethodsimple_test_mask
Simple test for mask head without augmentation.
mmdet/models/roi_heads/test_mixins.py:224
↓ 2 callersMethodsimple_test_mask
Obtain mask prediction without augmentation.
projects/models/co_detr.py:436
↓ 2 callersFunctionskip_pipeline_steps
(config)
tools/misc/browse_dataset.py:53
↓ 2 callersFunctionsmooth_l1_loss
Smooth L1 loss. Args: pred (torch.Tensor): The prediction. target (torch.Tensor): The learning target of the prediction.
mmdet/models/losses/smooth_l1_loss.py:12
↓ 2 callersFunctionssd_config
SSD Head Config.
tests/test_onnx/test_head.py:382
↓ 2 callersMethodto_bitmap
convert polygon masks to bitmap masks.
mmdet/core/mask/structures.py:860
↓ 2 callersMethodto_tensor
See :func:`BaseInstanceMasks.to_tensor`.
mmdet/core/mask/structures.py:910
↓ 2 callersFunctiontpfp_default
Check if detected bboxes are true positive or false positive. Args: det_bbox (ndarray): Detected bboxes of this image, of shape (m, 5).
mmdet/core/evaluation/mean_ap.py:169
↓ 2 callersFunctiontpfp_imagenet
Check if detected bboxes are true positive or false positive. Args: det_bbox (ndarray): Detected bboxes of this image, of shape (m, 5).
mmdet/core/evaluation/mean_ap.py:60
↓ 2 callersFunctionvoc_classes
()
mmdet/core/evaluation/class_names.py:20
↓ 2 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
projects/models/swin_transformer.py:42
↓ 2 callersMethodwindow_partition
Args: x: (B, H, W, C) Returns: windows: (num_windows*B, window_size, window_size, C)
mmdet/models/backbones/swin.py:272
↓ 2 callersFunctionyolo_config
YoloV3 Head Config.
tests/test_onnx/test_head.py:176
↓ 1 callersMethodGaussianMixture
(self, *args, **kwargs)
tests/test_models/test_dense_heads/test_lad_head.py:15
↓ 1 callersMethod__init__
(self, strides, ratios, scales=None, base_
mmdet/core/anchor/anchor_generator.py:61
↓ 1 callersMethod__init__
(self, *args, dynamic_intervals=None, **kwargs)
mmdet/core/evaluation/eval_hooks.py:26
↓ 1 callersMethod__init__
(self, annotation_file=None)
mmdet/datasets/coco_panoptic.py:37
↓ 1 callersMethod__init__
(self, ann_file, label_file='', image_level_ann_file='',
mmdet/datasets/openimages.py:63
↓ 1 callersMethod__init__
(self, num_max_proposals=None)
mmdet/datasets/pipelines/loading.py:536
↓ 1 callersMethod__init__
(self, num_scales, in_channels, out_channels,
mmdet/models/necks/yolo_neck.py:92
↓ 1 callersMethod__init__
L2 normalization layer. Args: n_dims (int): Number of dimensions to be normalized scale (float, optional): Defaults t
mmdet/models/necks/ssd_neck.py:108
↓ 1 callersMethod__init__
(self, rfp_steps, rfp_backbone, aspp_out_channels,
mmdet/models/necks/rfp.py:78
↓ 1 callersMethod__init__
(self, in_channels, out_channels, num_outs,
mmdet/models/necks/sfp.py:43
↓ 1 callersMethod__init__
(self, in_channels, mid_channels, dilation,
mmdet/models/necks/dilated_encoder.py:24
↓ 1 callersMethod__init__
(self, depth, with_last_pool=False, ceil_mode=True,
mmdet/models/backbones/ssd_vgg.py:50
↓ 1 callersMethod__init__
(self, groups=1, base_width=4, **kwargs)
mmdet/models/backbones/resnext.py:143
↓ 1 callersMethod__init__
(self, depth=53, out_indices=(3, 4, 5), frozen_stages=-1,
mmdet/models/backbones/darknet.py:101
↓ 1 callersMethod__init__
(self, extra, in_channels=3, conv_cfg=None,
mmdet/models/backbones/hrnet.py:281
↓ 1 callersMethod__init__
Bottleneck block for ResNeXt. If style is "pytorch", the stride-two layer is the 3x3 conv layer, if it is "caffe", the stride-two lay
mmdet/models/backbones/detectors_resnext.py:14
↓ 1 callersMethod__init__
(self, in_channels, out_channels, mid_channels,
mmdet/models/backbones/efficientnet.py:41
↓ 1 callersMethod__init__
(self, downsample_times=5, num_stacks=2, stage_channels=(25
mmdet/models/backbones/hourglass.py:131
↓ 1 callersMethod__init__
(self, *args, mask_feature_head, dynamic_conv_size=1,
mmdet/models/dense_heads/solov2_head.py:170
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size=3,
mmdet/models/dense_heads/guided_anchor_head.py:31
↓ 1 callersMethod__init__
(self, force_topk=False, topk=9, num_classes=80,
mmdet/models/dense_heads/autoassign_head.py:37
↓ 1 callersMethod__init__
(self, num_classes, in_channels, num_dcn=0,
mmdet/models/dense_heads/tood_head.py:113
↓ 1 callersMethod__init__
(self, num_classes, in_channels, base_edge_list=(16, 32, 64
mmdet/models/dense_heads/fovea_head.py:55
↓ 1 callersMethod__init__
(self, num_classes, in_channels, stacked_convs=4,
mmdet/models/dense_heads/gfl_head.py:88
↓ 1 callersMethod__init__
(self, num_classes, in_channels, num_feat_levels=2,
mmdet/models/dense_heads/corner_head.py:118
↓ 1 callersMethod__init__
(self, channels, ratio=4, conv_cfg=None, a
mmdet/models/utils/se_layer.py:88
↓ 1 callersMethod__init__
(self, num_feats, temperature=10000, normalize=False,
mmdet/models/utils/positional_encoding.py:36
↓ 1 callersMethod__init__
(self, in_channels, out_channels, expand_ratio=0.5,
mmdet/models/utils/csp_layer.py:96
↓ 1 callersMethod__init__
(self, *args, tempearture=20, power=1.0, eps=1e-6, **kwargs)
mmdet/models/utils/normed_predictor.py:21
↓ 1 callersMethod__init__
(self, block, inplanes, planes, num_blocks
mmdet/models/utils/res_layer.py:26
↓ 1 callersMethod__init__
(self, in_channels, out_channels, conv_cfg=None,
mmdet/models/roi_heads/bbox_heads/double_bbox_head.py:26
↓ 1 callersMethod__init__
(self, in_channels, feat_channels, out_channels,
mmdet/models/plugins/pixel_decoder.py:34
↓ 1 callersMethod__init__
(self, use_sigmoid=True, beta=2.0, reduction='mean',
mmdet/models/losses/gfocal_loss.py:146
↓ 1 callersMethod__init__
(self, reduction='mean', loss_weight=1.0)
mmdet/models/losses/smooth_l1_loss.py:117
↓ 1 callersMethod__init__
(self, bins=10, momentum=0, use_sigmoid=True,
mmdet/models/losses/ghm_loss.py:39
↓ 1 callersMethod__init__
CrossEntropyLoss. Args: use_sigmoid (bool, optional): Whether the prediction uses sigmoid of softmax. Defaults to
mmdet/models/losses/cross_entropy_loss.py:295
↓ 1 callersMethod__init__
(self, num_queries, hidden_dim, num_classes,
projects/models/query_denoising.py:10
↓ 1 callersMethod__repr__
str: the string of the module
mmdet/utils/util_mixins.py:87
↓ 1 callersMethod__setattr__
(self, name, val)
mmdet/core/data_structures/general_data.py:195
↓ 1 callersMethod__setattr__
Set attribute, i.e. self.name = value This reloading prevent the teacher model from being registered as a nn.Module. The teacher modu
mmdet/models/detectors/kd_one_stage.py:92
↓ 1 callersMethod_add_conv_branch
Add the fc branch which consists of a sequential of conv layers.
mmdet/models/roi_heads/bbox_heads/double_bbox_head.py:136
↓ 1 callersMethod_add_default_meta_keys
Add default meta keys. We set default meta keys including `pad_shape`, `scale_factor` and `img_norm_cfg` to avoid the case where no `
mmdet/datasets/pipelines/formatting.py:248
↓ 1 callersMethod_add_fc_branch
Add the fc branch which consists of a sequential of fc layers.
mmdet/models/roi_heads/bbox_heads/double_bbox_head.py:148
↓ 1 callersMethod_add_ground_truth
(self, runner)
mmdet/core/hook/wandblogger_hook.py:324
↓ 1 callersFunction_adjust_brightness
(img, factor)
tests/test_data/test_pipelines/test_transform/test_img_augment.py:81
↓ 1 callersMethod_adjust_brightness_img
Adjust the brightness of image.
mmdet/datasets/pipelines/auto_augment.py:821
↓ 1 callersMethod_adjust_color_img
Apply Color transformation to image.
mmdet/datasets/pipelines/auto_augment.py:733
↓ 1 callersMethod_adjust_contrast_img
Adjust the image contrast.
mmdet/datasets/pipelines/auto_augment.py:870
↓ 1 callersFunction_allreduce_coalesced
(tensors, world_size, bucket_size_mb=-1)
mmdet/core/utils/dist_utils.py:15
↓ 1 callersMethod_bbox_decode
(self, points, bbox_pred, stride, max_shape)
mmdet/models/dense_heads/reppoints_head.py:756
↓ 1 callersMethod_bbox_decode
(self, priors, bbox_pred, base_len, max_shape)
mmdet/models/dense_heads/fovea_head.py:358
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet/models/roi_heads/pisa_roi_head.py:100
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet/models/roi_heads/standard_roi_head.py:131
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet/models/roi_heads/scnet_roi_head.py:150
↓ 1 callersMethod_bbox_forward_train
(self, x, sampling_results, gt_bboxes, gt_labels, img_metas)
mmdet/models/roi_heads/dynamic_roi_head.py:109
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet/models/roi_heads/htc_roi_head.py:86
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
mmdet/models/roi_heads/cascade_roi_head.py:143
↓ 1 callersMethod_bbox_forward_train
Run forward function and calculate loss for box head in training.
projects/models/co_roi_head.py:155
↓ 1 callersMethod_bbox_post_process
bbox post-processing method. The boxes would be rescaled to the original image scale and do the nms operation. Usually `with_nms` is
mmdet/models/dense_heads/base_dense_head.py:226
↓ 1 callersMethod_bbox_post_process
bbox post-processing method. Do the nms operation for bboxes in same level. Args: mlvl_scores (list[Tensor]): Box scores
mmdet/models/dense_heads/rpn_head.py:189
↓ 1 callersMethod_bboxes_nms
(self, bboxes, labels, cfg)
mmdet/models/dense_heads/centernet_head.py:402
↓ 1 callersMethod_bboxes_nms
(self, bboxes, labels, cfg)
mmdet/models/dense_heads/corner_head.py:777
↓ 1 callersMethod_binary_cross_entropy
Args: cls_pred (Tensor): The prediction with shape (num_query, 1, *) or (num_query, *). gt_labels (Te
mmdet/core/bbox/match_costs/match_cost.py:320
↓ 1 callersFunction_build_demo_runner_without_hook
(runner_type='EpochBasedRunner', max_epochs=1,
tests/test_utils/test_hook.py:22
↓ 1 callersMethod_build_predictor
Initialize predictor layers of a single level head.
mmdet/models/dense_heads/yolox_head.py:169
↓ 1 callersFunction_check_bbox_head
(bbox_cfg, bbox_head)
tests/test_runtime/test_config.py:168
↓ 1 callersFunction_check_bbox_mask
(results, results_translated, offset,
tests/test_data/test_pipelines/test_transform/test_translate.py:201
↓ 1 callersMethod_check_branches
(self, num_branches, num_blocks, in_channels, num_channels)
mmdet/models/backbones/hrnet.py:49
↓ 1 callersFunction_check_keys
(results, results_translated)
tests/test_data/test_pipelines/test_transform/test_translate.py:13
↓ 1 callersFunction_check_mask_head
(mask_cfg, mask_head)
tests/test_runtime/test_config.py:145
↓ 1 callersMethod_compute_gamma
Compute the value of gamma according to paper. gamma is the parameter of bernoulli distribution, which controls the number of features
mmdet/models/plugins/dropblock.py:62
↓ 1 callersFunction_concat_dataset
(cfg, default_args=None)
mmdet/datasets/builder.py:32
↓ 1 callersFunction_construct_ann_info
(h=427, w=640, c=3)
tests/test_data/test_pipelines/test_transform/test_translate.py:40
↓ 1 callersFunction_construct_img
(results)
tests/test_data/test_pipelines/test_transform/test_translate.py:30
↓ 1 callersFunction_construct_semantic_seg
(results)
tests/test_data/test_pipelines/test_transform/test_translate.py:128
↓ 1 callersFunction_construct_toy_data
()
tests/test_data/test_pipelines/test_transform/test_transform.py:265
← previousnext →501–600 of 2,920, ranked by callers