Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Gus-Guo/AugFPN
/ functions
Functions
482 in github.com/Gus-Guo/AugFPN
⨍
Functions
482
◇
Types & classes
90
↓ 1 callers
Function
_minimal_ext_cmd
(cmd)
setup.py:24
↓ 1 callers
Function
_non_dist_train
(model, dataset, cfg, validate=False)
mmdet/apis/train.py:99
↓ 1 callers
Method
_output_size
(input, weight, padding, dilation, stride)
mmdet/ops/dcn/functions/deform_conv.py:92
↓ 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:56
↓ 1 callers
Function
_prepare_data
(img, img_transform, cfg, device)
mmdet/apis/inference.py:11
↓ 1 callers
Method
_rand_another
(self, idx)
mmdet/datasets/custom.py:151
↓ 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/models/registry.py:18
↓ 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:139
↓ 1 callers
Function
accuracy
(pred, target, topk=1)
mmdet/core/loss/losses.py:141
↓ 1 callers
Method
add_gt_
(self, gt_labels)
mmdet/core/bbox/assigners/assign_result.py:12
↓ 1 callers
Function
allreduce_grads
(model, coalesce=True, bucket_size_mb=-1)
mmdet/core/utils/dist_utils.py:31
↓ 1 callers
Function
anchor_inside_flags
(flat_anchors, valid_flags, img_shape, allowed_border=0)
mmdet/core/anchor/anchor_target.py:175
↓ 1 callers
Function
assign_and_sample
(bboxes, gt_bboxes, gt_bboxes_ignore, gt_labels, cfg)
mmdet/core/bbox/assign_sampling.py:28
↓ 1 callers
Method
assign_wrt_overlaps
Assign w.r.t. the overlaps of bboxes with gts. Args: overlaps (Tensor): Overlaps between k gt_bboxes and n bboxes,
mmdet/core/bbox/assigners/max_iou_assigner.py:93
↓ 1 callers
Method
aug_test
(self, imgs, img_metas, **kwargs)
mmdet/models/detectors/base.py:50
↓ 1 callers
Method
aug_test
(self, imgs, img_metas, rescale=False)
mmdet/models/detectors/rpn.py:66
↓ 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:9
↓ 1 callers
Method
backward
(ctx, grad_output)
mmdet/ops/roi_pool/functions/roi_pool.py:38
↓ 1 callers
Function
bbox_flip
Flip bboxes horizontally. Args: bboxes(ndarray): shape (..., 4*k) img_shape(tuple): (height, width)
mmdet/datasets/transforms.py:49
↓ 1 callers
Method
build_roi_layers
(self, layer_cfg, featmap_strides)
mmdet/models/roi_extractors/all_level_auxiliary.py:40
↓ 1 callers
Method
build_roi_layers
(self, layer_cfg, featmap_strides)
mmdet/models/roi_extractors/soft_roi_selection.py:68
↓ 1 callers
Method
build_roi_layers
(self, layer_cfg, featmap_strides)
mmdet/models/roi_extractors/single_level.py:43
↓ 1 callers
Method
centerness_target
(self, pos_bbox_targets)
mmdet/models/anchor_heads/fcos_head.py:397
↓ 1 callers
Function
customize_compiler_for_nvcc
inject deep into distutils to customize how the dispatch to cc/nvcc works. If you subclass UnixCCompiler, it's not trivial to get your subclas
mmdet/ops/nms/setup.py:25
↓ 1 callers
Function
det2json
(dataset, results)
mmdet/core/evaluation/coco_utils.py:103
↓ 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
Function
expand_binary_labels
(labels, label_weights, label_channels)
mmdet/core/anchor/anchor_target.py:165
↓ 1 callers
Function
expand_target
(bbox_targets, bbox_weights, labels, num_classes)
mmdet/core/bbox/bbox_target.py:67
↓ 1 callers
Method
extract_feat
(self, imgs)
mmdet/models/detectors/base.py:33
↓ 1 callers
Method
fcos_target
(self, points, gt_bboxes_list, gt_labels_list)
mmdet/models/anchor_heads/fcos_head.py:316
↓ 1 callers
Method
forward_test
(self, imgs, img_metas, **kwargs)
mmdet/models/detectors/base.py:58
↓ 1 callers
Method
forward_train
(self, imgs, img_metas, **kwargs)
mmdet/models/detectors/base.py:42
↓ 1 callers
Method
gen_base_anchors
(self)
mmdet/core/anchor/anchor_generator.py:18
↓ 1 callers
Method
get_ann_info
(self, idx)
mmdet/datasets/custom.py:128
↓ 1 callers
Method
get_bboxes_auxiliary
(self, cls_scores, bbox_preds, centernesses,
mmdet/models/anchor_heads/fcos_head.py:408
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, m
mmdet/models/anchor_heads/anchor_head.py:243
↓ 1 callers
Method
get_bboxes_single
(self, cls_scores, bbox_preds, c
mmdet/models/anchor_heads/fcos_head.py:238
↓ 1 callers
Method
get_bboxes_single_auxiliary
(self, cls_scores, bbox_preds, m
mmdet/models/anchor_heads/retina_head.py:98
↓ 1 callers
Method
get_bboxes_single_auxiliary
(self, cls_scores, bbox_preds, c
mmdet/models/anchor_heads/fcos_head.py:441
↓ 1 callers
Function
get_cls_results
Get det results and gt information of a certain class.
mmdet/core/evaluation/mean_ap.py:202
↓ 1 callers
Function
get_dataset
(data_cfg)
mmdet/datasets/utils.py:77
↓ 1 callers
Function
get_git_hash
()
setup.py:22
↓ 1 callers
Function
get_hash
()
setup.py:48
↓ 1 callers
Method
get_points_single
(self, featmap_size, stride, dtype, device)
mmdet/models/anchor_heads/fcos_head.py:305
↓ 1 callers
Function
get_version
()
setup.py:77
↓ 1 callers
Method
init_offset
(self)
mmdet/ops/dcn/modules/deform_conv.py:132
↓ 1 callers
Method
init_weights
(self)
mmdet/models/utils/conv_module.py:71
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/two_stage.py:71
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/two_stage_old.py:56
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/cascade_rcnn.py:83
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/single_stage_old.py:28
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/rpn.py:28
↓ 1 callers
Method
init_weights
(self, pretrained=None)
mmdet/models/detectors/single_stage.py:55
↓ 1 callers
Function
iou_loss
(pred_bboxes, target_bboxes, reduction='mean')
mmdet/core/loss/losses.py:122
↓ 1 callers
Method
load_annotations
(self, ann_file)
mmdet/datasets/custom.py:122
↓ 1 callers
Method
load_proposals
(self, proposal_file)
mmdet/datasets/custom.py:125
↓ 1 callers
Method
loss_aux
(self, mask_pred, mask_targets, labels, alpha=0.25)
mmdet/models/mask_heads/fcn_mask_head.py:110
↓ 1 callers
Function
main
()
tools/train.py:42
↓ 1 callers
Function
main
()
tools/test.py:62
↓ 1 callers
Function
main
()
tools/voc_eval.py:46
↓ 1 callers
Function
main
()
tools/coco_eval.py:6
↓ 1 callers
Function
main
()
tools/slurm_test.py:68
↓ 1 callers
Function
main
()
tools/convert_datasets/pascal_voc.py:103
↓ 1 callers
Function
make_res_layer
(block, inplanes, planes, blocks,
mmdet/models/backbones/resnext.py:87
↓ 1 callers
Function
make_res_layer
(block, inplanes, planes, blocks,
mmdet/models/backbones/resnet.py:231
↓ 1 callers
Method
map_roi_levels
Map rois to corresponding feature levels by scales. - scale < finest_scale: level 0 - finest_scale <= scale < finest_scale * 2: level
mmdet/models/roi_extractors/single_level.py:52
↓ 1 callers
Function
mask_target
(pos_proposals_list, pos_assigned_gt_inds_list, gt_masks_list, cfg)
mmdet/core/mask/mask_target.py:6
↓ 1 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:41
↓ 1 callers
Function
merge_aug_proposals
Merge augmented proposals (multiscale, flip, etc.) Args: aug_proposals (list[Tensor]): proposals from different testing schem
mmdet/core/post_processing/merge_augs.py:9
↓ 1 callers
Method
norm1
(self)
mmdet/models/backbones/resnet.py:57
↓ 1 callers
Method
norm1
(self)
mmdet/models/backbones/resnet.py:178
↓ 1 callers
Method
norm1
(self)
mmdet/models/backbones/resnet.py:383
↓ 1 callers
Method
norm2
(self)
mmdet/models/backbones/resnet.py:61
↓ 1 callers
Method
norm2
(self)
mmdet/models/backbones/resnet.py:182
↓ 1 callers
Method
norm3
(self)
mmdet/models/backbones/resnet.py:186
↓ 1 callers
Function
parse_args
()
tools/train.py:14
↓ 1 callers
Function
parse_args
()
tools/test.py:39
↓ 1 callers
Function
parse_args
()
tools/slurm_test.py:40
↓ 1 callers
Function
parse_args
()
tools/convert_datasets/pascal_voc.py:94
↓ 1 callers
Function
parse_losses
(losses)
mmdet/apis/train.py:16
↓ 1 callers
Method
prepare_test_img
Prepare an image for testing (multi-scale and flipping)
mmdet/datasets/custom.py:245
↓ 1 callers
Method
prepare_train_img
(self, idx)
mmdet/datasets/custom.py:165
↓ 1 callers
Function
print_map_summary
Print mAP and results of each class. Args: mean_ap(float): calculated from `eval_map` results(list): calculated from `eval_map`
mmdet/core/evaluation/mean_ap.py:332
↓ 1 callers
Function
print_recall_summary
Print recalls in a table. Args: recalls(ndarray): calculated from `bbox_recalls` proposal_nums(ndarray or list): top N proposals
mmdet/core/evaluation/recall.py:105
↓ 1 callers
Function
proposal2json
(dataset, results)
mmdet/core/evaluation/coco_utils.py:88
↓ 1 callers
Function
random_scale
Randomly select a scale from a list of scales or scale ranges. Args: img_scales (list[tuple]): Image scale or scale range. mode (
mmdet/datasets/utils.py:36
↓ 1 callers
Function
readme
()
setup.py:7
↓ 1 callers
Method
refine_bboxes
Refine bboxes during training. Args: rois (Tensor): Shape (n*bs, 5), where n is image number per GPU, and bs is t
mmdet/models/bbox_heads/bbox_head.py:134
↓ 1 callers
Method
regress_by_class
Regress the bbox for the predicted class. Used in Cascade R-CNN. Args: rois (Tensor): shape (n, 4) or (n, 5) label (T
mmdet/models/bbox_heads/bbox_head_auxiliary.py:174
↓ 1 callers
Method
reset_parameters
(self)
mmdet/ops/dcn/modules/deform_conv.py:46
↓ 1 callers
Method
reset_parameters
(self)
mmdet/ops/dcn/modules/deform_conv.py:91
↓ 1 callers
Method
sample
(self, assign_result, bboxes, gt_bboxes, **kwargs)
mmdet/core/bbox/samplers/pseudo_sampler.py:18
↓ 1 callers
Function
segm2json
(dataset, results)
mmdet/core/evaluation/coco_utils.py:120
↓ 1 callers
Function
set_random_seed
(seed)
mmdet/apis/env.py:51
↓ 1 callers
Function
set_recall_param
Check proposal_nums and iou_thrs and set correct format.
mmdet/core/evaluation/recall.py:40
↓ 1 callers
Function
sigmoid_focal_loss
(pred, target, weight, gamma=2.0,
mmdet/core/loss/losses.py:35
↓ 1 callers
Method
simple_test
(self, img, img_meta, **kwargs)
mmdet/models/detectors/base.py:46
← previous
next →
101–200 of 482, ranked by callers