MCPcopy Create free account

hub / github.com/TuSimple/mx-maskrcnn / functions

Functions350 in github.com/TuSimple/mx-maskrcnn

↓ 42 callersMethodupdate
(self, labels, preds)
rcnn/core/metric.py:41
↓ 15 callersMethodlist_arguments
(self)
rcnn/PY_OP/mask_roi.py:49
↓ 10 callersMethodinfo
Print information about the annotation file. :return:
rcnn/pycocotools/coco.py:116
↓ 9 callersFunctionbbox_overlaps
(boxes, query_boxes)
rcnn/processing/bbox_transform.py:5
↓ 8 callersFunctiongenerate_config
(_network, _dataset)
rcnn/config.py:163
↓ 8 callersFunctionload_param
wrapper for load checkpoint :param prefix: Prefix of model name. :param epoch: Epoch number of model we would like to load. :param co
rcnn/utils/load_model.py:39
↓ 8 callersFunctionresidual_unit
(data, num_filter, stride, dim_match, name)
rcnn/symbol/symbol_mask_fpn.py:13
↓ 8 callersFunctionrleInit
rcnn/pycocotools/maskApi.c:14
↓ 7 callersMethod__init__
(self)
rcnn/core/metric.py:37
↓ 7 callersFunctiontensor_vstack
vertically stack tensors :param tensor_list: list of tensor to be stacked vertically :param pad: label to pad with :return: tensor wi
rcnn/io/image.py:127
↓ 6 callersMethodinfer_shape
(self, in_shape)
rcnn/PY_OP/mask_roi.py:55
↓ 5 callersFunctionget_fpn_maskrcnn_batch
return a dictionary that contains raw data.
rcnn/io/rcnn.py:99
↓ 5 callersFunctionget_image
preprocess image and return processed roidb :param roidb: a list of roidb :return: list of img as in mxnet format roidb add new item[
rcnn/io/image.py:9
↓ 5 callersFunctionget_maskrcnn_fpn_name
()
rcnn/core/metric.py:26
↓ 5 callersMethodgt_roidb
(self)
rcnn/dataset/imdb.py:42
↓ 5 callersFunctionnms
greedily select boxes with high confidence and overlap with current maximum <= thresh rule out overlap >= thresh :param dets: [[x1, y1, x
rcnn/processing/nms.py:24
↓ 4 callersFunctionassign_anchor_fpn
assign ground truth boxes to anchor positions :param feat_shape: infer output shape :param gt_boxes: assign ground truth :param im_in
rcnn/io/rpn.py:74
↓ 4 callersFunctionclip_boxes
Clip boxes to image boundaries. :param boxes: [N, 4* num_classes] :param im_shape: tuple of 2 :return: [N, 4* num_classes]
rcnn/processing/bbox_transform.py:32
↓ 4 callersMethodgetAnnIds
Get ann ids that satisfy given filter conditions. default skips that filter :param imgIds (int array) : get anns for given imgs
rcnn/pycocotools/coco.py:124
↓ 4 callersFunctionget_resnet_conv
(data)
rcnn/symbol/symbol_mask_fpn.py:35
↓ 4 callersFunctionget_resnet_conv_down
(conv_feat)
rcnn/symbol/symbol_mask_fpn.py:75
↓ 4 callersMethodloadAnns
Load anns with the specified ids. :param ids (int array) : integer ids specifying anns :return: anns (object array) : l
rcnn/pycocotools/coco.py:195
↓ 4 callersMethodload_rpn_roidb
turn rpn detection boxes into roidb :param gt_roidb: [image_index]['boxes', 'gt_classes', 'gt_overlaps', 'flipped'] :return:
rcnn/dataset/imdb.py:78
↓ 4 callersFunctionumin
rcnn/pycocotools/maskApi.c:11
↓ 3 callersFunction_unmap
unmap a subset inds of data into original data of size count
rcnn/io/rpn.py:90
↓ 3 callersMethodbind
(self, data_shapes, label_shapes=None, for_training=True, inputs_need_grad=False, force_rebind=Fa
rcnn/core/module.py:92
↓ 3 callersFunctionexpand_bbox_regression_targets
expand from 5 to 4 * num_classes; only the right class has non-zero bbox regression targets :param bbox_targets_data: [k * 5] :param num_
rcnn/processing/bbox_regression.py:236
↓ 3 callersFunctionget_rpn_names
()
rcnn/core/metric.py:5
↓ 3 callersMethodlist_outputs
(self)
rcnn/PY_OP/mask_roi.py:52
↓ 3 callersFunctionload_checkpoint
Load model checkpoint from file. :param prefix: Prefix of model name. :param epoch: Epoch number of model we would like to load. :ret
rcnn/utils/load_model.py:4
↓ 3 callersFunctionpy_nms_wrapper
(thresh)
rcnn/processing/nms.py:6
↓ 3 callersFunctiontest_rpn
(network, dataset, image_set, root_path, dataset_path, ctx, prefix, epoch, vis, shuf
rcnn/tools/test_rpn.py:13
↓ 3 callersFunctiontrain_rpn
(network, dataset, image_set, root_path, dataset_path, frequent, kvstore, work_load_list, no_fli
rcnn/tools/train_rpn.py:15
↓ 3 callersFunctionumax
rcnn/pycocotools/maskApi.c:12
↓ 2 callersMethod__init__
(self, roidb, batch_size=1, shuffle=False, has_rpn=False)
rcnn/core/loader.py:12
↓ 2 callersMethod_clip_pad
Clip boxes of the pad area. :param tensor: [n, c, H, W] :param pad_shape: [h, w] :return: [n, c, h, w]
rcnn/PY_OP/proposal_fpn.py:142
↓ 2 callersMethod_insert_queue
(self)
rcnn/io/threaded_loader.py:60
↓ 2 callersFunction_mask_umap
(mask_targets, mask_labels, mask_inds)
rcnn/io/rcnn.py:224
↓ 2 callersFunction_mkanchors
Given a vector of widths (ws) and heights (hs) around a center (x_ctr, y_ctr), output a set of anchors (windows).
rcnn/processing/generate_anchor.py:49
↓ 2 callersMethod_thread_start
(self, num_thread)
rcnn/io/threaded_loader.py:64
↓ 2 callersFunction_whctrs
Return width, height, x center, and y center for an anchor (window).
rcnn/processing/generate_anchor.py:37
↓ 2 callersFunctionanchors_plane
(feat_h, feat_w, stride, base_anchor)
rcnn/processing/generate_anchor.py:9
↓ 2 callersMethodannToRLE
Convert annotation which can be polygons, uncompressed RLE to RLE. :return: binary mask (numpy 2D array)
rcnn/pycocotools/coco.py:400
↓ 2 callersMethodappend_flipped_images
append flipped images to an roidb flip boxes coordinates, images will be actually flipped when loading into network :param ro
rcnn/dataset/imdb.py:151
↓ 2 callersFunctionbbIou
rcnn/pycocotools/maskApi.c:109
↓ 2 callersFunctioncombine_model
(prefix1, epoch1, prefix2, epoch2, prefix_out, epoch_out)
rcnn/utils/combine_model.py:5
↓ 2 callersFunctionconvert_context
:param params: dict of str to NDArray :param ctx: the context to convert to :return: dict of str of NDArray with context ctx
rcnn/utils/load_model.py:27
↓ 2 callersMethodcreateIndex
(self)
rcnn/pycocotools/coco.py:85
↓ 2 callersFunctiondemo_maskrcnn
(network, dataset, image_set, root_path, dataset_path, result_path, ctx, prefix, epoch,
rcnn/tools/demo_maskrcnn.py:13
↓ 2 callersFunctionfilter_roidb
remove roidb entries without usable rois
rcnn/tools/train_maskrcnn.py:66
↓ 2 callersMethodfit
(self, train_data, eval_metric=None, epoch_end_callback=None, batch_end_callback=None, kvstore='lo
rcnn/core/solver.py:36
↓ 2 callersMethodforward
(self, data_batch, is_train=None)
rcnn/core/module.py:159
↓ 2 callersFunctiongenerate_anchors
Generate anchor (reference) windows by enumerating aspect ratios X scales wrt a reference (0, 0, 15, 15) window.
rcnn/processing/generate_anchor.py:12
↓ 2 callersMethodgetImgIds
Get img ids that satisfy given filter conditions. :param imgIds (int array) : get imgs for given ids :param catIds (int array
rcnn/pycocotools/coco.py:174
↓ 2 callersMethodget_batch
(self)
rcnn/core/loader.py:78
↓ 2 callersMethodget_batch
(self)
rcnn/core/loader.py:197
↓ 2 callersMethodget_batch
(self)
rcnn/core/loader.py:485
↓ 2 callersMethodget_batch
(self)
rcnn/io/threaded_loader.py:117
↓ 2 callersMethodget_outputs
(self, merge_multi_context=True)
rcnn/core/module.py:200
↓ 2 callersFunctionget_rpn_batch
prototype for rpn batch: data, im_info, gt_boxes :param roidb: ['image', 'flipped'] + ['gt_boxes', 'boxes', 'gt_classes'] :return: data
rcnn/io/rpn.py:41
↓ 2 callersMethodgetindex
(self)
rcnn/io/threaded_loader.py:108
↓ 2 callersMethodgetpad
(self)
rcnn/io/threaded_loader.py:111
↓ 2 callersFunctionim_detect_mask
(predictor, data_batch, data_names, scale=1)
rcnn/core/tester.py:105
↓ 2 callersMethodinfer_shape
Return maximum data and label shape for single gpu
rcnn/core/loader.py:461
↓ 2 callersMethodinit_params
(self, initializer=Uniform(0.01), arg_params=None, aux_params=None, allow_missing=False, f
rcnn/core/module.py:82
↓ 2 callersMethoditer_next
(self)
rcnn/io/threaded_loader.py:85
↓ 2 callersFunctionmerge_roidb
roidb are list, concat them together
rcnn/utils/load_data.py:27
↓ 2 callersMethodmerge_roidbs
merge roidbs into one :param a: roidb to be merged into :param b: roidb to be merged :return: merged imdb
rcnn/dataset/imdb.py:281
↓ 2 callersMethodpredict
(self, data_batch)
rcnn/core/tester.py:25
↓ 2 callersMethodreset
(self)
rcnn/io/threaded_loader.py:79
↓ 2 callersFunctionrleFree
rcnn/pycocotools/maskApi.c:19
↓ 2 callersFunctionrleToBbox
rcnn/pycocotools/maskApi.c:133
↓ 2 callersFunctiontest_maskrcnn
(network, dataset, image_set, root_path, dataset_path, result_path, ctx, prefix, epoch,
rcnn/tools/test_maskrcnn.py:13
↓ 2 callersFunctiontrain_maskrcnn
(network, dataset, image_set, root_path, dataset_path, frequent, kvstore, work_load_list, no_fl
rcnn/tools/train_maskrcnn.py:19
↓ 2 callersMethodupdate
(self)
rcnn/core/module.py:196
↓ 1 callersFunctionROIAlignBackwardAcc
rcnn/CXX_OP/roi_align.cc:32
↓ 1 callersFunctionROIAlignForward
rcnn/CXX_OP/roi_align.cc:21
↓ 1 callersMethod__init__
This iter will provide roi and mask data to Mask-RCNN with FPN backbone :param roidb: list of dict, must be preprocessed :par
rcnn/io/threaded_loader.py:12
↓ 1 callersMethod__init__
(self, num_classes)
rcnn/PY_OP/mask_roi.py:45
↓ 1 callersMethod__init__
(self)
rcnn/PY_OP/mask_output.py:33
↓ 1 callersMethod__init__
(self, rcnn_strides='(32,16,8,4)', pool_h='7', pool_w='7')
rcnn/PY_OP/fpn_roi_pooling.py:92
↓ 1 callersMethod__init__
(self, feat_stride='(64,32,16,8,4)', scales='(8)', ratios='(0.5, 1, 2)', output_score='False',
rcnn/PY_OP/proposal_fpn.py:160
↓ 1 callersMethod_filter_boxes
Remove all boxes with any side smaller than min_size
rcnn/PY_OP/proposal_fpn.py:134
↓ 1 callersMethod_group_indexes
(self)
rcnn/io/threaded_loader.py:202
↓ 1 callersMethod_insert_queue
(self)
rcnn/io/threaded_loader.py:385
↓ 1 callersMethod_make_data_and_labels
(self, im_array_list, levels_data_list)
rcnn/core/loader.py:224
↓ 1 callersMethod_prepare
Prepare ._gts and ._dts for evaluation based on params :return: None
rcnn/pycocotools/cocoeval.py:85
↓ 1 callersFunction_ratio_enum
Enumerate a set of anchors for each aspect ratio wrt an anchor.
rcnn/processing/generate_anchor.py:64
↓ 1 callersMethod_reset_bind
(self)
rcnn/core/module.py:51
↓ 1 callersFunction_scale_enum
Enumerate a set of anchors for each scale wrt an anchor.
rcnn/processing/generate_anchor.py:78
↓ 1 callersMethod_thread_start
(self, num_thread)
rcnn/io/threaded_loader.py:389
↓ 1 callersFunction_unmap
unmap a subset inds of data into original data of size count
rcnn/PY_OP/fpn_roi_pooling.py:11
↓ 1 callersFunctionadd_assign_targets
given roidb, add ['assign_level'] :param roidb: roidb to be processed. must have gone through imdb.prepare_roidb
rcnn/processing/assign_levels.py:18
↓ 1 callersFunctionadd_bbox_regression_targets
given roidb, add ['bbox_targets'] and normalize bounding box regression targets :param roidb: roidb to be processed. must have gone through i
rcnn/processing/bbox_regression.py:54
↓ 1 callersFunctionadd_mask_targets
given roidb, add ['bbox_targets'] and normalize bounding box regression targets :param roidb: roidb to be processed. must have gone through i
rcnn/processing/bbox_regression.py:193
↓ 1 callersFunctionalternate_train
(args, ctx, pretrained, epoch, rpn_epoch, rpn_lr, rpn_lr_step, rcnn_ep
train_alternate_mask_fpn.py:13
↓ 1 callersMethodcheck_aspect_queues
(self)
rcnn/io/threaded_loader.py:256
↓ 1 callersMethodcheck_params
(self, arg_params, aux_params)
rcnn/core/solver.py:29
↓ 1 callersFunctioncompute_assign_targets
(rois, threshold)
rcnn/processing/assign_levels.py:5
↓ 1 callersFunctioncompute_bbox_mask_targets_and_label
given rois, overlaps, gt labels, seg, compute bounding box mask targets :param rois: roidb[i]['boxes'] k * 4 :param overlaps: roidb[i]['m
rcnn/processing/bbox_regression.py:158
next →1–100 of 350, ranked by callers