MCPcopy Create free account

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

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

↓ 1 callersMethod_partition_max_cluster
Partition the largest cluster into two sub-clusters.
OpenSelfSup/openselfsup/models/memories/odc_memory.py:176
↓ 1 callersFunction_paste_mask_lists_in_image
Paste a list of masks that are of various resolutions (e.g., 28 x 28) into an image. The location, height, and width for pasting each mask is
OpenSelfSup/benchmarks/detection/detection/projects/TensorMask/tensormask/arch.py:136
↓ 1 callersFunction_postprocess
Post-process the output boxes for TensorMask. The input images are often resized when entering an object detector. As a result, we often
OpenSelfSup/benchmarks/detection/detection/projects/TensorMask/tensormask/arch.py:182
↓ 1 callersMethod_redirect_empty_clusters
Re-direct empty clusters.
OpenSelfSup/openselfsup/models/memories/odc_memory.py:197
↓ 1 callersMethod_reduce_pred_densepose
(self, outputs, tfms)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/test_time_augmentation.py:114
↓ 1 callersMethod_register_module
Register a module. Args: module (:obj:`nn.Module`): Module to be registered.
OpenSelfSup/openselfsup/utils/registry.py:29
↓ 1 callersFunction_rename_HRNet_weights
(weights)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/densepose_checkpoint.py:7
↓ 1 callersMethod_resize
(self, mask, matrix, w, h)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/vis/base.py:59
↓ 1 callersMethod_rot90
(self, img, num)
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_roi_align_rotated.py:24
↓ 1 callersMethod_run_stage
Args: features (list[Tensor]): #lvl input features to ROIHeads proposals (list[Instances]): #image Instances, with th
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/roi_heads/cascade_rcnn.py:258
↓ 1 callersFunction_set
(key_seq: List[str], val: str)
OpenSelfSup/benchmarks/detection/detection/detectron2/config/compat.py:120
↓ 1 callersMethod_setup_figure
Args: Same as in :meth:`__init__()`. Returns: fig (matplotlib.pyplot.figure): top level container for all th
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:256
↓ 1 callersMethod_simple_roialign_with_grad
(self, img, box, resolution, device)
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_roi_align.py:79
↓ 1 callersMethod_tasks_from_config
Returns: tuple[str]: tasks that can be evaluated under the given configuration.
OpenSelfSup/benchmarks/detection/detection/detectron2/evaluation/coco_evaluation.py:97
↓ 1 callersMethod_tasks_from_config
Returns: tuple[str]: tasks that can be evaluated under the given configuration.
OpenSelfSup/benchmarks/detection/detection/detectron2/evaluation/lvis_evaluation.py:57
↓ 1 callersMethod_test_eval
(self, sizes)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/tests/test_model_e2e.py:33
↓ 1 callersMethod_test_eval
(self, input_sizes)
OpenSelfSup/benchmarks/detection/detection/tests/modeling/test_model_e2e.py:56
↓ 1 callersMethod_write_metrics
Args: metrics_dict (dict): dict of scalar metrics
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:269
↓ 1 callersFunctionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
utils/simclr-converter/eval.py:162
↓ 1 callersFunctionactivation_count_operators
Implement operator-level activations counting using jit. This is a wrapper of fvcore.nn.activation_count, that supports standard detection mo
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/analysis.py:78
↓ 1 callersFunctionadd_batch_counter_hook_function
(module)
OpenSelfSup/openselfsup/utils/flops_counter.py:401
↓ 1 callersFunctionadd_batch_counter_variables_or_reset
(module)
OpenSelfSup/openselfsup/utils/flops_counter.py:397
↓ 1 callersFunctionadd_bootstrap_config
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/config.py:18
↓ 1 callersFunctionadd_dataset_category_config
Add config for additional category-related dataset options - category whitelisting - category mapping
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/config.py:7
↓ 1 callersFunctionadd_densepose_head_config
Add config for densepose head.
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/config.py:70
↓ 1 callersMethodadd_edge
(self, u, v)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:789
↓ 1 callersFunctionadd_flops_counting_methods
(net_main_module)
OpenSelfSup/openselfsup/utils/flops_counter.py:157
↓ 1 callersFunctionadd_ground_truth_to_proposals_single_image
Augment `proposals` with ground-truth boxes from `gt_boxes`. Args: Same as `add_ground_truth_to_proposals`, but with gt_boxes and pr
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/proposal_generator/proposal_utils.py:147
↓ 1 callersFunctionadd_panoptic_deeplab_config
Add config for Panoptic-DeepLab.
OpenSelfSup/benchmarks/detection/detection/projects/Panoptic-DeepLab/panoptic_deeplab/config.py:8
↓ 1 callersFunctionadd_pointrend_config
Add config for PointRend.
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/point_rend/config.py:7
↓ 1 callersFunctionadd_tensormask_config
Add config for TensorMask.
OpenSelfSup/benchmarks/detection/detection/projects/TensorMask/tensormask/config.py:7
↓ 1 callersFunctionadd_tridentnet_config
Add config for tridentnet.
OpenSelfSup/benchmarks/detection/detection/projects/TridentNet/tridentnet/config.py:7
↓ 1 callersMethodafter_step
Called after each iteration.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:79
↓ 1 callersMethodafter_step
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:171
↓ 1 callersMethodafter_train
Called after the last iteration.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:67
↓ 1 callersMethodafter_train
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:158
↓ 1 callersFunctionaggregate_stats
(opts)
OpenSelfSup/benchmarks/svm_tools/aggregate_low_shot_svm_stats.py:39
↓ 1 callersMethodapply_deltas
Apply transformation `deltas` (dx, dy, dw, dh, da) to `boxes`. Args: deltas (Tensor): transformation deltas of shape (N,
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/box_regression.py:177
↓ 1 callersMethodapply_image
(self, img, interp=None)
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/point_rend/color_augmentation.py:49
↓ 1 callersFunctionassign_boxes_to_levels
Map each box in `box_lists` to a feature map level index and return the assignment vector. Args: box_lists (list[Boxes] | list[R
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/poolers.py:22
↓ 1 callersMethodassign_labels
(self, labels)
OpenSelfSup/openselfsup/datasets/deepcluster.py:26
↓ 1 callersMethodauto_scale_workers
When the config is defined for certain number of workers (according to ``cfg.SOLVER.REFERENCE_WORLD_SIZE``) that's different from the
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:543
↓ 1 callersMethodbbox
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/visualizer.py:143
↓ 1 callersMethodbefore_step
Called before each iteration.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:73
↓ 1 callersMethodbefore_step
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:163
↓ 1 callersMethodbefore_train
Called before the first iteration.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:61
↓ 1 callersMethodbefore_train
(self)
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/train_loop.py:154
↓ 1 callersFunctionbenchmark_paste
()
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_mask_ops.py:156
↓ 1 callersFunctionbenchmark_roi_align
()
OpenSelfSup/benchmarks/detection/detection/tests/layers/test_roi_align.py:114
↓ 1 callersFunctionbenchmark_rotated_iou
()
OpenSelfSup/benchmarks/detection/detection/tests/structures/test_rotated_boxes.py:317
↓ 1 callersFunctionbilinear_interpolate_gradient
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/csrc/ROIAlign/ROIAlign_cpu.cpp:221
↓ 1 callersFunctionbilinear_interpolate_gradient
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/csrc/ROIAlignRotated/ROIAlignRotated_cpu.cpp:132
↓ 1 callersFunctionbox_iou_rotated_cpu
OpenSelfSup/benchmarks/detection/detection/detectron2/layers/csrc/box_iou_rotated/box_iou_rotated_cpu.cpp:23
↓ 1 callersMethodbox_reg_loss
Compute the smooth L1 loss for box regression. Returns: scalar Tensor
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/roi_heads/fast_rcnn.py:233
↓ 1 callersMethodbrightness
(self, img)
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/point_rend/color_augmentation.py:70
↓ 1 callersFunctionbuild_backbone
Build backbone.
OpenSelfSup/openselfsup/models/builder.py:46
↓ 1 callersFunctionbuild_densepose_data_filter
Build DensePose data filter which selects data for training Args: cfg (CfgNode): configuration options Return: Callable
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/build.py:26
↓ 1 callersFunctionbuild_densepose_head
Build DensePose head based on configurations options Args: cfg (CfgNode): configuration options input_channels (int): input
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/build.py:42
↓ 1 callersFunctionbuild_densepose_losses
Build DensePose loss based on configurations options Args: cfg (CfgNode): configuration options Return: An instance of D
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/build.py:58
↓ 1 callersFunctionbuild_densepose_predictor
Create an instance of DensePose predictor based on configuration options. Args: cfg (CfgNode): configuration options input_c
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/build.py:10
↓ 1 callersMethodbuild_evaluator
Create evaluator(s) for a given dataset. This uses the special metadata "evaluator_type" associated with each builtin dataset.
OpenSelfSup/benchmarks/detection/detection/tools/train_net.py:52
↓ 1 callersMethodbuild_evaluator
Returns: DatasetEvaluator or None It is not implemented by default.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:474
↓ 1 callersMethodbuild_evaluator
(cls, cfg: CfgNode, dataset_name, output_folder=None)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/engine/trainer.py:68
↓ 1 callersMethodbuild_hooks
Build a list of default hooks, including timing, evaluation, checkpointing, lr scheduling, precise BN, writing events. Retur
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:328
↓ 1 callersFunctionbuild_ins_embed_branch
Build a instance embedding branch from `cfg.MODEL.INS_EMBED_HEAD.NAME`.
OpenSelfSup/benchmarks/detection/detection/projects/Panoptic-DeepLab/panoptic_deeplab/panoptic_seg.py:327
↓ 1 callersFunctionbuild_keypoint_head
Build a keypoint head from `cfg.MODEL.ROI_KEYPOINT_HEAD.NAME`.
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/roi_heads/keypoint_head.py:32
↓ 1 callersMethodbuild_lr_scheduler
It now calls :func:`detectron2.solver.build_lr_scheduler`. Overwrite it if you'd like a different scheduler.
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:444
↓ 1 callersMethodbuild_optimizer
Returns: torch.optim.Optimizer: It now calls :func:`detectron2.solver.build_optimizer`. Overwrite it if you'd li
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:433
↓ 1 callersFunctionbuild_pose_hrnet_backbone
(cfg, input_shape: ShapeSpec)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/modeling/hrnet.py:471
↓ 1 callersFunctionbuild_rpn_head
Build an RPN head defined by `cfg.MODEL.RPN.HEAD_NAME`.
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/proposal_generator/rpn.py:59
↓ 1 callersFunctionbuild_sem_seg_train_aug
(cfg)
OpenSelfSup/benchmarks/detection/detection/projects/Panoptic-DeepLab/train_net.py:34
↓ 1 callersFunctionbuild_sem_seg_train_aug
(cfg)
OpenSelfSup/benchmarks/detection/detection/projects/DeepLab/train_net.py:23
↓ 1 callersFunctionbuild_sem_seg_train_aug
(cfg)
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/train_net.py:31
↓ 1 callersMethodbuild_test_loader
Returns: iterable It now calls :func:`detectron2.data.build_detection_test_loader`. Overwrite it if you'd like a
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:463
↓ 1 callersMethodbuild_writers
Build a list of writers to be used. By default it contains writers that write metrics to the screen, a json file, and a tenso
OpenSelfSup/benchmarks/detection/detection/detectron2/engine/defaults.py:375
↓ 1 callersMethodc2_postprocess
(im_info, rpn_rois, rpn_roi_probs, tensor_mode)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/c10.py:248
↓ 1 callersMethodc2_preprocess
(box_lists)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/c10.py:266
↓ 1 callersFunctioncalculate_ap
Computes the AP under the precision recall curve.
OpenSelfSup/benchmarks/svm_tools/svm_helper.py:68
↓ 1 callersFunctioncalculate_uncertainty
For each location of the prediction `sem_seg_logits` we estimate uncerainty as the difference between top first and top second predicted
OpenSelfSup/benchmarks/detection/detection/projects/PointRend/point_rend/semantic_seg.py:19
↓ 1 callersMethodcat
Concatenates a list of BitMasks into a single BitMasks Arguments: bitmasks_list (list[BitMasks]) Returns:
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/masks.py:221
↓ 1 callersMethodcenter_losses
(self, predictions, targets, weights)
OpenSelfSup/benchmarks/detection/detection/projects/Panoptic-DeepLab/panoptic_deeplab/panoptic_seg.py:502
↓ 1 callersMethodclear_histograms
Delete all the stored histograms for visualization. This should be called after histograms are written to tensorboard.
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:469
↓ 1 callersMethodclear_images
Delete all the stored images for visualization. This should be called after images are written to tensorboard.
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/events.py:462
↓ 1 callersMethodclone
Clone the RotatedBoxes. Returns: RotatedBoxes
OpenSelfSup/benchmarks/detection/detection/detectron2/structures/rotated_boxes.py:223
↓ 1 callersFunctioncocofy_lvis
Filter LVIS instance segmentation annotations to remove all categories that are not included in COCO. The new json files can be used to evalu
OpenSelfSup/benchmarks/detection/detection/datasets/prepare_cocofied_lvis.py:96
↓ 1 callersFunctioncollect_torch_env
()
OpenSelfSup/benchmarks/detection/detection/detectron2/utils/collect_env.py:17
↓ 1 callersFunctioncompress_quantized_densepose_chart_result
Compresses quantized DensePose chart-based result Args: result (DensePoseChartResultQuantized): quantized DensePose chart-based resu
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/structures/chart_result.py:78
↓ 1 callersMethodcomputeDPIoU
(self, imgId, catId)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/densepose_coco_evaluation.py:365
↓ 1 callersMethodcompute_iou_dt_gt
(self, dt, gt, is_crowd)
OpenSelfSup/benchmarks/detection/detection/detectron2/evaluation/rotated_coco_evaluation.py:57
↓ 1 callersFunctionconstruct_init_net_from_params
Construct the init_net from params dictionary
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:290
↓ 1 callersFunctioncontain_targets
(op_ssa)
OpenSelfSup/benchmarks/detection/detection/detectron2/export/shared.py:701
↓ 1 callersMethodcontext_to_image_bgr
(self, context)
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/vis/densepose_results.py:100
↓ 1 callersFunctionconv_ws_2d
(input, weight, bias=None, stride=1, padding=0,
OpenSelfSup/openselfsup/models/utils/conv_ws.py:5
↓ 1 callersFunctionconvert
(input, output)
OpenSelfSup/benchmarks/detection/detection/datasets/prepare_ade20k_sem_seg.py:11
↓ 1 callersMethodconvert
Convert DensePose predictor outputs to DensePoseResult using some registered converter. Does recursive lookup for base classes, so th
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/densepose/converters/to_chart_result.py:21
↓ 1 callersFunctionconvert_basic_c2_names
Apply some basic name conversion to names in C2 weights. It only deals with typical backbone models. Args: original_keys (list[s
OpenSelfSup/benchmarks/detection/detection/detectron2/checkpoint/c2_model_loading.py:12
↓ 1 callersFunctionconvert_boxes_to_pooler_format
Convert all boxes in `box_lists` to the low-level format used by ROI pooling ops (see description under Returns). Args: box_list
OpenSelfSup/benchmarks/detection/detection/detectron2/modeling/poolers.py:68
↓ 1 callersFunctionconvert_c2_detectron_names
Map Caffe2 Detectron weight names to Detectron2 names. Args: weights (dict): name -> tensor Returns: dict: detectron2 n
OpenSelfSup/benchmarks/detection/detection/detectron2/checkpoint/c2_model_loading.py:68
↓ 1 callersFunctioncreate_argument_parser
()
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/apply_net.py:296
↓ 1 callersFunctioncreate_argument_parser
()
OpenSelfSup/benchmarks/detection/detection/projects/DensePose/query_db.py:227
← previousnext →601–700 of 2,192, ranked by callers