MCPcopy Create free account

hub / github.com/Megvii-BaseDetection/cvpods / functions

Functions2,734 in github.com/Megvii-BaseDetection/cvpods

↓ 262 callersMethodto
(self, device: str)
cvpods/structures/boxes.py:166
↓ 261 callersMethodjoin
(*paths)
cvpods/utils/file/file_io.py:272
↓ 185 callersMethodcat
Concatenates a list of Boxes into a single Boxes Arguments: boxes_list (list[Boxes]) Returns: Boxes
cvpods/structures/boxes.py:272
↓ 167 callersMethodget
Collects all the annotations / detections for that particular image.
cvpods/analyser/tide/data.py:139
↓ 99 callersMethoddevice
(self)
cvpods/structures/boxes.py:292
↓ 90 callersFunctioncat
Efficient version of torch.cat that avoids a copy if there is only a single element in a list
cvpods/layers/wrappers.py:37
↓ 83 callersFunctionget_norm
Args: norm (str or callable): Returns: nn.Module or None: the normalization layer
cvpods/layers/batch_norm.py:146
↓ 79 callersMethod_register_configuration
(self, config)
cvpods/configs/base_config.py:401
↓ 65 callersMethodwrite
(self)
cvpods/utils/dump/events.py:99
↓ 53 callersMethodpop
(self, k, d=None)
cvpods/configs/base_config.py:276
↓ 46 callersMethodclone
Clone the Boxes. Returns: Boxes
cvpods/structures/boxes.py:157
↓ 44 callersFunctionpermute_to_N_HWA_K
Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
cvpods/modeling/meta_arch/retinanet.py:18
↓ 43 callersMethodupdate
(self, e=None, **f)
cvpods/configs/base_config.py:270
↓ 36 callersMethodhas
Returns: bool: whether the field called `name` exists.
cvpods/structures/instances.py:76
↓ 35 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
cvpods/structures/image_list.py:50
↓ 35 callersMethodoutput_shape
(self)
cvpods/modeling/backbone/fpn.py:170
↓ 35 callersMethodvalues
Returns: list[(number, iteration)]: content of the current buffer.
cvpods/utils/dump/history_buffer.py:67
↓ 32 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
cvpods/structures/boxes.py:39
↓ 29 callersMethod__init__
Args: top (int): number of rows of `pad_value` to add on top. left (int): number of columns of `pad_value` to add on
cvpods/data/transforms/transform_gen.py:1107
↓ 28 callersMethodget_deltas
Get box regression transformation deltas (dx, dy, dw, dh) that can be used to transform the `src_boxes` into the `target_boxes`. That
cvpods/modeling/box_regression.py:35
↓ 27 callersMethod__init__
Args: top (int): number of rows of `pad_value` to add on top. left (int): number of columns of `pad_value` to add on
cvpods/data/transforms/transform.py:915
↓ 27 callersMethodextend
(self, buffers)
cvpods/modeling/anchor_generator.py:34
↓ 27 callersMethodput_scalar
Add a scalar `value` to the `HistoryBuffer` associated with `name`. Args: smoothing_hint (bool): a 'hint' on whether thi
cvpods/utils/dump/events.py:324
↓ 26 callersMethod_init
(self, params=None)
cvpods/data/transforms/transform_gen.py:131
↓ 24 callersFunctionbuild_resnet_fpn_backbone
(cfg, input_shape: ShapeSpec)
cvpods/modeling/backbone/fpn.py:304
↓ 22 callersMethod_set_attributes
Set attributes from the input list of parameters. Args: params (list): list of parameters.
cvpods/data/transforms/transform.py:73
↓ 22 callersFunctionlog_first_n
Log only for the first n times. Args: lvl (int): the logging level msg (str): n (int): key (str or tuple[str
cvpods/utils/dump/logger.py:88
↓ 21 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
cvpods/modeling/postprocessing.py:8
↓ 20 callersMethodload
:meth: read the object from a dict
cvpods/evaluation/crowdhumantools.py:26
↓ 19 callersMethodarea
Computes the area of all the boxes. Returns: torch.Tensor: a vector with areas of each box.
cvpods/structures/boxes.py:169
↓ 19 callersMethodget
(name)
cvpods/checkpoint/catalog.py:61
↓ 19 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
cvpods/structures/boxes.py:304
↓ 19 callersMethodseconds
Returns: (float): the total number of seconds since the start/reset of the timer, excluding the time when the tim
cvpods/utils/benchmark/timer.py:49
↓ 18 callersMethodclip
Clip (in place) the boxes by limiting x coordinates to the range [0, width] and y coordinates to the range [0, height]. Args
cvpods/structures/boxes.py:180
↓ 18 callersFunctionget_event_storage
Returns: The :class:`EventStorage` object that's currently being used. Throws an error if no :class`EventStorage` is currently en
cvpods/utils/dump/events.py:18
↓ 17 callersFunctionbuild_resnet_backbone
Create a ResNet instance from config. Returns: ResNet: a :class:`ResNet` instance.
cvpods/modeling/backbone/resnet.py:671
↓ 17 callersMethodclose
(self)
cvpods/utils/dump/events.py:120
↓ 17 callersMethodsave
Dump model and checkpointables to a file. Args: name (str): name of the file. kwargs (dict): extra arbitrary
cvpods/checkpoint/checkpoint.py:64
↓ 16 callersMethoddecode
r""" decode feature maps, width height, regression to detections results. Args: fmap (Tensor): input feature map.
cvpods/modeling/meta_arch/centernet.py:284
↓ 15 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
cvpods/layers/rotated_boxes.py:10
↓ 14 callersMethodapply_deltas
Apply transformation `deltas` (dx, dy, dw, dh) to `boxes`. Args: deltas (Tensor): transformation deltas of shape (N, k*4
cvpods/modeling/box_regression.py:70
↓ 14 callersFunctionfunc
(device, n=3)
tests/layers/test_mask_ops.py:175
↓ 14 callersMethodget
(self, name: str)
cvpods/utils/registry.py:61
↓ 13 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
cvpods/utils/dump/logger.py:153
↓ 13 callersFunctiongeneralized_batched_nms
(boxes, scores, idxs, iou_threshold, score_threshold=0.001, nms_type="normal", **k
cvpods/layers/nms.py:89
↓ 13 callersMethodregister
Register the given object under the the name `obj.__name__`. Can be used as either a decorator or not. See docstring of this class fo
cvpods/utils/registry.py:40
↓ 12 callersMethod_apply_transforms
Apply a list of :class:`TransformGen` on the input image, and returns the transformed image and a list of transforms. We can
cvpods/data/base_dataset.py:98
↓ 12 callersMethodfind
find a given key and its value in config Args: key (str): the string you want to find show (bool): if show i
cvpods/configs/base_config.py:356
↓ 11 callersMethodassertEqualsArrays
(self, a1, a2)
tests/data/test_rotation_transform.py:9
↓ 11 callersFunctionbuild_retinanet_resnet_fpn_backbone
(cfg, input_shape: ShapeSpec)
cvpods/modeling/backbone/fpn.py:320
↓ 11 callersFunctionensure_dir
(path: str)
cvpods/utils/file/file_io.py:19
↓ 10 callersFunctionCOCO
Loads ground truth from a COCO-style annotation file. If path is not specified, this will download the COCO annotations for the year and
cvpods/analyser/tide/datasets.py:65
↓ 10 callersMethod__init__
(self, stride)
cvpods/modeling/backbone/dynamic_arch/op_with_flops.py:509
↓ 10 callersMethod_set_group_flag
Set flag according to image aspect ratio. Images with aspect ratio greater than 1 will be set as group 1, otherwise group 0.
cvpods/data/base_dataset.py:176
↓ 10 callersMethodapply_image
Apply the transform on an image. Args: img (ndarray): of shape NxHxWxC, or HxWxC or HxW. The array can be
cvpods/data/transforms/transform.py:87
↓ 10 callersFunctionbatched_nms
Same as torchvision.ops.boxes.batched_nms, but safer.
cvpods/layers/nms.py:35
↓ 10 callersMethodiou
(self, dts, gts, pyiscrowd)
cvpods/evaluation/eval_MR_multisetup.py:187
↓ 10 callersFunctionpermute_to_N_HWA_K
Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
playground/detection/coco/free_anchor/free_anchor.res50.fpn.coco.800size.1x/retinanet.py:18
↓ 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
cvpods/modeling/meta_arch/pointrend.py:33
↓ 10 callersFunctionread_image
Read an image into the given format. Will apply rotation and flipping if the image has such exif information. Args: file_name (s
cvpods/data/detection_utils.py:147
↓ 10 callersFunctionsmooth_l1_loss
Smooth L1 loss defined in the Fast R-CNN paper as: | 0.5 * x ** 2 / beta if abs(x) < beta smoothl1(x) = |
cvpods/modeling/losses/smooth_l1_loss.py:5
↓ 9 callersMethod__init__
Extra keyword arguments supported in addition to those in `torch.nn.Conv2d`: Args: norm (nn.Module, optional): a normali
cvpods/layers/wrappers.py:64
↓ 9 callersMethod_create_tensors
(self, N)
tests/layers/test_nms_rotated.py:71
↓ 9 callersMethodconvert_frozen_batchnorm
Convert BatchNorm/SyncBatchNorm in module into FrozenBatchNorm. Args: module (torch.nn.Module): Returns:
cvpods/layers/batch_norm.py:113
↓ 9 callersFunctionget_activation
Args: activation (EasyDict or str): Returns: nn.Module or None: the activation layer
cvpods/layers/batch_norm.py:168
↓ 9 callersMethodget_fields
Returns: dict: a dict which maps names (str) to data of the fields Modifying the returned dict will modify this instance
cvpods/structures/instances.py:95
↓ 9 callersFunctionget_shape
Get the shapes from a jit value object. Args: val (torch._C.Value): jit value object. Returns: list(int): return a list
cvpods/modeling/nn_utils/jit_handles.py:160
↓ 9 callersMethodmerge_from_list
Merge config (keys, values) in a list (e.g., from command line) into this CfgNode. Args: cfg_list (list): cfg_li
cvpods/configs/base_config.py:304
↓ 9 callersMethodnonempty
Find boxes that are non-empty. A box is considered empty, if either of its side is no larger than threshold. Returns:
cvpods/structures/boxes.py:195
↓ 9 callersMethodopen
Open a stream to a URI, similar to the built-in `open`. Args: path (str): A URI supported by this PathHandler R
cvpods/utils/file/file_io.py:116
↓ 9 callersMethodoverlay_instances
Args: boxes (Boxes, RotatedBoxes or ndarray): either a :class:`Boxes`, or an Nx4 numpy array of XYXY_ABS format f
cvpods/utils/visualizer/visualizer.py:507
↓ 9 callersMethodset
Set the field named `name` to `value`. The length of `value` must be the number of instances, and must agree with other exist
cvpods/structures/instances.py:63
↓ 9 callersFunctionsetup_logger
Initialize the cvpods logger and set its verbosity level to "INFO". Args: output (str): a file name or a directory to save log. If N
cvpods/utils/dump/logger.py:25
↓ 9 callersMethodstep
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model and returns the
cvpods/solver/lars_sgd.py:140
↓ 9 callersMethodtrain
Run training. Returns: OrderedDict of results, if evaluation is enabled. Otherwise None.
cvpods/engine/runner.py:279
↓ 8 callersMethod_add_error
(self, error)
cvpods/analyser/tide/quantify.py:224
↓ 8 callersMethod_filter_annotations
Load and prepare dataset dicts for instance detection/segmentation and semantic segmentation. Args: dataset_name
cvpods/data/base_dataset.py:134
↓ 8 callersMethod_make_cbl
cbl = conv + batch_norm + leaky_relu
cvpods/modeling/meta_arch/yolov3.py:81
↓ 8 callersFunctionannotations_to_instances
Create an :class:`Instances` object used by the models, from instance annotations in the dataset dict. Args: annos (list[dict]):
cvpods/data/detection_utils.py:238
↓ 8 callersMethodapply_coords
Apply the transform on coordinates. Args: coords (ndarray): floating point array of shape Nx2. Each row is (x, y).
cvpods/data/transforms/transform.py:101
↓ 8 callersFunctioncheck_image_size
Raise an error if the image does not match the size specified in the dict.
cvpods/data/detection_utils.py:171
↓ 8 callersMethodcopy
Copies a source path to a destination path. Args: src_path (str): A URI supported by this PathHandler dst_pa
cvpods/utils/file/file_io.py:130
↓ 8 callersFunctioncreate_keypoint_hflip_indices
Args: dataset_names (list[str]): list of dataset names Returns: ndarray[int]: a vector of size=#keypoints, storing the
cvpods/data/detection_utils.py:369
↓ 8 callersFunctionfilter_empty_instances
Filter out empty instances in an `Instances` object. Args: instances (Instances): by_box (bool): whether to filter out insta
cvpods/data/detection_utils.py:340
↓ 8 callersMethodget
Returns the field called `name`.
cvpods/structures/instances.py:89
↓ 8 callersMethodget_image
Returns: ndarray: the visualized image of shape (H, W, 3) (RGB) in uint8 type. The shape is scaled w.r.t the input
cvpods/utils/visualizer/visualizer.py:262
↓ 8 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
tests/layers/test_nms_rotated.py:17
↓ 8 callersMethodplot
Plots a summary model for each run in this TIDE object. Images will be outputted to out_dir, which will be created if it doesn't exis
cvpods/analyser/tide/quantify.py:695
↓ 8 callersMethodstride
(self)
cvpods/modeling/backbone/resnet.py:582
↓ 7 callersMethod__init__
Args: stem (nn.Module): a stem module stages (list[list[ResNetBlock]]): several (typically 4) stages,
cvpods/modeling/backbone/resnet.py:587
↓ 7 callersFunction_check_args_tf
(kwargs)
cvpods/data/transforms/auto_aug.py:38
↓ 7 callersMethodbackward
(ctx, grad_output)
cvpods/layers/roi_align.py:29
↓ 7 callersMethodexists
Checks if there is a resource at the given URI. Args: path (str): A URI supported by this PathHandler Returns:
cvpods/utils/file/file_io.py:164
↓ 7 callersMethodforward
(self, x)
cvpods/layers/wrappers.py:81
↓ 7 callersMethodfreeze
(self)
cvpods/modeling/backbone/mobilenet.py:46
↓ 7 callersMethodisfile
Checks if there the resource at the given URI is a file. Args: path (str): A URI supported by this PathHandler
cvpods/utils/file/file_io.py:178
↓ 7 callersMethodmedian
Return the median of the latest `window_size` values in the buffer.
cvpods/utils/dump/history_buffer.py:48
↓ 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
cvpods/structures/rotated_boxes.py:487
↓ 7 callersMethodsummarize
Compute and display summary metrics for evaluation results. Note this functin can *only* be applied on the default parameter setting
cvpods/evaluation/widerfacetools.py:452
↓ 7 callersMethodto
(self, device: str)
cvpods/structures/masks.py:106
next →1–100 of 2,734, ranked by callers