MCPcopy Create free account

hub / github.com/chengyangfu/retinamask / functions

Functions379 in github.com/chengyangfu/retinamask

↓ 1 callersFunctionbuild_roi_heads
(cfg)
maskrcnn_benchmark/modeling/roi_heads/roi_heads.py:41
↓ 1 callersFunctionbuild_rpn
This gives the gist of it. Not super important because it doesn't change as much
maskrcnn_benchmark/modeling/rpn/rpn.py:135
↓ 1 callersMethodbuild_transform
Creates a basic transformation that was used to train the models
demo/predictor.py:129
↓ 1 callersFunctionbuild_transforms
(cfg, is_train=True)
maskrcnn_benchmark/data/transforms/build.py:5
↓ 1 callersFunctioncache_url
r"""Loads the Torch serialized object at the given URL. If the object is already present in `model_dir`, it's deserialized and returned. The f
maskrcnn_benchmark/utils/model_zoo.py:15
↓ 1 callersFunctioncompute_on_dataset
(model, data_loader, device)
maskrcnn_benchmark/engine/inference.py:23
↓ 1 callersMethodcompute_prediction
Arguments: original_image (np.ndarray): an image as returned by OpenCV Returns: prediction (BoxList): the de
demo/predictor.py:182
↓ 1 callersMethodconvert_to_roi_format
(self, boxes)
maskrcnn_benchmark/modeling/poolers.py:80
↓ 1 callersMethodcreate_complex_model
(self)
tests/checkpoint.py:18
↓ 1 callersMethodcreate_mask_montage
Create a montage showing the probability heatmaps for each one one of the detected objects Arguments: image (np.
demo/predictor.py:296
↓ 1 callersMethodcrop
(self, box)
maskrcnn_benchmark/structures/segmentation_mask.py:41
↓ 1 callersMethodcrop
Cropss a rectangular region from this bounding box. The box is a 4-tuple defining the left, upper, right, and lower pixel coo
maskrcnn_benchmark/structures/bounding_box.py:167
↓ 1 callersFunctiondo_train
( model, data_loader, optimizer, scheduler, checkpointer, device, checkpoint_perio
maskrcnn_benchmark/engine/trainer.py:38
↓ 1 callersFunctionevaluate_box_proposals
Evaluate detection proposal recall metrics. This function is a much faster alternative to the official COCO API recall evaluation code. However,
maskrcnn_benchmark/engine/inference.py:125
↓ 1 callersFunctionevaluate_predictions_on_coco
( coco_gt, coco_results, json_result_file, iou_type="bbox" )
maskrcnn_benchmark/engine/inference.py:241
↓ 1 callersFunctionexpand_boxes
(boxes, scale)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:91
↓ 1 callersFunctionexpand_masks
(mask, padding)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:108
↓ 1 callersMethodfilter_results
Returns bounding-box detection results by thresholding on scores and applying non-maximum suppression (NMS).
maskrcnn_benchmark/modeling/roi_heads/box_head/inference.py:95
↓ 1 callersMethodforward
Arguments: x (Tensor): the mask logits boxes (list[BoxList]): bounding boxes that are used as referen
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:27
↓ 1 callersMethodforward
(self, x)
maskrcnn_benchmark/layers/misc.py:31
↓ 1 callersMethodforward_for_single_feature_map
Arguments: anchors: list[BoxList] box_cls: tensor of size N, A * C, H, W box_regression: tensor of size N
maskrcnn_benchmark/modeling/rpn/retinanet_infer.py:47
↓ 1 callersMethodforward_for_single_feature_map
Arguments: anchors: list[BoxList] box_cls: tensor of size N, A * C, H, W box_regression: tensor of size N
maskrcnn_benchmark/modeling/rpn/retinanet_detail_infer.py:47
↓ 1 callersMethodforward_for_single_feature_map
Arguments: anchors: list[BoxList] objectness: tensor of size N, A, H, W box_regression: tensor of size N,
maskrcnn_benchmark/modeling/rpn/inference.py:74
↓ 1 callersMethodforward_single_image
(self, masks, boxes)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:164
↓ 1 callersMethodget_c2_detectron_12_2017_baselines
(name)
maskrcnn_benchmark/config/paths_catalog.py:95
↓ 1 callersMethodget_c2_imagenet_pretrained
(name)
maskrcnn_benchmark/config/paths_catalog.py:87
↓ 1 callersFunctionget_extensions
()
setup.py:17
↓ 1 callersMethodget_idxs
(self, idx)
maskrcnn_benchmark/data/datasets/concat_dataset.py:13
↓ 1 callersMethodget_img_info
(self, index)
maskrcnn_benchmark/data/datasets/coco.py:62
↓ 1 callersFunctionget_pil_version
()
maskrcnn_benchmark/utils/collect_env.py:7
↓ 1 callersMethodget_size
(self, image_size)
maskrcnn_benchmark/data/transforms/transforms.py:33
↓ 1 callersMethodgrid_anchors
(self, grid_sizes)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:73
↓ 1 callersFunctionkeep_only_positive_boxes
Given a set of BoxList containing the `labels` field, return a set of BoxList for which `labels > 0`. Arguments: boxes (list of
maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py:13
↓ 1 callersFunctionload_c2_format
(cfg, f)
maskrcnn_benchmark/utils/c2_model_loading.py:138
↓ 1 callersFunctionmain
()
tools/test_net.py:21
↓ 1 callersFunctionmain
()
tools/train_net.py:109
↓ 1 callersFunctionmain
()
demo/webcam.py:11
↓ 1 callersFunctionmake_anchor_generator
(config)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:129
↓ 1 callersFunctionmake_anchor_generator_retinanet
(config)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:147
↓ 1 callersFunctionmake_batch_data_sampler
( dataset, sampler, aspect_grouping, images_per_batch, num_iters=None, start_iter=0 )
maskrcnn_benchmark/data/build.py:82
↓ 1 callersFunctionmake_data_sampler
(dataset, shuffle, distributed)
maskrcnn_benchmark/data/build.py:56
↓ 1 callersFunctionmake_lr_scheduler
(cfg, optimizer)
maskrcnn_benchmark/solver/build.py:23
↓ 1 callersFunctionmake_optimizer
(cfg, model)
maskrcnn_benchmark/solver/build.py:7
↓ 1 callersFunctionmake_retinanet_detail_postprocessor
( config, fpn_post_nms_top_n, rpn_box_coder)
maskrcnn_benchmark/modeling/rpn/retinanet_detail_infer.py:206
↓ 1 callersFunctionmake_retinanet_loss_evaluator
(cfg, box_coder)
maskrcnn_benchmark/modeling/rpn/retinanet_loss.py:160
↓ 1 callersFunctionmake_roi_box_feature_extractor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_feature_extractors.py:86
↓ 1 callersFunctionmake_roi_box_loss_evaluator
(cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/loss.py:159
↓ 1 callersFunctionmake_roi_box_post_processor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/inference.py:139
↓ 1 callersFunctionmake_roi_box_predictor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_predictors.py:60
↓ 1 callersFunctionmake_roi_mask_feature_extractor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/roi_mask_feature_extractors.py:66
↓ 1 callersFunctionmake_roi_mask_loss_evaluator
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/loss.py:133
↓ 1 callersFunctionmake_roi_mask_post_processor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:186
↓ 1 callersFunctionmake_roi_mask_predictor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/roi_mask_predictors.py:42
↓ 1 callersFunctionmake_rpn_loss_evaluator
(cfg, box_coder)
maskrcnn_benchmark/modeling/rpn/loss.py:139
↓ 1 callersMethodmatch_targets_to_anchors
(self, anchor, target)
maskrcnn_benchmark/modeling/rpn/loss.py:36
↓ 1 callersMethodmatch_targets_to_anchors
(self, anchor, target)
maskrcnn_benchmark/modeling/rpn/retinanet_loss.py:50
↓ 1 callersMethodmatch_targets_to_proposals
(self, proposal, target)
maskrcnn_benchmark/modeling/roi_heads/box_head/loss.py:32
↓ 1 callersMethodmatch_targets_to_proposals
(self, proposal, target)
maskrcnn_benchmark/modeling/roi_heads/mask_head/loss.py:57
↓ 1 callersFunctionnms_cpu
maskrcnn_benchmark/csrc/cpu/nms_cpu.cpp:67
↓ 1 callersMethodnum_anchors_per_location
(self)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:70
↓ 1 callersMethodoverlay_boxes
Adds the predicted boxes on top of the image Arguments: image (np.ndarray): an image as returned by OpenCV p
demo/predictor.py:247
↓ 1 callersMethodoverlay_class_names
Adds detected class names and scores in the positions defined by the top-left corner of the predicted bounding box Arguments
demo/predictor.py:332
↓ 1 callersMethodoverlay_mask
Adds the instances contours for each predicted object. Each label has a different color. Arguments: image (np.nd
demo/predictor.py:270
↓ 1 callersFunctionparse
(log_path)
tools/parse_log.py:5
↓ 1 callersFunctionpaste_mask_in_image
(mask, box, im_h, im_w, thresh=0.5, padding=1)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:118
↓ 1 callersFunctionpre_calc_for_bilinear_interpolate
maskrcnn_benchmark/csrc/cpu/ROIAlign_cpu.cpp:18
↓ 1 callersMethodprepare_boxlist
Returns BoxList from `boxes` and adds probability scores information as an extra field `boxes` has shape (#detections, 4 * #c
maskrcnn_benchmark/modeling/roi_heads/box_head/inference.py:76
↓ 1 callersFunctionprepare_for_coco_detection
(predictions, dataset)
maskrcnn_benchmark/engine/inference.py:39
↓ 1 callersFunctionprepare_for_coco_segmentation
(predictions, dataset)
maskrcnn_benchmark/engine/inference.py:73
↓ 1 callersMethodprepare_targets
(self, proposals, targets)
maskrcnn_benchmark/modeling/roi_heads/box_head/loss.py:45
↓ 1 callersMethodprepare_targets
(self, proposals, targets)
maskrcnn_benchmark/modeling/roi_heads/mask_head/loss.py:70
↓ 1 callersMethodprepare_targets
(self, anchors, targets)
maskrcnn_benchmark/modeling/rpn/loss.py:50
↓ 1 callersMethodprepare_targets
(self, anchors, targets)
maskrcnn_benchmark/modeling/rpn/retinanet_loss.py:64
↓ 1 callersFunctionproject_masks_on_boxes
Given segmentation masks and the bounding boxes corresponding to the location of the masks in the image, this function crops and resizes
maskrcnn_benchmark/modeling/roi_heads/mask_head/loss.py:11
↓ 1 callersFunctionreduce_loss_dict
Reduce the loss dictionary from all processes so that process with rank 0 has the averaged results. Returns a dict with the same fields as
maskrcnn_benchmark/engine/trainer.py:13
↓ 1 callersMethodresize
(self, size, *args, **kwargs)
maskrcnn_benchmark/structures/segmentation_mask.py:47
↓ 1 callersMethodrun_on_opencv_image
Arguments: image (np.ndarray): an image as returned by OpenCV Returns: prediction (BoxList): the detected ob
demo/predictor.py:159
↓ 1 callersFunctionscatter_gather
This function gathers data from multiple processes, and returns them in a list, as they were obtained from each process. This function i
maskrcnn_benchmark/utils/comm.py:81
↓ 1 callersMethodselect_over_all_levels
(self, boxlists)
maskrcnn_benchmark/modeling/rpn/retinanet_infer.py:151
↓ 1 callersMethodselect_over_all_levels
(self, boxlists)
maskrcnn_benchmark/modeling/rpn/retinanet_detail_infer.py:152
↓ 1 callersMethodselect_over_all_levels
(self, boxlists)
maskrcnn_benchmark/modeling/rpn/inference.py:152
↓ 1 callersMethodselect_top_predictions
Select only predictions which have a `score` > self.confidence_threshold, and returns the predictions in descending order of score
demo/predictor.py:218
↓ 1 callersMethodset_epoch
(self, epoch)
maskrcnn_benchmark/data/samplers/distributed.py:66
↓ 1 callersMethodset_low_quality_matches_
Produce additional matches for predictions that have only low-quality matches. Specifically, for each ground-truth find the set of pr
maskrcnn_benchmark/modeling/matcher.py:79
↓ 1 callersFunctionsetup_custom_environment
Load custom environment setup from a Python source file and run the setup function.
maskrcnn_benchmark/utils/env.py:20
↓ 1 callersFunctionsetup_environment
Perform environment setup work. The default setup is a no-op, but this function allows the user to specify a Python source file that performs
maskrcnn_benchmark/utils/env.py:7
↓ 1 callersFunctionstrip_prefix_if_present
(state_dict, prefix)
maskrcnn_benchmark/utils/model_serialization.py:61
↓ 1 callersMethodsubsample
This method performs the positive/negative sampling, and return the sampled proposals. Note: this function keeps a state.
maskrcnn_benchmark/modeling/roi_heads/box_head/loss.py:75
↓ 1 callersMethodtag_last_checkpoint
(self, last_filename)
maskrcnn_benchmark/utils/checkpoint.py:88
↓ 1 callersFunctiontest
(cfg, model, distributed)
tools/train_net.py:79
↓ 1 callersFunctiontrain
(cfg, local_rank, distributed)
tools/train_net.py:29
↓ 1 callersMethodtranspose
(self, method)
maskrcnn_benchmark/structures/segmentation_mask.py:23
FunctionPYBIND11_MODULE
maskrcnn_benchmark/csrc/vision.cpp:7
FunctionROIAlignForward_cpu_kernel
maskrcnn_benchmark/csrc/cpu/ROIAlign_cpu.cpp:114
FunctionROIAlign_backward
maskrcnn_benchmark/csrc/ROIAlign.h:27
FunctionROIAlign_forward
Interface for Python
maskrcnn_benchmark/csrc/ROIAlign.h:11
FunctionROIPool_backward
maskrcnn_benchmark/csrc/ROIPool.h:26
FunctionROIPool_forward
maskrcnn_benchmark/csrc/ROIPool.h:11
FunctionSigmoidFocalLoss_backward
maskrcnn_benchmark/csrc/SigmoidFocalLoss.h:26
FunctionSigmoidFocalLoss_forward
Interface for Python
maskrcnn_benchmark/csrc/SigmoidFocalLoss.h:10
← previousnext →101–200 of 379, ranked by callers