MCPcopy Create free account

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

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

↓ 41 callersMethodget_field
(self, field)
maskrcnn_benchmark/structures/bounding_box.py:42
↓ 27 callersMethodadd_field
(self, field, field_data)
maskrcnn_benchmark/structures/bounding_box.py:39
↓ 22 callersMethodto
(self, *args, **kwargs)
maskrcnn_benchmark/structures/image_list.py:24
↓ 16 callersMethodcreate_model
(self)
tests/checkpoint.py:15
↓ 14 callersFunction_assert_with_logging
(cond, msg)
maskrcnn_benchmark/config/yacs.py:483
↓ 14 callersMethodconvert
(self, mode)
maskrcnn_benchmark/structures/bounding_box.py:55
↓ 12 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
maskrcnn_benchmark/modeling/utils.py:9
↓ 11 callersMethodupdate
(self, **kwargs)
maskrcnn_benchmark/utils/metric_logger.py:45
↓ 10 callersMethodload
(self, f=None)
maskrcnn_benchmark/utils/checkpoint.py:52
↓ 10 callersMethodresize
Returns a resized copy of this bounding box :param size: The requested size in pixels, as a 2-tuple: (width, height).
maskrcnn_benchmark/structures/bounding_box.py:91
↓ 9 callersMethodclone
Recursively copy this CfgNode.
maskrcnn_benchmark/config/yacs.py:231
↓ 9 callersMethodsave
(self, name, **kwargs)
maskrcnn_benchmark/utils/checkpoint.py:32
↓ 8 callersMethodextend
(self, buffers)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:21
↓ 7 callersFunctionsynchronize
Helper function to synchronize between multiple processes when using distributed training
maskrcnn_benchmark/utils/comm.py:33
↓ 5 callersFunctionboxlist_iou
Compute the intersection over union of two set of boxes. The box order must be (xmin, ymin, xmax, ymax). Arguments: box1: (BoxList) bou
maskrcnn_benchmark/structures/boxlist_ops.py:53
↓ 5 callersMethoddecode
From a set of original boxes and encoded relative box offsets, get the decoded boxes. Arguments: rel_codes (Tens
maskrcnn_benchmark/modeling/box_coder.py:52
↓ 5 callersMethodget
(name)
maskrcnn_benchmark/config/paths_catalog.py:146
↓ 5 callersMethodstate_dict
Returns the state of the scheduler as a :class:`dict`. It contains an entry for every variable in self.__dict__ which is not the opti
maskrcnn_benchmark/solver/lr_scheduler.py:24
↓ 4 callersMethod_split_into_xyxy
(self)
maskrcnn_benchmark/structures/bounding_box.py:75
↓ 4 callersFunctioncat_boxlist
Concatenates a list of BoxList (having the same image size) into a single BoxList Arguments: bboxes (list[BoxList])
maskrcnn_benchmark/structures/boxlist_ops.py:102
↓ 4 callersMethodclip_to_image
(self, remove_empty=True)
maskrcnn_benchmark/structures/bounding_box.py:214
↓ 4 callersMethodcopy_with_fields
(self, fields)
maskrcnn_benchmark/structures/bounding_box.py:238
↓ 4 callersMethodencode
Encode a set of proposals with respect to some reference boxes Arguments: reference_boxes (Tensor): reference bo
maskrcnn_benchmark/modeling/box_coder.py:22
↓ 4 callersMethodfreeze
Make this CfgNode and all of its children immutable.
maskrcnn_benchmark/config/yacs.py:206
↓ 4 callersFunctionmake_data_loader
(cfg, is_train=True, is_distributed=False, start_iter=0, flip=False)
maskrcnn_benchmark/data/build.py:107
↓ 4 callersFunctionmkdir
(path)
maskrcnn_benchmark/utils/miscellaneous.py:6
↓ 3 callersMethod__init__
(self, cfg, **kwargs)
maskrcnn_benchmark/modeling/backbone/hrnet.py:219
↓ 3 callersMethod__init__
(self, cfg)
maskrcnn_benchmark/modeling/backbone/resnet.py:60
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
maskrcnn_benchmark/modeling/backbone/hrnet.py:320
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
maskrcnn_benchmark/modeling/backbone/hrnet.py:276
↓ 3 callersFunction_valid_type
(value, allow_cfg_node=False)
maskrcnn_benchmark/config/yacs.py:366
↓ 3 callersMethodarea
(self)
maskrcnn_benchmark/structures/bounding_box.py:226
↓ 3 callersFunctionbuild_detection_model
(cfg)
maskrcnn_benchmark/modeling/detector/detectors.py:8
↓ 3 callersMethodfields
(self)
maskrcnn_benchmark/structures/bounding_box.py:48
↓ 3 callersMethodget_checkpoint_file
(self)
maskrcnn_benchmark/utils/checkpoint.py:77
↓ 3 callersFunctionget_rank
()
maskrcnn_benchmark/utils/comm.py:21
↓ 3 callersMethodhas_checkpoint
(self)
maskrcnn_benchmark/utils/checkpoint.py:73
↓ 3 callersFunctionimport_file
(module_name, file_path, make_importable=False)
maskrcnn_benchmark/utils/imports.py:11
↓ 3 callersFunctionis_main_process
()
maskrcnn_benchmark/utils/comm.py:27
↓ 3 callersMethodload_state_dict
Loads the schedulers state. Arguments: state_dict (dict): scheduler state. Should be an object returned from a ca
maskrcnn_benchmark/solver/lr_scheduler.py:32
↓ 3 callersMethodmerge_from_file
Load a yaml config file and merge it this CfgNode.
maskrcnn_benchmark/config/yacs.py:167
↓ 3 callersMethodmerge_from_list
Merge config (keys, values) in a list (e.g., from command line) into this CfgNode. For example, `cfg_list = ['FOO.BAR', 0.5]`.
maskrcnn_benchmark/config/yacs.py:177
↓ 3 callersMethodstep
(self, epoch=None)
maskrcnn_benchmark/solver/lr_scheduler.py:44
↓ 3 callersFunctionto_image_list
tensors can be an ImageList, a torch.Tensor or an iterable of Tensors. It can't be a numpy array. When tensors is an iterable of Tensors,
maskrcnn_benchmark/structures/image_list.py:29
↓ 3 callersMethodtranspose
Transpose bounding box (flip or rotate in 90 degree steps) :param method: One of :py:attr:`PIL.Image.FLIP_LEFT_RIGHT`, :py:
maskrcnn_benchmark/structures/bounding_box.py:129
↓ 2 callersMethod__init__
(self, cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_predictors.py:55
↓ 2 callersMethod__init__
(self, cfg)
maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_feature_extractors.py:89
↓ 2 callersMethod__init__
Arguments: cfg : config in_channels (int): number of channels of the input feature num_ancho
maskrcnn_benchmark/modeling/rpn/rpn.py:54
↓ 2 callersFunction_cat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
maskrcnn_benchmark/structures/boxlist_ops.py:92
↓ 2 callersFunction_check_and_coerce_cfg_value_type
Checks that `replacement`, which is intended to replace `original` is of the right type. The type is correct if it matches exactly or is one of a
maskrcnn_benchmark/config/yacs.py:441
↓ 2 callersFunction_decode_cfg_value
Decodes a raw config value (e.g., from a yaml config files or command line argument) into a Python object.
maskrcnn_benchmark/config/yacs.py:408
↓ 2 callersMethod_immutable
Set immutability to is_immutable and recursively apply the setting to all nested CfgNodes.
maskrcnn_benchmark/config/yacs.py:218
↓ 2 callersFunction_load_cfg_from_yaml_str
Load a config from a YAML string encoding.
maskrcnn_benchmark/config/yacs.py:319
↓ 2 callersMethod_load_file
(self, f)
maskrcnn_benchmark/utils/checkpoint.py:93
↓ 2 callersFunction_make_stage
( transformation_module, in_channels, bottleneck_channels, out_channels, block_count,
maskrcnn_benchmark/modeling/backbone/resnet.py:172
↓ 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).
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:228
↓ 2 callersMethod_prepare_batches
(self)
maskrcnn_benchmark/data/samplers/grouped_batch_sampler.py:40
↓ 2 callersFunction_register_generic
(module_dict, module_name, module)
maskrcnn_benchmark/utils/registry.py:4
↓ 2 callersFunction_send_and_wait
(r)
maskrcnn_benchmark/utils/comm.py:45
↓ 2 callersFunction_whctrs
Return width, height, x center, and y center for an anchor (window).
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:219
↓ 2 callersMethodbackward
(ctx, grad_output)
maskrcnn_benchmark/layers/roi_pool.py:25
↓ 2 callersFunctionboxlist_nms
Performs non-maximum suppression on a boxlist, with scores specified in a boxlist field via score_field. Arguments: boxlist(BoxL
maskrcnn_benchmark/structures/boxlist_ops.py:9
↓ 2 callersFunctioncheck_expected_results
(results, expected_results, sigma_tol)
maskrcnn_benchmark/data/datasets/evaluation/coco/coco_eval.py:336
↓ 2 callersFunctioncollect_env_info
()
maskrcnn_benchmark/utils/collect_env.py:11
↓ 2 callersMethodcompute_colors_for_labels
Simple function that adds fixed colors depending on the class
demo/predictor.py:241
↓ 2 callersFunctionevaluate
evaluate dataset using different methods based on dataset type. Args: dataset: Dataset object predictions(list[BoxList]): each ite
maskrcnn_benchmark/data/datasets/evaluation/__init__.py:7
↓ 2 callersFunctiongenerate_anchors
Generates a matrix of anchor boxes in (x1, y1, x2, y2) format. Anchors are centered on stride / 2, have (approximate) sqrt areas of the specified
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:193
↓ 2 callersMethodget_groundtruth
(self, index)
maskrcnn_benchmark/data/datasets/voc.py:76
↓ 2 callersMethodget_img_info
(self, index)
maskrcnn_benchmark/data/datasets/coco.py:98
↓ 2 callersFunctionget_world_size
()
maskrcnn_benchmark/utils/comm.py:15
↓ 2 callersMethodhas_field
(self, field)
maskrcnn_benchmark/structures/bounding_box.py:45
↓ 2 callersFunctioninference
( model, data_loader, dataset_name, iou_types=("bbox",), box_only=Fals
maskrcnn_benchmark/engine/inference.py:54
↓ 2 callersMethodkey_is_deprecated
Test if a key is deprecated.
maskrcnn_benchmark/config/yacs.py:259
↓ 2 callersMethodkey_is_renamed
Test if a key is renamed.
maskrcnn_benchmark/config/yacs.py:266
↓ 2 callersFunctionload_state_dict
(model, loaded_state_dict)
maskrcnn_benchmark/utils/model_serialization.py:71
↓ 2 callersFunctionmake_rpn_postprocessor
(config, rpn_box_coder, is_train)
maskrcnn_benchmark/modeling/rpn/inference.py:183
↓ 2 callersMethodraise_key_rename_error
(self, full_key)
maskrcnn_benchmark/config/yacs.py:270
↓ 2 callersFunctionsetup_logger
(name, save_dir, distributed_rank)
maskrcnn_benchmark/utils/logger.py:7
↓ 2 callersFunctionsmooth_l1_loss
very similar to the smooth_l1_loss from pytorch, but with the extra beta parameter
maskrcnn_benchmark/layers/smooth_l1_loss.py:6
↓ 1 callersFunctionROIAlign_forward_cpu
maskrcnn_benchmark/csrc/cpu/ROIAlign_cpu.cpp:221
↓ 1 callersMethod__init__
(self, threshold=0.5, padding=1)
maskrcnn_benchmark/modeling/roi_heads/mask_head/inference.py:163
↓ 1 callersMethod__init__
(self, buffers=None)
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:16
↓ 1 callersMethod__init__
( self, model, optimizer=None, scheduler=None, save_dir="", sa
maskrcnn_benchmark/utils/checkpoint.py:14
↓ 1 callersMethod__init__
(self, optimizer, last_epoch=-1)
maskrcnn_benchmark/solver/lr_scheduler.py:7
↓ 1 callersMethod__len__
(self)
maskrcnn_benchmark/data/datasets/coco.py:84
↓ 1 callersFunction_accumulate_predictions_from_multiple_gpus
(predictions_per_gpu)
maskrcnn_benchmark/engine/inference.py:32
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
maskrcnn_benchmark/modeling/backbone/hrnet.py:118
↓ 1 callersFunction_check_size_scale_factor
(dim)
maskrcnn_benchmark/layers/misc.py:75
↓ 1 callersFunction_compute_aspect_ratios
(dataset)
maskrcnn_benchmark/data/build.py:76
↓ 1 callersMethod_copy_extra_fields
(self, bbox)
maskrcnn_benchmark/structures/bounding_box.py:51
↓ 1 callersFunction_decode
(encoded_data)
maskrcnn_benchmark/utils/comm.py:73
↓ 1 callersFunction_encode
(encoded_data, data)
maskrcnn_benchmark/utils/comm.py:59
↓ 1 callersMethod_forward_test
(self, anchors, objectness, rpn_box_regression)
maskrcnn_benchmark/modeling/rpn/rpn.py:162
↓ 1 callersMethod_forward_train
(self, anchors, objectness, rpn_box_regression, targets)
maskrcnn_benchmark/modeling/rpn/rpn.py:139
↓ 1 callersMethod_freeze_backbone
(self, freeze_at)
maskrcnn_benchmark/modeling/backbone/resnet.py:106
↓ 1 callersFunction_generate_anchors
Generate anchor (reference) windows by enumerating aspect ratios X scales wrt a reference (0, 0, base_size - 1, base_size - 1) window.
maskrcnn_benchmark/modeling/rpn/anchor_generator.py:207
↓ 1 callersMethod_init_weights
(self)
maskrcnn_benchmark/modeling/backbone/hrfpn.py:39
↓ 1 callersFunction_load_C_extensions
()
maskrcnn_benchmark/layers/_utils.py:14
↓ 1 callersFunction_load_c2_pickled_weights
(file_path)
maskrcnn_benchmark/utils/c2_model_loading.py:121
↓ 1 callersFunction_load_cfg_from_file
Load a config from a YAML file or a Python source file.
maskrcnn_benchmark/config/yacs.py:305
next →1–100 of 431, ranked by callers