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
↓ 3 callers
Method
get_ann_info
(self, idx)
mmdet/datasets/coco.py:41
↓ 3 callers
Function
get_results
(filename, dataset='coco', task='bbox', metric=None,
tools/robustness_eval.py:161
↓ 3 callers
Function
images_to_levels
Convert targets by image to targets by feature level. [target_img0, target_img1] -> [target_level0, target_level1, ...]
mmdet/core/anchor/guided_anchor_target.py:205
↓ 3 callers
Function
inference_detector
Inference image(s) with the detector. Args: model (nn.Module): The loaded detector. imgs (str/ndarray or list[str/ndarray]): Eith
mmdet/apis/inference.py:65
↓ 3 callers
Function
merge_aug_bboxes
Merge augmented detection bboxes and scores. Args: aug_bboxes (list[Tensor]): shape (n, 4*#class) aug_scores (list[Tensor] or Non
mmdet/core/post_processing/merge_augs.py:46
↓ 3 callers
Method
norm1
(self)
mmdet/models/backbones/hrnet.py:345
↓ 3 callers
Function
print_coco_results
(results)
tools/robustness_eval.py:8
↓ 3 callers
Function
show_result
Visualize the detection results on the image. Args: img (str or np.ndarray): Image filename or loaded image. result (tuple[list]
mmdet/apis/inference.py:121
↓ 3 callers
Method
valid_flags
(self, featmap_size, valid_size, device='cuda')
mmdet/core/anchor/point_generator.py:24
↓ 2 callers
Method
__init__
(self, spatial_scale, out_size, out_channels,
mmdet/ops/dcn/deform_pool.py:84
↓ 2 callers
Method
__init__
(self, *args, **kwargs)
mmdet/models/necks/nas_fpn.py:58
↓ 2 callers
Method
__init__
(self, depth, in_channels=3, num_stages=4,
mmdet/models/backbones/resnet.py:354
↓ 2 callers
Method
__init__
(self, eps=1e-6, reduction='mean', loss_weight=1.0)
mmdet/models/losses/iou_loss.py:120
↓ 2 callers
Method
__nice__
(self)
mmdet/utils/util_mixins.py:79
↓ 2 callers
Function
_dummy_bbox_sampling
Create sample results that can be passed to BBoxHead.get_target
tests/test_heads.py:93
↓ 2 callers
Method
_meshgrid
(self, x, y, row_major=True)
mmdet/core/anchor/point_generator.py:6
↓ 2 callers
Method
_meshgrid
(self, x, y, row_major=True)
mmdet/core/anchor/anchor_generator.py:58
↓ 2 callers
Method
_resize
(self, x, size)
mmdet/models/necks/nas_fpn.py:26
↓ 2 callers
Function
accuracy
(pred, target, topk=1)
mmdet/models/losses/accuracy.py:4
↓ 2 callers
Method
albu_builder
Import a module from albumentations. Inherits some of `build_from_cfg` logic. Args: cfg (dict): Config dict. It should at
mmdet/datasets/pipelines/transforms.py:773
↓ 2 callers
Function
allreduce_grads
(params, coalesce=True, bucket_size_mb=-1)
mmdet/core/utils/dist_utils.py:31
↓ 2 callers
Function
assign_and_sample
(bboxes, gt_bboxes, gt_bboxes_ignore, gt_labels, cfg)
mmdet/core/bbox/assign_sampling.py:26
↓ 2 callers
Method
backward
(ctx, grad_output)
mmdet/ops/roi_pool/roi_pool.py:33
↓ 2 callers
Function
bbox_flip
Flip bboxes horizontally. Args: bboxes(Tensor or ndarray): Shape (..., 4*k) img_shape(tuple): Image shape. Returns:
mmdet/core/bbox/transforms.py:114
↓ 2 callers
Method
bbox_flip
Flip bboxes horizontally. Args: bboxes(ndarray): shape (..., 4*k) img_shape(tuple): (height, width)
mmdet/datasets/pipelines/transforms.py:209
↓ 2 callers
Function
build_optimizer
Build optimizer from configs. Args: model (:obj:`nn.Module`): The model with parameters to be optimized. optimizer_cfg (dict): Th
mmdet/apis/train.py:114
↓ 2 callers
Function
concurrent
Run code concurrently in different streams. :param streamqueue: asyncio.Queue instance. Queue tasks define the pool of streams used for conc
mmdet/utils/contextmanagers.py:95
↓ 2 callers
Function
cvt_annotations
(devkit_path, years, split, out_file)
tools/convert_datasets/pascal_voc.py:67
↓ 2 callers
Function
ensure_rng
Simple version of the ``kwarray.ensure_rng`` Args: rng (int | numpy.random.RandomState | None): if None, then defaults t
mmdet/core/bbox/demodata.py:5
↓ 2 callers
Function
flops_to_string
(flops, units='GMac', precision=2)
mmdet/utils/flops_counter.py:69
↓ 2 callers
Method
gen_grid_from_reg
Base on the previous bboxes and regression values, we compute the regressed bboxes and generate the grids on the bboxes.
mmdet/models/anchor_heads/reppoints_head.py:215
↓ 2 callers
Method
get_anchors
Get squares according to feature map sizes and guided anchors. Args: featmap_sizes (list[tuple]): Multi-level feature map
mmdet/models/anchor_heads/guided_anchor_head.py:271
↓ 2 callers
Function
get_masks
(result, num_classes=80)
tools/test_ins.py:22
↓ 2 callers
Method
get_points
Get points according to feature map sizes. Args: featmap_sizes (list[tuple]): Multi-level feature map sizes. img_meta
mmdet/models/anchor_heads/reppoints_head.py:293
↓ 2 callers
Method
get_points
(self, featmap_sizes, dtype, device, flatten=False)
mmdet/models/anchor_heads/fovea_head.py:172
↓ 2 callers
Method
get_points
Get points according to feature map sizes. Args: featmap_sizes (list[tuple]): Multi-level feature map sizes. dtype (t
mmdet/models/anchor_heads/fcos_head.py:288
↓ 2 callers
Method
grid_points
(self, featmap_size, stride=16, device='cuda')
mmdet/core/anchor/point_generator.py:14
↓ 2 callers
Method
hard_mining
(self, inds, num_expected, bboxes, labels, feats)
mmdet/core/bbox/samplers/ohem_sampler.py:32
↓ 2 callers
Function
last_zero_init
(m)
mmdet/ops/context_block.py:6
↓ 2 callers
Function
make_res_layer
(block, inplanes, planes, blocks,
mmdet/models/backbones/resnet.py:243
↓ 2 callers
Function
makeplot
(rs, ps, outDir, class_name, iou_type)
tools/coco_error_analysis.py:12
↓ 2 callers
Method
mapper
Dictionary mapper. Renames keys according to keymap provided. Args: d (dict): old dict keymap (dict)
mmdet/datasets/pipelines/transforms.py:806
↓ 2 callers
Method
norm
(self)
mmdet/models/utils/conv_module.py:150
↓ 2 callers
Method
offset_to_pts
Change from point offset to point coordinate.
mmdet/models/anchor_heads/reppoints_head.py:349
↓ 2 callers
Function
parse_require_file
(fpath)
setup.py:189
↓ 2 callers
Function
point_target
Compute corresponding GT box and classification targets for proposals. Args: points_list (list[list]): Multi level points of each image.
mmdet/core/anchor/point_target.py:7
↓ 2 callers
Method
pre_pipeline
(self, results)
mmdet/datasets/custom.py:95
↓ 2 callers
Function
profile_time
Print time spent by CPU and GPU. Useful as a temporary context manager to find sweet spots of code suitable for async implementation.
mmdet/utils/profiling.py:10
↓ 2 callers
Function
random_boxes
Simple version of ``kwimage.Boxes.random`` Returns: Tensor: shape (n, 4) in x1, y1, x2, y2 format. References: https://
mmdet/core/bbox/demodata.py:30
↓ 2 callers
Method
random_sample
(img_scales)
mmdet/datasets/pipelines/transforms.py:82
↓ 2 callers
Function
reduce_mean
(tensor)
mmdet/models/anchor_heads/atss_head.py:16
↓ 2 callers
Function
set_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:19
↓ 2 callers
Method
show_result
Show RPN proposals on the image. Although we assume batch size is 1, this method supports arbitrary batch size.
mmdet/models/detectors/rpn.py:84
↓ 2 callers
Function
show_result_ins
Visualize the instance segmentation results on the image. Args: img (str or np.ndarray): Image filename or loaded image. result (
mmdet/apis/inference.py:209
↓ 2 callers
Function
sigmoid_focal_loss
(pred, target, weight=None, gamma=2.0,
mmdet/models/losses/focal_loss.py:28
↓ 2 callers
Method
simple_test_bboxes
Test only det bboxes without augmentation.
mmdet/models/detectors/test_mixins.py:97
↓ 2 callers
Function
smooth_l1_loss
(pred, target, beta=1.0)
mmdet/models/losses/smooth_l1_loss.py:9
↓ 1 callers
Method
__init__
(self, dataset, interval=1)
mmdet/core/evaluation/eval_hooks.py:20
↓ 1 callers
Method
__init__
(self, dataset, samples_per_gpu=1)
mmdet/datasets/loader/sampler.py:39
↓ 1 callers
Method
__init__
(self, input_size, depth, with_last_pool=False,
mmdet/models/backbones/ssd_vgg.py:38
↓ 1 callers
Method
__init__
(self, groups=1, base_width=4, **kwargs)
mmdet/models/backbones/resnext.py:189
↓ 1 callers
Method
__init__
(self, extra, in_channels=3, conv_cfg=None,
mmdet/models/backbones/hrnet.py:256
↓ 1 callers
Method
__init__
(self, in_channels, out_channels, conv_cfg=None,
mmdet/models/bbox_heads/double_bbox_head.py:23
↓ 1 callers
Method
__init__
(self, num_shared_convs=0, num_shared_fcs=0, num_cls_convs=
mmdet/models/bbox_heads/convfc_bbox_head.py:18
↓ 1 callers
Method
__init__
(self, bins=10, momentum=0, use_sigmoid=True, loss_weight=1.0)
mmdet/models/losses/ghm_loss.py:34
↓ 1 callers
Method
__init__
(self, in_channels, out_channels, kernel_size=3,
mmdet/models/anchor_heads/guided_anchor_head.py:32
↓ 1 callers
Method
__init__
(self, num_classes, in_channels, feat_channels=256,
mmdet/models/anchor_heads/fovea_head.py:49
↓ 1 callers
Method
__repr__
(self)
mmdet/utils/util_mixins.py:89
↓ 1 callers
Method
_add_conv_branch
Add the fc branch which consists of a sequential of conv layers
mmdet/models/bbox_heads/double_bbox_head.py:120
↓ 1 callers
Method
_add_fc_branch
Add the fc branch which consists of a sequential of fc layers
mmdet/models/bbox_heads/double_bbox_head.py:132
↓ 1 callers
Function
_allreduce_coalesced
(tensors, world_size, bucket_size_mb=-1)
mmdet/core/utils/dist_utils.py:9
↓ 1 callers
Method
_bbox_forward_train
(self, stage, x, sampling_
mmdet/models/detectors/htc.py:43
↓ 1 callers
Method
_binary_op
(self, x1, x2)
mmdet/models/necks/nas_fpn.py:23
↓ 1 callers
Method
_check_branches
(self, num_branches, num_blocks, in_channels, num_channels)
mmdet/models/backbones/hrnet.py:43
↓ 1 callers
Function
_concat_dataset
(cfg, default_args=None)
mmdet/datasets/builder.py:8
↓ 1 callers
Function
_demodata_refine_boxes
Create random test data for the ``mmdet.models.bbox_heads.bbox_head.BBoxHead.refine_boxes`` method
tests/test_heads.py:299
↓ 1 callers
Function
_dist_train
(model, dataset, cfg, validate=False, logger=N
mmdet/apis/train.py:194
↓ 1 callers
Function
_expand_binary_labels
(labels, label_weights, label_channels)
mmdet/models/losses/ghm_loss.py:8
↓ 1 callers
Function
_expand_binary_labels
(labels, label_weights, label_channels)
mmdet/models/losses/cross_entropy_loss.py:22
↓ 1 callers
Method
_filter_imgs
Filter images too small.
mmdet/datasets/custom.py:103
↓ 1 callers
Function
_get_config_directory
Find the predefined detector config directory
tests/test_config.py:4
↓ 1 callers
Function
_get_config_directory
Find the predefined detector config directory
tests/test_forward.py:11
↓ 1 callers
Function
_get_config_module
Load a configuration as a python module
tests/test_forward.py:26
↓ 1 callers
Method
_infer_shape
(ctx, input, weight)
mmdet/ops/dcn/deform_conv.py:175
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/decoupled_solo_light_head.py:82
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/decoupled_solo_head.py:80
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/reppoints_head.py:109
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/solov2_head.py:85
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/guided_anchor_head.py:172
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/anchor_head.py:83
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/fovea_head.py:83
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/solov2_light_head.py:85
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/fcos_head.py:71
↓ 1 callers
Method
_init_layers
(self)
mmdet/models/anchor_heads/solo_head.py:81
↓ 1 callers
Method
_load_anns
(self, results)
mmdet/datasets/pipelines/instaboost.py:35
↓ 1 callers
Method
_load_bboxes
(self, results)
mmdet/datasets/pipelines/loading.py:52
↓ 1 callers
Method
_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
mmdet/ops/dcn/deform_conv.py:283
↓ 1 callers
Method
_load_labels
(self, results)
mmdet/datasets/pipelines/loading.py:63
↓ 1 callers
Method
_load_masks
(self, results)
mmdet/datasets/pipelines/loading.py:82
↓ 1 callers
Method
_load_semantic_seg
(self, results)
mmdet/datasets/pipelines/loading.py:91
← previous
next →
101–200 of 1,004, ranked by callers