Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WXinlong/SOLO
/ functions
Functions
1,004 in github.com/WXinlong/SOLO
⨍
Functions
1,004
◇
Types & classes
169
↓ 1 callers
Method
_make_branches
(self, num_branches, block, num_blocks, num_channels)
mmdet/models/backbones/hrnet.py:104
↓ 1 callers
Method
_make_extra_layers
(self, outplanes)
mmdet/models/backbones/ssd_vgg.py:112
↓ 1 callers
Method
_make_fuse_layers
(self)
mmdet/models/backbones/hrnet.py:113
↓ 1 callers
Method
_make_layer
(self, block, inplanes, planes, blocks, stride=1)
mmdet/models/backbones/hrnet.py:398
↓ 1 callers
Method
_make_one_branch
(self, branch_index, block, num_blo
mmdet/models/backbones/hrnet.py:60
↓ 1 callers
Method
_make_stem_layer
(self, in_channels)
mmdet/models/backbones/resnet.py:443
↓ 1 callers
Method
_mask_forward_train
(self, stage, x, sampling_
mmdet/models/detectors/htc.py:73
↓ 1 callers
Function
_minimal_ext_cmd
(cmd)
setup.py:37
↓ 1 callers
Function
_non_dist_train
(model, dataset, cfg, validate=False,
mmdet/apis/train.py:253
↓ 1 callers
Method
_output_size
(input, weight, padding, dilation, stride)
mmdet/ops/dcn/deform_conv.py:98
↓ 1 callers
Method
_pad_img
(self, results)
mmdet/datasets/pipelines/transforms.py:285
↓ 1 callers
Method
_pad_masks
(self, results)
mmdet/datasets/pipelines/transforms.py:296
↓ 1 callers
Method
_pad_seg
(self, results)
mmdet/datasets/pipelines/transforms.py:308
↓ 1 callers
Method
_parse_ann_info
Parse bbox and mask annotation. Args: ann_info (list[dict]): Annotation info of an image. with_mask (bool): Whether t
mmdet/datasets/coco.py:58
↓ 1 callers
Method
_parse_anns
(self, results, anns, img)
mmdet/datasets/pipelines/instaboost.py:56
↓ 1 callers
Method
_poly2mask
(self, mask_ann, img_h, img_w)
mmdet/datasets/pipelines/loading.py:67
↓ 1 callers
Method
_rand_another
(self, idx)
mmdet/datasets/custom.py:123
↓ 1 callers
Method
_random_jitter
Ramdom jitter positive proposals for training.
mmdet/models/detectors/grid_rcnn.py:58
↓ 1 callers
Method
_random_scale
(self, results)
mmdet/datasets/pipelines/transforms.py:104
↓ 1 callers
Function
_recalls
(all_ious, proposal_nums, thrs)
mmdet/core/evaluation/recall.py:7
↓ 1 callers
Method
_register_module
Register a module. Args: module (:obj:`nn.Module`): Module to be registered.
mmdet/utils/registry.py:29
↓ 1 callers
Method
_resize_bboxes
(self, results)
mmdet/datasets/pipelines/transforms.py:135
↓ 1 callers
Method
_resize_img
(self, results)
mmdet/datasets/pipelines/transforms.py:120
↓ 1 callers
Method
_resize_masks
(self, results)
mmdet/datasets/pipelines/transforms.py:143
↓ 1 callers
Method
_resize_seg
(self, results)
mmdet/datasets/pipelines/transforms.py:161
↓ 1 callers
Method
_sample_neg
(self, assign_result, num_expected, **kwargs)
mmdet/core/bbox/samplers/base_sampler.py:28
↓ 1 callers
Method
_sample_pos
(self, assign_result, num_expected, **kwargs)
mmdet/core/bbox/samplers/base_sampler.py:24
↓ 1 callers
Method
_set_group_flag
Set flag according to image aspect ratio. Images with aspect ratio greater than 1 will be set as group 1, otherwise group 0.
mmdet/datasets/custom.py:111
↓ 1 callers
Function
add_batch_counter_hook_function
(module)
mmdet/utils/flops_counter.py:401
↓ 1 callers
Function
add_batch_counter_variables_or_reset
(module)
mmdet/utils/flops_counter.py:397
↓ 1 callers
Function
add_flops_counting_methods
(net_main_module)
mmdet/utils/flops_counter.py:157
↓ 1 callers
Method
add_gt_
(self, gt_labels)
mmdet/core/bbox/assigners/assign_result.py:183
↓ 1 callers
Function
add_plot_parser
(subparsers)
tools/analyze_logs.py:94
↓ 1 callers
Function
add_time_parser
(subparsers)
tools/analyze_logs.py:122
↓ 1 callers
Method
aforward_test
(self, *, img, img_meta, **kwargs)
mmdet/models/detectors/base.py:82
↓ 1 callers
Function
analyze_results
(res_file, ann_file, res_types, out_dir)
tools/coco_error_analysis.py:104
↓ 1 callers
Method
apredict
(self, img)
tests/test_async.py:53
↓ 1 callers
Method
async_simple_test
(self, img, img_meta, **kwargs)
mmdet/models/detectors/base.py:67
↓ 1 callers
Method
async_test_bboxes
Async test only det bboxes without augmentation.
mmdet/models/detectors/test_mixins.py:64
↓ 1 callers
Method
async_test_mask
(self, x, img_meta,
mmdet/models/detectors/test_mixins.py:164
↓ 1 callers
Method
async_test_rpn
(self, x, img_meta, rpn_test_cfg)
mmdet/models/detectors/test_mixins.py:19
↓ 1 callers
Method
atss_target
almost the same with anchor_target, with a little modification, here we need return the anchor
mmdet/models/anchor_heads/atss_head.py:349
↓ 1 callers
Method
aug_test
(self, imgs, img_metas, **kwargs)
mmdet/models/detectors/base.py:75
↓ 1 callers
Method
aug_test
(self, imgs, img_metas, rescale=False)
mmdet/models/detectors/rpn.py:71
↓ 1 callers
Method
aug_test_bboxes
(self, feats, img_metas, proposal_list, rcnn_test_cfg)
mmdet/models/detectors/test_mixins.py:122
↓ 1 callers
Method
aug_test_mask
(self, feats, img_metas, det_bboxes, det_labels)
mmdet/models/detectors/test_mixins.py:234
↓ 1 callers
Function
average_precision
Calculate average precision (for single or multiple scales). Args: recalls (ndarray): shape (num_scales, num_dets) or (num_dets, )
mmdet/core/evaluation/mean_ap.py:12
↓ 1 callers
Function
balanced_l1_loss
(pred, target, beta=1.0, alpha=0.5,
mmdet/models/losses/balanced_l1_loss.py:10
↓ 1 callers
Function
bbox_target
(pos_bboxes_list, neg_bboxes_list, pos_gt_bboxes_list, pos_gt_
mmdet/core/bbox/bbox_target.py:7
↓ 1 callers
Function
bounded_iou_loss
Improving Object Localization with Fitness NMS and Bounded IoU Loss, https://arxiv.org/abs/1711.00164. Args: pred (tensor): Predicted
mmdet/models/losses/iou_loss.py:31
↓ 1 callers
Method
build_roi_layers
(self, layer_cfg, featmap_strides)
mmdet/models/roi_extractors/single_level.py:45
↓ 1 callers
Method
calc_sub_regions
Compute point specific representation regions. See Grid R-CNN Plus (https://arxiv.org/abs/1906.05688) for details.
mmdet/models/mask_heads/grid_head.py:191
↓ 1 callers
Function
center_of_mass
(bitmasks)
mmdet/models/anchor_heads/decoupled_solo_light_head.py:14
↓ 1 callers
Function
center_of_mass
(bitmasks)
mmdet/models/anchor_heads/decoupled_solo_head.py:14
↓ 1 callers
Function
center_of_mass
(bitmasks)
mmdet/models/anchor_heads/solov2_head.py:14
↓ 1 callers
Function
center_of_mass
(bitmasks)
mmdet/models/anchor_heads/solov2_light_head.py:14
↓ 1 callers
Function
center_of_mass
(bitmasks)
mmdet/models/anchor_heads/solo_head.py:14
↓ 1 callers
Method
centerness_target
(self, anchors, bbox_targets)
mmdet/models/anchor_heads/atss_head.py:236
↓ 1 callers
Method
centerness_target
(self, pos_bbox_targets)
mmdet/models/anchor_heads/fcos_head.py:401
↓ 1 callers
Method
centers_to_bboxes
Get bboxes according to center points. Only used in MaxIOUAssigner.
mmdet/models/anchor_heads/reppoints_head.py:333
↓ 1 callers
Function
coco_eval_with_return
(result_files, result_types, coco,
tools/test_robustness.py:26
↓ 1 callers
Function
collect_env
()
tools/collect_env.py:15
↓ 1 callers
Function
collect_results
(result_part, size, tmpdir=None)
tools/test_robustness.py:145
↓ 1 callers
Function
collect_results
(result_part, size, tmpdir=None)
tools/test_ins.py:89
↓ 1 callers
Function
collect_results
(result_part, size, tmpdir=None)
tools/test_ins_vis.py:130
↓ 1 callers
Function
collect_results_cpu
(result_part, size, tmpdir=None)
tools/test.py:81
↓ 1 callers
Function
collect_results_gpu
(result_part, size)
tools/test.py:123
↓ 1 callers
Function
conv_ws_2d
(input, weight, bias=None, stride=1, padding=0,
mmdet/models/utils/conv_ws.py:5
↓ 1 callers
Function
convert
Convert keys in checkpoints. There can be some breaking changes during the development of mmdetection, and this tool is used for upgrading ch
tools/upgrade_model_version.py:8
↓ 1 callers
Function
convert
Convert keys in detectron pretrained ResNet models to pytorch style.
tools/detectron2pytorch.py:34
↓ 1 callers
Method
copy_grads_to_fp32
Copy gradients from fp16 model to fp32 weight copy.
mmdet/core/fp16/hooks.py:46
↓ 1 callers
Method
copy_params_to_fp16
Copy updated params from fp32 weight copy to fp16 model.
mmdet/core/fp16/hooks.py:54
↓ 1 callers
Function
det2json
(dataset, results)
mmdet/core/evaluation/coco_utils.py:146
↓ 1 callers
Function
detect
(img)
tests/async_benchmark.py:69
↓ 1 callers
Function
dice_loss
(input, target)
mmdet/models/anchor_heads/decoupled_solo_light_head.py:33
↓ 1 callers
Function
dice_loss
(input, target)
mmdet/models/anchor_heads/decoupled_solo_head.py:33
↓ 1 callers
Function
dice_loss
(input, target)
mmdet/models/anchor_heads/solov2_head.py:33
↓ 1 callers
Function
dice_loss
(input, target)
mmdet/models/anchor_heads/solov2_light_head.py:33
↓ 1 callers
Function
dice_loss
(input, target)
mmdet/models/anchor_heads/solo_head.py:33
↓ 1 callers
Function
eval_recalls
Calculate recalls. Args: gts(list or ndarray): a list of arrays of shape (n, 4) proposals(list or ndarray): a list of arrays of s
mmdet/core/evaluation/recall.py:62
↓ 1 callers
Method
extract_feat
(self, imgs)
mmdet/models/detectors/base.py:41
↓ 1 callers
Method
fcos_target
(self, points, gt_bboxes_list, gt_labels_list)
mmdet/models/anchor_heads/fcos_head.py:317
↓ 1 callers
Method
forward
(self, logits, targets)
mmdet/ops/sigmoid_focal_loss/sigmoid_focal_loss.py:46
↓ 1 callers
Method
forward
(self, features, rois)
mmdet/ops/roi_align/roi_align.py:73
↓ 1 callers
Method
forward
(self, features, rois)
mmdet/ops/roi_pool/roi_pool.py:62
↓ 1 callers
Method
forward_test
Args: imgs (List[Tensor]): the outer list indicates test-time augmentations and inner Tensor should have a shape
mmdet/models/detectors/base.py:102
↓ 1 callers
Method
forward_train
Args: img (list[Tensor]): list of tensors of shape (1, C, H, W). Typically these should be mean centered and std
mmdet/models/detectors/base.py:50
↓ 1 callers
Method
fovea_target
(self, gt_bbox_list, gt_label_list, featmap_sizes, points)
mmdet/models/anchor_heads/fovea_head.py:232
↓ 1 callers
Function
ga_loc_target
Compute location targets for guided anchoring. Each feature map is divided into positive, negative and ignore regions. - positive regions: ta
mmdet/core/anchor/guided_anchor_target.py:32
↓ 1 callers
Function
ga_shape_target
Compute guided anchoring targets. Args: approx_list (list[list]): Multi level approxs of each image. inside_flag_list (list[list]
mmdet/core/anchor/guided_anchor_target.py:133
↓ 1 callers
Method
gen_base_anchors
(self)
mmdet/core/anchor/anchor_generator.py:29
↓ 1 callers
Function
gen_packages_items
()
setup.py:197
↓ 1 callers
Method
get_ann_info
(self, idx)
mmdet/datasets/custom.py:92
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, m
mmdet/models/anchor_heads/reppoints_head.py:542
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, m
mmdet/models/anchor_heads/guided_anchor_head.py:560
↓ 1 callers
Method
get_bboxes_single
Transform outputs for a single batch item into labeled boxes.
mmdet/models/anchor_heads/anchor_head.py:280
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, f
mmdet/models/anchor_heads/fovea_head.py:340
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, c
mmdet/models/anchor_heads/atss_head.py:294
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, c
mmdet/models/anchor_heads/fcos_head.py:239
↓ 1 callers
Function
get_cls_results
Get det results and gt information of a certain class. Args: det_results (list[list]): Same as `eval_map()`. annotations (list[di
mmdet/core/evaluation/mean_ap.py:240
← previous
next →
201–300 of 1,004, ranked by callers