MCPcopy Create free account

hub / github.com/cjrd/self-supervised-pretraining / functions

Functions2,192 in github.com/cjrd/self-supervised-pretraining

↓ 126 callersMethodto
(self, *args: Any, **kwargs: Any)
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:170
↓ 80 callersMethodget
Returns the field called `name`.
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/instances.py:94
↓ 75 callersMethodcat
Concatenates a list of Boxes into a single Boxes Arguments: boxes_list (list[Boxes]) Returns: Boxes
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:279
↓ 69 callersMethodcuda
(self)
OpenSelfSup/openselfsup/utils/alias_multinomial.py:51
↓ 52 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/wrappers.py:20
↓ 52 callersMethodget
(self)
OpenSelfSup/benchmarks/detection/detection/demo/predictor.py:191
↓ 44 callersMethodclone
Clone the Boxes. Returns: Boxes
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:160
↓ 42 callersMethoddevice
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:307
↓ 42 callersMethodupdate
Update features in memory bank. Args: ind (Tensor): Indices for the batch of features. feature (Tensor): Batch of fea
OpenSelfSup/openselfsup/models/memories/simple_memory.py:29
↓ 39 callersFunctionget_norm
Args: norm (str or callable): either one of BN, SyncBN, FrozenBN, GN; or a callable that takes a channel number and returns
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/batch_norm.py:127
↓ 36 callersFunctionget_cfg
Get a copy of the default config. Returns: a detectron2 CfgNode instance.
OpenSelfSup/benchmarks/detection/detection/detectron2/config/config.py:79
↓ 36 callersMethodsave
Args: filepath (str): a string that contains the absolute path, including the file name, where the visualized ima
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:283
↓ 34 callersMethodwrite
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:103
↓ 32 callersMethodget
(self, key)
OpenSelfSup/openselfsup/utils/registry.py:26
↓ 32 callersMethodhas
(self, name)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/c10.py:72
↓ 28 callersMethodput_scalar
Add a scalar `value` to the `HistoryBuffer` associated with `name`. Args: smoothing_hint (bool): a 'hint' on whether thi
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:297
↓ 24 callersMethodflatten
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/c10.py:97
↓ 21 callersMethodfreeze
Make this block not trainable. This method sets all parameters to `requires_grad=False`, and convert all BatchNorm layers to
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/blocks.py:36
↓ 21 callersMethodmerge_from_file
(self, cfg_filename: str, allow_unsafe: bool = True)
OpenSelfSup/benchmarks/detection/detection/detectron2/config/config.py:24
↓ 21 callersFunctionto_device
This function is a replacement of .to(another_device) such that it allows the casting to be traced properly by explicitly calling the underly
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:25
↓ 20 callersFunctioncheck_set_pb_arg
(pb, arg_name, arg_attr, arg_value, allow_override=False)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:204
↓ 20 callersFunctionget_event_storage
Returns: The :class:`EventStorage` object that's currently being used. Throws an error if no :class:`EventStorage` is currently e
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:24
↓ 20 callersFunctionprint_log
Print a log message. Args: msg (str): The message to be logged. logger (logging.Logger | str | None): The logger to be used. Some
OpenSelfSup/openselfsup/utils/logger.py:43
↓ 20 callersMethodresume_or_load
If `resume==True` and `cfg.OUTPUT_DIR` contains the last checkpoint (defined by a `last_checkpoint` file), resume from the file. Resu
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:308
↓ 18 callersMethodget_image
Returns: ndarray: the visualized image of shape (H, W, 3) (RGB) in uint8 type. The shape is scale
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:291
↓ 18 callersMethodtrain
Args: start_iter, max_iter (int): See docs above
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:126
↓ 17 callersFunctionbuild_norm_layer
Build normalization layer. Args: cfg (dict): cfg should contain: type (str): identify norm layer type. layer args
OpenSelfSup/openselfsup/models/utils/norm.py:12
↓ 17 callersMethodfrom_tensors
Args: tensors: a tuple or list of `torch.Tensors`, each of shape (Hi, Wi) or (C_1, ..., C_K, Hi, Wi) where K >= 1
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/image_list.py:58
↓ 16 callersFunctionalias
(x, name, is_backward=False)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:592
↓ 16 callersMethodclip
Clip (in place) the boxes by limiting x coordinates to the range [0, width] and y coordinates to the range [0, height]. Args
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:184
↓ 16 callersFunctionsetup_logger
Initialize the detectron2 logger and set its verbosity level to "DEBUG". Args: output (str): a file name or a directory to save log.
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/logger.py:35
↓ 15 callersFunctionbuild_model
Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. Note that it does not load any weights from ``cfg``.
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/meta_arch/build.py:15
↓ 15 callersMethoddump
Returns: str: a yaml string representation of the config
OpenSelfSup/benchmarks/detection/detection/detectron2/config/config.py:67
↓ 15 callersMethodoutput_shape
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/backbone/fpn.py:147
↓ 15 callersMethodoverlay_instances
Args: boxes (Boxes, RotatedBoxes or ndarray): either a :class:`Boxes`, or an Nx4 numpy array of XYXY_ABS format f
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:572
↓ 15 callersMethodset
(self, name, value)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/c10.py:75
↓ 14 callersMethodnorm
(self)
OpenSelfSup/openselfsup/models/utils/conv_module.py:146
↓ 13 callersFunctionbuild_conv_layer
Build convolution layer. Args: cfg (None or dict): Cfg should contain: type (str): Identify conv layer type. laye
OpenSelfSup/openselfsup/models/utils/conv_module.py:15
↓ 13 callersMethodconvert
Args: box: can be a k-tuple, k-list or an Nxk array/tensor, where k = 4 or 5 from_mode, to_mode (BoxMode) Re
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:44
↓ 13 callersFunctionpairwise_iou_rotated
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x_center, y_center, width, height, an
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/rotated_boxes.py:7
↓ 11 callersMethodclose
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:125
↓ 11 callersFunctiondefault_argument_parser
Create a parser with some common arguments used by detectron2 users. Args: epilog (str): epilog passed to ArgumentParser describing
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:49
↓ 10 callersFunctionbuild_backbone
Build a backbone from `cfg.MODEL.BACKBONE.NAME`. Returns: an instance of :class:`Backbone`
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/backbone/build.py:20
↓ 10 callersFunctionbuild_from_cfg
Build a module from config dict. Args: cfg (dict): Config dict. It should at least contain the key "type". registry (:obj:`Regist
OpenSelfSup/openselfsup/utils/registry.py:51
↓ 10 callersMethodbuild_model
Returns: torch.nn.Module: It now calls :func:`detectron2.modeling.build_model`. Overwrite it if you'd like a dif
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:419
↓ 10 callersFunctiondefault_setup
Perform some basic common setups at the beginning of a job, including: 1. Set up the detectron2 logger 2. Log basic information about en
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:111
↓ 10 callersFunctionget_pb_arg
(pb, arg_name)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:167
↓ 10 callersMethodinterp2d
Bilinear interpolation method to be used for upscaling Args: tensor_nchw (tensor): tensor of shape (N, C, H, W)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/predictors/chart.py:66
↓ 10 callersFunctionlaunch
Launch multi-gpu or distributed training. This function must be called on all machines involved in the training. It will spawn child proc
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/launch.py:24
↓ 10 callersFunctionpoint_sample
A wrapper around :function:`torch.nn.functional.grid_sample` to support 3D point_coords tensors. Unlike :function:`torch.nn.functional.grid_s
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/point_rend/point_features.py:19
↓ 10 callersMethodset
Set the field named `name` to `value`. The length of `value` must be the number of instances, and must agree with other exist
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/instances.py:68
↓ 10 callersMethodtest
(self)
OpenSelfSup/benchmarks/detection/detection/tests/test_packaging.py:23
↓ 9 callersMethodarea
Computes the area of all the boxes. Returns: torch.Tensor: a vector with areas of each box.
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:173
↓ 9 callersFunctionbatched_nms
Same as torchvision.ops.boxes.batched_nms, but safer.
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/nms.py:10
↓ 9 callersFunctionget_pb_arg_vali
(pb, arg_name, default_val)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:189
↓ 9 callersMethodget_sample
(self, idx)
OpenSelfSup/openselfsup/datasets/data_sources/cifar.py:29
↓ 9 callersFunctionpatch_instances
A contextmanager, under which the Instances class in detectron2 is replaced by a statically-typed scriptable class, defined by `fields`.
OpenSelfSup/benchmarks/detection/detection/detectron2/export/torchscript_patch.py:17
↓ 9 callersMethodremove
Remove the field called `name`.
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/instances.py:88
↓ 9 callersMethodstep
User should either: (1) Call this function to increment storage.iter when needed. Or (2) Set `storage.iter` to the correct iteration
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:415
↓ 8 callersFunction_rename
(cfg: CN, old: str, new: str)
OpenSelfSup/benchmarks/detection/detection/detectron2/config/compat.py:116
↓ 8 callersMethodapply_deltas
Apply transformation `deltas` (dx, dy, dw, dh) to `boxes`. Args: deltas (Tensor): transformation deltas of shape (N, k*4
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/box_regression.py:73
↓ 8 callersFunctiondetector_postprocess
Resize the output instances. The input images are often resized when entering an object detector. As a result, we often need the outputs
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/postprocessing.py:11
↓ 8 callersMethodextract_at_points
Extract ground truth values z_gt for valid point indices and estimated values z_est using bilinear interpolation over top-left (y_lo,
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/losses/utils.py:195
↓ 8 callersMethodloss
(self, cls_score, labels)
OpenSelfSup/openselfsup/models/heads/cls_head.py:60
↓ 8 callersFunctionpermute_to_N_HWA_K
Transpose/reshape a tensor from (N, (Ai x K), H, W) to (N, (HxWxAi), K)
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/meta_arch/retinanet.py:27
↓ 7 callersMethod_random_data
(self)
OpenSelfSup/benchmarks/detection/detection/tests/test_visualizer.py:17
↓ 7 callersMethodbackward
(ctx, grad_output)
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/roi_align.py:38
↓ 7 callersMethodconvert
Convert an instance to the destination type using some registered converter. Does recursive lookup for base classes, so there's no ne
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/converters/base.py:63
↓ 7 callersFunctioncreate_small_table
Create a small table using the keys of small_dict as headers. This is only suitable for small dictionaries. Args: small_dict (di
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/logger.py:204
↓ 7 callersMethoddraw_instance_predictions
Draw instance-level prediction results on an image. Args: predictions (Instances): the output of an instance detection/s
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:359
↓ 7 callersFunctionget_root_logger
Get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHandler will be added. If `log_file` i
OpenSelfSup/openselfsup/utils/logger.py:6
↓ 7 callersFunctionpairwise_iou
Given two lists of rotated boxes of size N and M, compute the IoU (intersection over union) between __all__ N x M pairs of boxes. The
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/rotated_boxes.py:466
↓ 6 callersMethod__init__
(self, in_channels, out_channels, with_avg_pool=True)
OpenSelfSup/openselfsup/models/necks.py:32
↓ 6 callersMethod_create_tensors
(self, N)
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_nms_rotated.py:65
↓ 6 callersFunction_get_base_config_dir
Return the base directory for configurations
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/tests/common.py:20
↓ 6 callersFunction_init_weights
(module, init_linear='normal', std=0.01, bias=0.)
OpenSelfSup/openselfsup/models/necks.py:10
↓ 6 callersFunctionbuild
Build a module. Args: cfg (dict, list[dict]): The config of modules, it is either a dict or a list of configs. regist
OpenSelfSup/openselfsup/models/builder.py:8
↓ 6 callersFunctionbuild_proposal_generator
Build a proposal generator from `cfg.MODEL.PROPOSAL_GENERATOR.NAME`. The name can be "PrecomputedProposals" to use no proposal generator.
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/proposal_generator/build.py:15
↓ 6 callersMethoddraw_binary_mask
Args: binary_mask (ndarray): numpy array of shape (H, W), where H is the image height and W is the image width. E
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:1000
↓ 6 callersFunctionf
()
OpenSelfSup/benchmarks/detection/detection/tools/benchmark.py:104
↓ 6 callersFunctionget_pb_arg_vals
(pb, arg_name, default_val)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:194
↓ 6 callersFunctionis_supported_instance
(module)
OpenSelfSup/openselfsup/utils/flops_counter.py:238
↓ 6 callersFunctionnms_edit_distance
Compare the "keep" result of two nms call. They are allowed to be different in terms of edit distance due to floating point precision iss
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_nms_rotated.py:11
↓ 6 callersFunctionnms_rotated
Performs non-maximum suppression (NMS) on the rotated boxes according to their intersection-over-union (IoU). Rotated NMS iteratively re
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/nms.py:34
↓ 6 callersFunctionnonzero_tuple
A 'as_tuple=True' version of torch.nonzero to support torchscript. because of https://github.com/pytorch/pytorch/issues/38718
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/wrappers.py:233
↓ 6 callersFunctionpairwise_iou
Given two lists of boxes of size N and M, compute the IoU (intersection over union) between __all__ N x M pairs of boxes. The box ord
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/boxes.py:344
↓ 6 callersMethodrun
(self)
OpenSelfSup/benchmarks/detection/detection/demo/predictor.py:149
↓ 6 callersFunctionsem_seg_postprocess
Return semantic segmentation predictions in the original resolution. The input images are often resized when entering semantic segmentor. Mo
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/postprocessing.py:75
↓ 6 callersFunctionslice_if_not_none
(data, item)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/structures/chart_confidence.py:53
↓ 6 callersMethodsummarize
Compute and display summary metrics for evaluation results. Note this function can *only* be applied on the default parameter setting
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/densepose_coco_evaluation.py:891
↓ 6 callersMethodto
(self, *args: Any, **kwargs: Any)
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/masks.py:104
↓ 5 callersMethod__init__
(self, in_channels, atrous_rates, out_channels)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/roi_heads/deeplab.py:109
↓ 5 callersMethod__init__
(self, arg1, arg2, arg3=3)
OpenSelfSup/benchmarks/detection/detection/tests/test_config.py:65
↓ 5 callersFunction_broadcast_params
If one size (or aspect ratio) is specified and there are multiple feature maps, we "broadcast" anchors of that single size (or aspect ratio)
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/anchor_generator.py:52
↓ 5 callersFunction_cast_to_f32
(f64)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/caffe2_modeling.py:98
↓ 5 callersMethod_convert_xy_to_wh
(self, x)
OpenSelfSup/benchmarks/detection/detection/tests/structures/test_boxes.py:13
↓ 5 callersMethod_simple_roialign
RoiAlign with scale 1.0 and 0 sample ratio.
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_roi_align.py:62
↓ 5 callersMethodadd_arguments
(cls: type, parser: argparse.ArgumentParser)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/apply_net.py:43
↓ 5 callersMethodadd_arguments
(cls: type, parser: argparse.ArgumentParser)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/query_db.py:41
↓ 5 callersFunctionbuild_dataloader
Build PyTorch DataLoader. In distributed training, each GPU/process has a dataloader. In non-distributed training, there is only one dataload
OpenSelfSup/openselfsup/datasets/loader/build_loader.py:22
next →1–100 of 2,192, ranked by callers