MCPcopy Create free account

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

Functions350 in github.com/TuSimple/mx-maskrcnn

↓ 1 callersFunctioncompute_bbox_regression_targets
given rois, overlaps, gt labels, compute bounding box regression targets :param rois: roidb[i]['boxes'] k * 4 :param overlaps: roidb[i]['
rcnn/processing/bbox_regression.py:18
↓ 1 callersFunctioncompute_mask_and_label
(ex_rois, ex_labels, seg, flipped)
rcnn/processing/bbox_regression.py:106
↓ 1 callersMethodcreate_roidb_from_box_list
given ground truth, prepare roidb :param box_list: [image_index] ndarray of [box_index][x1, x2, y1, y2] :param gt_roidb: [ima
rcnn/dataset/imdb.py:102
↓ 1 callersFunctioncustomize_compiler_for_nvcc
inject deep into distutils to customize how the dispatch to gcc/nvcc works. If you subclass UnixCCompiler, it's not trivial to get your subcl
rcnn/cython/setup.py:68
↓ 1 callersFunctiondemo_maskrcnn
(network, ctx, prefix, epoch,img_path, vis= True, has_rpn = True, thresh = 0.001)
rcnn/tools/demo_single_image.py:14
↓ 1 callersFunctiondraw_detection_mask
(im_array, boxes_this_image, masks_this_image, scale, filename)
rcnn/core/tester.py:291
↓ 1 callersMethodevaluate_mask
(self, results_pack)
rcnn/dataset/cityscape.py:189
↓ 1 callersMethodevaluate_recall
evaluate detection proposal recall metrics record max overlap value for each gt box; return vector of overlap values :param r
rcnn/dataset/imdb.py:184
↓ 1 callersFunctionfind_in_path
Find a file in a search path
rcnn/cython/setup.py:16
↓ 1 callersFunctiongenerate_anchors_fpn
Generate anchor (reference) windows by enumerating aspect ratios X scales wrt a reference (0, 0, 15, 15) window.
rcnn/processing/generate_anchor.py:25
↓ 1 callersFunctiongenerate_proposals
Generate detections results using RPN. :param predictor: Predictor :param test_data: data iterator, must be non-shuffled :param imdb:
rcnn/core/tester.py:44
↓ 1 callersMethodgetCatIds
filtering parameters. default skips that filter. :param catNms (str array) : get cats for given cat names :param supNms (str
rcnn/pycocotools/coco.py:152
↓ 1 callersMethodget_feat_shape_list
(feat_sym, data_shape)
rcnn/io/threaded_loader.py:460
↓ 1 callersFunctionget_fpn_rcnn_testbatch
return a dict of testbatch :param roidb: ['image', 'flipped'] + ['boxes'] :return: data, label, im_info
rcnn/io/rcnn.py:46
↓ 1 callersMethodget_params
(self)
rcnn/core/module.py:78
↓ 1 callersFunctionget_rcnn_fpn_names
()
rcnn/core/metric.py:17
↓ 1 callersFunctionget_rcnn_testbatch
return a dict of testbatch :param roidb: ['image', 'flipped'] + ['boxes'] :return: data, label, im_info
rcnn/io/rcnn.py:24
↓ 1 callersFunctionget_rpn_testbatch
return a dict of testbatch :param roidb: ['image', 'flipped'] :return: data, label, im_info
rcnn/io/rpn.py:23
↓ 1 callersMethodgetindex
(self)
rcnn/core/loader.py:69
↓ 1 callersMethodgetindex
(self)
rcnn/core/loader.py:188
↓ 1 callersMethodgetindex
(self)
rcnn/core/loader.py:452
↓ 1 callersMethodgetindex
(self)
rcnn/io/threaded_loader.py:427
↓ 1 callersMethodgetpad
(self)
rcnn/core/loader.py:72
↓ 1 callersMethodgetpad
(self)
rcnn/core/loader.py:191
↓ 1 callersMethodgetpad
(self)
rcnn/core/loader.py:455
↓ 1 callersMethodgetpad
(self)
rcnn/io/threaded_loader.py:430
↓ 1 callersFunctiongpu_nms_wrapper
(thresh, device_id)
rcnn/processing/nms.py:18
↓ 1 callersFunctionim_proposal
(predictor, data_batch, data_names, scale)
rcnn/core/tester.py:30
↓ 1 callersMethodimage_path_from_index
(self, index)
rcnn/dataset/imdb.py:39
↓ 1 callersMethodinfer_label_shape
calculate label shape according to data shape :param data_shape: dict of shape {"data": (1, 3, 600, 1000)} :param fea
rcnn/io/threaded_loader.py:436
↓ 1 callersFunctionis_valid
valid images have at least 1 fg or bg roi
rcnn/tools/train_maskrcnn.py:69
↓ 1 callersFunctionis_valid
valid images have at least 1 fg or bg roi
rcnn/utils/load_data.py:38
↓ 1 callersMethoditer_next
(self)
rcnn/core/loader.py:55
↓ 1 callersMethoditer_next
(self)
rcnn/core/loader.py:175
↓ 1 callersMethoditer_next
(self)
rcnn/core/loader.py:439
↓ 1 callersMethoditer_next
(self)
rcnn/io/threaded_loader.py:406
↓ 1 callersMethodloadCats
Load cats with the specified ids. :param ids (int array) : integer ids specifying cats :return: cats (object array) : l
rcnn/pycocotools/coco.py:206
↓ 1 callersMethodloadImgs
Load anns with the specified ids. :param ids (int array) : integer ids specifying img :return: imgs (object array) : lo
rcnn/pycocotools/coco.py:217
↓ 1 callersMethodloadNumpyAnnotations
Convert result data from a numpy array [Nx7] where each row contains {imageID,x1,y1,w,h,score,class} :param data (numpy.ndarray)
rcnn/pycocotools/coco.py:377
↓ 1 callersMethodload_cityscape_annotations
for a given index, load image and bounding boxes info from a single image list :return: list of record['boxes', 'gt_classes', 'ins_id
rcnn/dataset/cityscape.py:109
↓ 1 callersMethodload_from_seg
(self, ins_seg_path)
rcnn/dataset/cityscape.py:76
↓ 1 callersFunctionload_gt_roidb
load ground truth roidb
rcnn/utils/load_data.py:6
↓ 1 callersMethodload_horizontal_batch
(self)
rcnn/io/threaded_loader.py:265
↓ 1 callersMethodload_image_set_index
find out which indexes correspond to given image set (train or val) :return:
rcnn/dataset/cityscape.py:34
↓ 1 callersFunctionload_proposal_roidb
load proposal roidb (append_gt when training)
rcnn/utils/load_data.py:16
↓ 1 callersMethodload_rpn_data
(self, full=False)
rcnn/dataset/imdb.py:67
↓ 1 callersMethodload_vertical_batch
(self)
rcnn/io/threaded_loader.py:260
↓ 1 callersFunctionlocate_cuda
Locate the CUDA environment on the system Returns a dict with keys 'home', 'nvcc', 'include', and 'lib64' and values giving the absolute path
rcnn/cython/setup.py:27
↓ 1 callersFunctionmain
()
demo_mask.py:32
↓ 1 callersFunctionmain
()
train_alternate_mask_fpn.py:105
↓ 1 callersFunctionmain
()
eval_maskrcnn.py:32
↓ 1 callersFunctionmain
()
rcnn/tools/test_rpn.py:89
↓ 1 callersFunctionmain
()
rcnn/tools/train_rpn.py:183
↓ 1 callersFunctionmain
()
rcnn/tools/demo_maskrcnn.py:94
↓ 1 callersFunctionmain
()
rcnn/tools/demo_single_image.py:145
↓ 1 callersFunctionmain
()
rcnn/tools/test_maskrcnn.py:97
↓ 1 callersFunctionparse_args
()
demo_mask.py:8
↓ 1 callersFunctionparse_args
()
train_alternate_mask_fpn.py:72
↓ 1 callersFunctionparse_args
()
eval_maskrcnn.py:8
↓ 1 callersFunctionparse_args
()
rcnn/tools/test_rpn.py:67
↓ 1 callersFunctionparse_args
()
rcnn/tools/train_rpn.py:152
↓ 1 callersFunctionparse_args
()
rcnn/tools/demo_maskrcnn.py:69
↓ 1 callersFunctionparse_args
()
rcnn/tools/demo_single_image.py:123
↓ 1 callersFunctionparse_args
()
rcnn/tools/test_maskrcnn.py:72
↓ 1 callersFunctionpred_demo_mask
wrapper for calculating offline validation for faster data analysis in this example, all threshold are set by hand :param predictor: Pred
rcnn/core/tester.py:198
↓ 1 callersFunctionpred_eval_mask
wrapper for calculating offline validation for faster data analysis in this example, all threshold are set by hand :param predictor: Pred
rcnn/core/tester.py:130
↓ 1 callersMethodreset
(self)
rcnn/core/loader.py:50
↓ 1 callersMethodreset
(self)
rcnn/core/loader.py:150
↓ 1 callersMethodreset
(self)
rcnn/core/loader.py:414
↓ 1 callersMethodreset
(self)
rcnn/io/threaded_loader.py:400
↓ 1 callersFunctionresize
only resize input image to target size and return scale :param im: BGR image input by opencv :param target_size: one dimensional size (th
rcnn/io/image.py:50
↓ 1 callersFunctionrleArea
rcnn/pycocotools/maskApi.c:72
↓ 1 callersFunctionrleFrPoly
rcnn/pycocotools/maskApi.c:161
↓ 1 callersFunctionrleIou
rcnn/pycocotools/maskApi.c:77
↓ 1 callersFunctionsample_rois_fpn
generate random sample of ROIs comprising foreground and background examples :param rois: all_rois [n, 4]; e2e: [n, 5] with batch_index :
rcnn/io/rcnn.py:253
↓ 1 callersFunctionsave_checkpoint
Checkpoint the model data into file. :param prefix: Prefix of model name. :param epoch: The epoch number of the model. :param arg_params:
rcnn/utils/save_model.py:4
↓ 1 callersMethodsetDetParams
(self)
rcnn/pycocotools/cocoeval.py:497
↓ 1 callersMethodsetKpParams
(self)
rcnn/pycocotools/cocoeval.py:508
↓ 1 callersMethodsummarize
Compute and display summary metrics for evaluation results. Note this functin can *only* be applied on the default parameter setting
rcnn/pycocotools/cocoeval.py:417
↓ 1 callersFunctiontransform
transform into mxnet tensor substract pixel size and transform to correct format :param im: [height, width, channel] in BGR :param pi
rcnn/io/image.py:71
↓ 1 callersFunctionvis_all_detection
visualize all detections in one image :param im_array: [b=1 c h w] in rgb :param detections: [ numpy.ndarray([[x1 y1 x2 y2 score]]) for j
rcnn/core/tester.py:260
MethodBackward
rcnn/CXX_OP/roi_align-inl.h:82
MethodCopy
rcnn/CXX_OP/roi_align-inl.h:203
FunctionCreateOp<cpu>
rcnn/CXX_OP/roi_align.cc:47
MethodCreateOperator
rcnn/CXX_OP/roi_align-inl.h:221
MethodCreateOperatorEx
rcnn/CXX_OP/roi_align.cc:55
MethodDMLC_DECLARE_PARAMETER
rcnn/CXX_OP/roi_align_v1-inl.h:31
MethodDMLC_DECLARE_PARAMETER
rcnn/CXX_OP/roi_align-inl.h:34
MethodDeclareBackwardDependency
decalre dependency and inplace optimization options
rcnn/CXX_OP/roi_align-inl.h:214
MethodForward
rcnn/CXX_OP/roi_align-inl.h:51
MethodGetParams
rcnn/CXX_OP/roi_align-inl.h:156
MethodInferShape
rcnn/CXX_OP/roi_align-inl.h:160
MethodInferType
rcnn/CXX_OP/roi_align-inl.h:188
MethodInit
rcnn/CXX_OP/roi_align-inl.h:152
MethodListArguments
rcnn/CXX_OP/roi_align-inl.h:136
MethodListOutputs
rcnn/CXX_OP/roi_align-inl.h:140
MethodMap
rcnn/CXX_OP/roi_align_v1-inl.h:62
MethodMap
rcnn/CXX_OP/roi_align_v1-inl.h:215
MethodNumOutputs
rcnn/CXX_OP/roi_align-inl.h:144
MethodNumVisibleOutputs
rcnn/CXX_OP/roi_align-inl.h:148
← previousnext →101–200 of 350, ranked by callers