MCPcopy Create free account

hub / github.com/HRNet/HRNet-MaskRCNN-Benchmark / functions

Functions431 in github.com/HRNet/HRNet-MaskRCNN-Benchmark

↓ 1 callersFunction_load_cfg_py_source
Load a config from a Python source file.
maskrcnn_benchmark/config/yacs.py:325
↓ 1 callersMethod_load_model
(self, checkpoint)
maskrcnn_benchmark/utils/checkpoint.py:96
↓ 1 callersFunction_load_module_from_file
(name, filename)
maskrcnn_benchmark/config/yacs.py:489
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
maskrcnn_benchmark/modeling/backbone/hrnet.py:148
↓ 1 callersMethod_make_fuse_layers
(self)
maskrcnn_benchmark/modeling/backbone/hrnet.py:157
↓ 1 callersMethod_make_layer
(self, block, inplanes, planes, blocks, stride=1)
maskrcnn_benchmark/modeling/backbone/hrnet.py:311
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
maskrcnn_benchmark/modeling/backbone/hrnet.py:135
↓ 1 callersFunction_merge_a_into_b
Merge config dictionary a into config dictionary b, clobbering the options in b whenever they are also specified in a.
maskrcnn_benchmark/config/yacs.py:370
↓ 1 callersFunction_output_size
(dim)
maskrcnn_benchmark/layers/misc.py:90
↓ 1 callersMethod_preprocess_annotation
(self, target)
maskrcnn_benchmark/data/datasets/voc.py:87
↓ 1 callersFunction_quantize
(x, bins)
maskrcnn_benchmark/data/build.py:69
↓ 1 callersFunction_ratio_enum
Enumerate a set of anchors for each aspect ratio wrt an anchor.
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:245
↓ 1 callersFunction_rename_basic_resnet_weights
(layer_keys)
maskrcnn_benchmark/utils/c2_model_loading.py:12
↓ 1 callersFunction_rename_fpn_weights
(layer_keys, stage_names)
maskrcnn_benchmark/utils/c2_model_loading.py:52
↓ 1 callersFunction_rename_weights_for_resnet
(weights, stage_names)
maskrcnn_benchmark/utils/c2_model_loading.py:72
↓ 1 callersFunction_scale_enum
Enumerate a set of anchors for each scale wrt an anchor.
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:256
↓ 1 callersFunction_to_dict
Recursively convert all CfgNode objects to dict objects.
maskrcnn_benchmark/config/yacs.py:345
↓ 1 callersMethodadd_gt_proposals
Arguments: proposals: list[BoxList] targets: list[BoxList]
maskrcnn_benchmark/modeling/rpn/inference.py:51
↓ 1 callersFunctionadd_path
(path)
tools/_init_paths.py:13
↓ 1 callersMethodadd_visibility_to
(self, boxlist)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:92
↓ 1 callersFunctionalign_and_update_state_dicts
Strategy: suppose that the models that we will create will have prefixes appended to each of its keys, for example due to an extra level of n
maskrcnn_benchmark/utils/model_serialization.py:10
↓ 1 callersFunctionbuild_backbone
(cfg)
maskrcnn_benchmark/modeling/backbone/backbone.py:56
↓ 1 callersFunctionbuild_dataset
Arguments: dataset_list (list[str]): Contains the names of the datasets, i.e., coco_2014_trian, coco_2014_val, etc tr
maskrcnn_benchmark/data/build.py:17
↓ 1 callersFunctionbuild_roi_box_head
Constructs a new box head. By default, uses ROIBoxHead, but if it turns out not to be enough, just register a new class and make it a par
maskrcnn_benchmark/modeling/roi_heads/box_head/box_head.py:64
↓ 1 callersFunctionbuild_roi_heads
(cfg)
maskrcnn_benchmark/modeling/roi_heads/roi_heads.py:41
↓ 1 callersFunctionbuild_roi_mask_head
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/mask_head.py:81
↓ 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:176
↓ 1 callersMethodbuild_transform
Creates a basic transformation that was used to train the models
demo/predictor.py:130
↓ 1 callersFunctionbuild_transforms
(cfg, is_train=True, flip=False)
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 callersFunctioncalc_detection_voc_ap
Calculate average precisions based on evaluation code of PASCAL VOC. This function calculates average precisions from given precisions and rec
maskrcnn_benchmark/data/datasets/evaluation/voc/voc_eval.py:160
↓ 1 callersFunctioncalc_detection_voc_prec_rec
Calculate precision and recall based on evaluation code of PASCAL VOC. This function calculates precision and recall of predicted bounding box
maskrcnn_benchmark/data/datasets/evaluation/voc/voc_eval.py:68
↓ 1 callersFunctioncalc_iou
(a, b)
tools/test_net.py:24
↓ 1 callersFunctioncoco_evaluation
( dataset, predictions, output_folder, box_only, iou_types, expected_results, expe
maskrcnn_benchmark/data/datasets/evaluation/coco/__init__.py:4
↓ 1 callersFunctioncompute_on_dataset
(model, data_loader, device)
maskrcnn_benchmark/engine/inference.py:16
↓ 1 callersMethodcompute_prediction
Arguments: original_image (np.ndarray): an image as returned by OpenCV Returns: prediction (BoxList): the de
demo/predictor.py:183
↓ 1 callersFunctionconditional_cast
(from_type, to_type)
maskrcnn_benchmark/config/yacs.py:455
↓ 1 callersFunctionconvert_cityscapes_instance_only
Convert from cityscapes format to COCO instance seg format - polygons
tools/cityscapes/convert_cityscapes_to_coco.py:101
↓ 1 callersFunctionconvert_coco_stuff_mat
Convert to png and save json with path. This currently only contains the segmentation labels for objects+stuff in cocostuff - if we need to co
tools/cityscapes/convert_cityscapes_to_coco.py:53
↓ 1 callersFunctionconvert_to_dict
(cfg_node, key_list)
maskrcnn_benchmark/config/yacs.py:348
↓ 1 callersMethodconvert_to_roi_format
(self, boxes)
maskrcnn_benchmark/modeling/poolers.py:78
↓ 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:298
↓ 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 callersMethoddefrost
Make this CfgNode and all of its children mutable.
maskrcnn_benchmark/config/yacs.py:210
↓ 1 callersFunctiondo_coco_evaluation
( dataset, predictions, box_only, output_folder, iou_types, expected_results, expe
maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py:13
↓ 1 callersFunctiondo_train
( model, data_loader, optimizer, scheduler, checkpointer, device, checkpoint_perio
maskrcnn_benchmark/engine/trainer.py:38
↓ 1 callersFunctiondo_voc_evaluation
(dataset, predictions, output_folder, logger)
maskrcnn_benchmark/data/datasets/evaluation/voc/voc_eval.py:12
↓ 1 callersMethoddump
Dump to a string.
maskrcnn_benchmark/config/yacs.py:162
↓ 1 callersFunctioneval_detection_voc
Evaluate on voc dataset. Args: pred_boxlists(list[BoxList]): pred boxlist, has labels and scores fields. gt_boxlists(list[BoxList]
maskrcnn_benchmark/data/datasets/evaluation/voc/voc_eval.py:48
↓ 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/data/datasets/evaluation/coco/coco_eval.py:156
↓ 1 callersFunctionevaluate_predictions_on_coco
( coco_gt, coco_results, json_result_file, iou_type="bbox" )
maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py:272
↓ 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] objectness: tensor of size N, A, H, W box_regression: tensor of size N,
maskrcnn_benchmark/modeling/rpn/inference.py:75
↓ 1 callersMethodforward_single_image
(self, masks, boxes)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:167
↓ 1 callersMethodget_c2_detectron_12_2017_baselines
(name)
maskrcnn_benchmark/config/paths_catalog.py:162
↓ 1 callersMethodget_c2_imagenet_pretrained
(name)
maskrcnn_benchmark/config/paths_catalog.py:154
↓ 1 callersFunctionget_extensions
()
setup.py:17
↓ 1 callersMethodget_idxs
(self, idx)
maskrcnn_benchmark/data/datasets/concat_dataset.py:13
↓ 1 callersMethodget_lr
(self)
maskrcnn_benchmark/solver/lr_scheduler.py:41
↓ 1 callersMethodget_num_inchannels
(self)
maskrcnn_benchmark/modeling/backbone/hrnet.py:195
↓ 1 callersMethodget_pad_size
(self, image_size)
maskrcnn_benchmark/data/transforms/transforms.py:66
↓ 1 callersFunctionget_pil_version
()
maskrcnn_benchmark/utils/collect_env.py:7
↓ 1 callersMethodget_size
(self, image_size)
maskrcnn_benchmark/data/transforms/transforms.py:35
↓ 1 callersMethodgrid_anchors
(self, grid_sizes)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:68
↓ 1 callersFunctioninstances2dict_with_polygons
(imageFileList, verbose=False)
tools/cityscapes/instances2dict_with_polygons.py:17
↓ 1 callersMethodis_frozen
Return mutability.
maskrcnn_benchmark/config/yacs.py:214
↓ 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:154
↓ 1 callersFunctionload_cfg
Load a cfg. Supports loading from: - A file object backed by a YAML file - A file object backed by a Python source file that exports a
maskrcnn_benchmark/config/yacs.py:284
↓ 1 callersFunctionmain
()
tools/test_net.py:135
↓ 1 callersFunctionmain
()
tools/train_net.py:113
↓ 1 callersFunctionmain
(argv)
tools/cityscapes/instances2dict_with_polygons.py:70
↓ 1 callersFunctionmain
()
demo/webcam.py:11
↓ 1 callersFunctionmake_anchor_generator
(config)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:123
↓ 1 callersFunctionmake_batch_data_sampler
( dataset, sampler, aspect_grouping, images_per_batch, num_iters=None, start_iter=0 )
maskrcnn_benchmark/data/build.py:85
↓ 1 callersFunctionmake_data_sampler
(dataset, shuffle, distributed)
maskrcnn_benchmark/data/build.py:59
↓ 1 callersFunctionmake_lr_scheduler
(cfg, optimizer)
maskrcnn_benchmark/solver/build.py:23
↓ 1 callersFunctionmake_optimizer
(cfg, model)
maskrcnn_benchmark/solver/build.py:5
↓ 1 callersFunctionmake_roi_box_feature_extractor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_feature_extractors.py:122
↓ 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:82
↓ 1 callersFunctionmake_roi_mask_feature_extractor
(cfg)
maskrcnn_benchmark/modeling/roi_heads/mask_head/roi_mask_feature_extractors.py:65
↓ 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:197
↓ 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 callersMethodmap_class_id_to_class_name
(self, class_id)
maskrcnn_benchmark/data/datasets/voc.py:133
↓ 1 callersMethodmatch_targets_to_anchors
(self, anchor, target)
maskrcnn_benchmark/modeling/rpn/loss.py:36
↓ 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 callersMethodmerge_from_other_cfg
Merge `cfg_other` into this CfgNode.
maskrcnn_benchmark/config/yacs.py:173
↓ 1 callersFunctionnms_cpu
maskrcnn_benchmark/csrc/cpu/nms_cpu.cpp:67
↓ 1 callersMethodnum_anchors_per_location
(self)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:65
← previousnext →101–200 of 431, ranked by callers