MCPcopy Create free account

hub / github.com/NVlabs/blade / functions

Functions19,095 in github.com/NVlabs/blade

↓ 88 callersMethodnormalize
(x)
sapiens/det/mmdet/models/dense_heads/autoassign_head.py:329
↓ 88 callersMethodupdate
(self, vars: dict, count: int = 1)
mmcv/mmcv/runner/log_buffer.py:47
↓ 86 callersMethodrmtree
Recursively delete a directory tree. Args: dir_path (str or Path): A directory to be removed. Examples: >>>
sapiens/engine/mmengine/fileio/backends/local_backend.py:451
↓ 84 callersMethoddim
(self)
mmcv/mmcv/parallel/data_container.py:111
↓ 84 callersMethodto
(self, device)
aios_repo/util/misc.py:416
↓ 83 callersFunctioninverse_sigmoid
(x, eps=1e-3)
aios_repo/util/misc.py:677
↓ 83 callersFunctionmulti_apply
Apply function to a list of arguments. Note: This function applies the ``func`` to multiple inputs and map the multiple outputs o
sapiens/det/mmdet/models/utils/misc.py:228
↓ 82 callersMethodclose
close an opened object.
sapiens/engine/mmengine/visualization/visualizer.py:1174
↓ 81 callersMethodkeys
(self, )
aios_repo/detrsmpl/core/conventions/segmentation/__init__.py:48
↓ 80 callersMethoddetach
Detach all tensors in data.
sapiens/pose/mmpose/structures/multilevel_pixel_data.py:278
↓ 79 callersMethoddevice
(self)
aios_repo/util/misc.py:451
↓ 79 callersMethodfromfile
Build a Config instance from config file. Args: filename (str or Path): Name of config file. use_predefined_variables
sapiens/engine/mmengine/config/config.py:458
↓ 78 callersFunctionfunc
(input, grid)
mmcv/tests/test_ops/test_onnx.py:120
↓ 78 callersMethodpop
pop property in data and metainfo as the same as python.
sapiens/pose/mmpose/structures/multilevel_pixel_data.py:230
↓ 77 callersFunctionbuild_activation_layer
Build activation layer. Args: cfg (dict): The activation layer config, which should contain: - type (str): Layer type.
sapiens/cv/mmcv/cnn/bricks/activation.py:130
↓ 76 callersMethodnew_ones
Reload ``new_ones`` from self.tensor.
sapiens/det/mmdet/structures/bbox/base_boxes.py:225
↓ 76 callersMethodto
(self, device)
mmhuman3d/core/renderer/torch3d_renderer/uv_renderer.py:107
↓ 75 callersMethoddecode
Decode the predicted bboxes according to prediction and base boxes.
sapiens/det/mmdet/models/task_modules/coders/base_bbox_coder.py:52
↓ 74 callersMethodvalues
Yield the values of the dictionary. If class attribute ``lazy`` is False, the value of ``LazyObject`` or ``LazyAttr`` will be built a
sapiens/engine/mmengine/config/config.py:245
↓ 73 callersMethodclone
Modified from pytorch3d and add `model_type` in __class__.__init__.
mmhuman3d/core/renderer/torch3d_renderer/meshes.py:212
↓ 73 callersMethodjoin_path
r"""Concatenate all file paths. Join one or more filepath components intelligently. The return value is the concatenation of filepath
sapiens/engine/mmengine/fileio/file_client.py:417
↓ 73 callersMethodremove
Remove a file. Args: filepath (str, Path): Path to be removed.
mmcv/mmcv/fileio/file_client.py:1080
↓ 72 callersMethodload
(self, load_path)
aios_repo/datasets/humandata.py:64
↓ 72 callersMethodset
sapiens/cv/mmcv/ops/csrc/pytorch/mlu/mlu_common_helper.cpp:72
↓ 71 callersFunctionconvert_kps
Convert keypoints following the mapping correspondence between src and dst keypoints definition. Supported conventions by now: agora, coco, smplx,
mmhuman3d/core/conventions/keypoints_mapping/__init__.py:97
↓ 71 callersMethodndim
mmcv/mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h:358
↓ 71 callersMethodndim
sapiens/cv/mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h:359
↓ 68 callersFunctionbuild_norm_layer
Build normalization layer. Args: cfg (dict): The norm layer config, which should contain: - type (str): Layer type.
sapiens/pretrain/mmpretrain/models/utils/norm.py:121
↓ 66 callersMethodcpu
Overrides this method to set the :attr:`device` Returns: nn.Module: The model itself.
sapiens/det/mmdet/models/data_preprocessors/data_preprocessor.py:586
↓ 66 callersMethodsave
(cls, save_path, data_list, data_strategy)
aios_repo/datasets/humandata.py:73
↓ 65 callersMethodnumpy
Convert all tensor to np.narray in data.
sapiens/pose/mmpose/structures/multilevel_pixel_data.py:282
↓ 65 callersMethodto
(self, device)
aios_repo/detrsmpl/core/renderer/torch3d_renderer/uv_renderer.py:106
↓ 64 callersMethoddata
sapiens/cv/mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h:824
↓ 64 callersMethodget
Get the value of the key. If class attribute ``lazy`` is True, the LazyObject will be built and returned. Args: key (str)
sapiens/engine/mmengine/config/config.py:185
↓ 63 callersFunctionbuild_conv_layer
Build convolution layer. Args: cfg (None or dict): The conv layer config, which should contain: - type (str): Layer type.
mmcv/mmcv/cnn/bricks/conv.py:35
↓ 63 callersMethodfromfile
(filename, use_predefined_variables=True, import_custom_modules=True)
mmcv/mmcv/utils/config.py:358
↓ 63 callersFunctionget_dist_info
()
mmcv/mmcv/runner/dist_utils.py:142
↓ 62 callersFunctionbuild_norm_layer
Build normalization layer. Args: cfg (dict): The norm layer config, which should contain: - type (str): Layer type.
mmcv/mmcv/cnn/bricks/norm.py:95
↓ 62 callersMethodrun
(self, data_loaders, workflow, **kwargs)
mmcv/mmcv/runner/base_runner.py:218
↓ 61 callersFunctionimages_to_levels
Convert targets by image to targets by feature level. [target_img0, target_img1] -> [target_level0, target_level1, ...]
sapiens/det/mmdet/models/utils/misc.py:455
↓ 61 callersFunctionresize
(input, size=None, scale_factor=None, mode='nearest', align_corner
sapiens/seg/mmseg/models/utils/wrappers.py:36
↓ 60 callersFunctionnormal_init
(module, mean=0, std=1, bias=0)
sapiens/engine/mmengine/model/weight_init.py:93
↓ 60 callersMethodunbind
Reload ``unbind`` from self.tensor.
sapiens/det/mmdet/structures/bbox/base_boxes.py:326
↓ 58 callersMethodempty
Whether the buffer is empty or not.
sapiens/det/mmdet/models/trackers/base_tracker.py:67
↓ 58 callersMethodregister_module
Register a module. A record will be added to ``self._module_dict``, whose key is the class name or the specified name, and value is t
sapiens/engine/mmengine/registry/registry.py:643
↓ 57 callersMethodadd
(self, module: Any, name: Optional[str] = None)
sapiens/cv/mmcv/ops/sparse_modules.py:152
↓ 57 callersMethodconcatenate
Concatenate two human_data. All keys will be kept it the returned human_data. If either value from human_data_0 or human_data_1 matches
aios_repo/detrsmpl/data/data_structures/human_data.py:1197
↓ 57 callersFunctionconstant_init
(module, val, bias=0)
sapiens/engine/mmengine/model/weight_init.py:75
↓ 57 callersMethodisdir
Check whether a file path is a directory. Args: filepath (str or Path): Path to be checked whether it is a direct
mmcv/mmcv/fileio/file_client.py:1099
↓ 56 callersFunctionget_file_backend
Return a file backend based on the prefix of uri or backend_args. Args: uri (str or Path): Uri to be parsed that contains the file prefix
sapiens/engine/mmengine/fileio/io.py:131
↓ 55 callersMethodbuild
(self, *args, **kwargs)
mmcv/mmcv/utils/registry.py:259
↓ 55 callersMethodto
Convert values in numpy.ndarray type to torch.Tensor, and move Tensors to the target device. All keys will exist in the returned dict.
aios_repo/detrsmpl/data/data_structures/human_data.py:406
↓ 55 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
sapiens/engine/mmengine/model/weight_init.py:688
↓ 55 callersMethodupdate
Update a new data sample.
aios_repo/detrsmpl/utils/demo_utils.py:751
↓ 54 callersMethodcall_hook
Call all hooks. Args: fn_name (str): The function name in each hook to be called, such as "before_train_epoch".
sapiens/engine/mmengine/runner/runner.py:1847
↓ 54 callersMethoddevice
Reload ``device`` from self.tensor.
sapiens/det/mmdet/structures/bbox/base_boxes.py:243
↓ 54 callersMethoditems
(self)
aios_repo/util/smplx/smplx/utils.py:72
↓ 53 callersMethodbackward
(ctx, a=None)
mmcv/mmcv/ops/knn.py:97
↓ 53 callersMethoddownsample
Downsample mesh.
blade/structures/meshes/utils.py:388
↓ 53 callersFunctionto_tensor
Convert objects of various python types to :obj:`torch.Tensor`. Supported types are: :class:`numpy.ndarray`, :class:`torch.Tensor`, :class:`S
sapiens/cv/mmcv/transforms/formatting.py:40
↓ 51 callersFunctionbuild_loss
Build loss.
aios_repo/detrsmpl/models/losses/builder.py:79
↓ 51 callersMethoditems
(self, )
aios_repo/detrsmpl/core/conventions/segmentation/__init__.py:45
↓ 51 callersMethodnew_empty
Reload ``new_empty`` from self.tensor.
sapiens/det/mmdet/structures/bbox/base_boxes.py:221
↓ 51 callersMethodstep
Performs a dilation searching step. Args: model (nn.Module): pytorch model work_dir (str): Directory to save the sear
sapiens/cv/mmcv/cnn/rfsearch/search.py:137
↓ 50 callersMethodisdir
Check whether a file path is a directory. Args: filepath (str or Path): Path to be checked whether it is a direct
sapiens/engine/mmengine/fileio/file_client.py:392
↓ 50 callersMethodset_field
Special method for set union field, used as property.setter functions.
sapiens/pose/mmpose/structures/multilevel_pixel_data.py:144
↓ 50 callersFunctionunmap
Unmap a subset of item (data) back to the original set of items (of size count)
sapiens/det/mmdet/models/utils/misc.py:250
↓ 49 callersMethodextend
Create new camera class which contains each input camera N times. Args: N: number of new copies of each camera. Returns:
aios_repo/detrsmpl/core/cameras/cameras.py:244
↓ 49 callersFunctionrotmat_to_aa
Convert rotation matrixs to axis angles. Args: matrix (Union[torch.Tensor, numpy.ndarray]): input shape should be (..., 3
aios_repo/detrsmpl/utils/transforms.py:374
↓ 49 callersMethodset_metainfo
Set or change key-value pairs in ``metainfo_field`` by parameter ``metainfo``. Args: metainfo (dict): A dict contains the
sapiens/engine/mmengine/structures/base_data_element.py:248
↓ 49 callersMethodzero_grad
A wrapper of ``Optimizer.zero_grad``.
sapiens/engine/mmengine/optim/optimizer/base.py:64
↓ 48 callersMethodinfo
dict: a dictionary of info about the object
aios_repo/models/utils.py:218
↓ 48 callersMethodtile
(x, dim, n_tile)
sapiens/pretrain/mmpretrain/models/heads/vqa_head.py:208
↓ 47 callersMethoditems
(self)
smplx_repo/smplx/utils.py:75
↓ 47 callersMethodread
Read the next frame. If the next frame have been decoded before and in the cache, then return it directly, otherwise decode, cache an
sapiens/cv/mmcv/video/io.py:160
↓ 47 callersFunctionreduce_mean
Obtain the mean of tensor on different GPUs.
sapiens/det/mmdet/utils/dist_utils.py:87
↓ 47 callersFunctionto_numpy
Convert torch tensor to numpy.ndarray. Args: x (Tensor | Sequence[Tensor]): A single tensor or a sequence of tensors
sapiens/pose/mmpose/utils/tensor_utils.py:39
↓ 47 callersMethodupdate
Override this method to make sure the LazyObject will not be built during updating.
sapiens/engine/mmengine/config/config.py:211
↓ 47 callersMethodwrite
Write image data. Args: grid (Optional[List], optional): Grid to write. Defaults to None. data (Optional[np.ndarray],
sapiens/seg/mmseg/apis/remote_sense_inferencer.py:89
↓ 46 callersMethoddump
Dump config to file or return config text. Args: file (str or Path, optional): If not specified, then the object is d
sapiens/engine/mmengine/config/config.py:1572
↓ 46 callersMethodisfile
Check whether a file path is a file. Args: filepath (str or Path): Path to be checked whether it is a file. Returns:
sapiens/engine/mmengine/fileio/file_client.py:405
↓ 45 callersFunctionbbox2roi
Convert a list of bboxes to roi format. Args: bbox_list (List[Union[Tensor, :obj:`BaseBoxes`]): a list of bboxes correspondin
sapiens/det/mmdet/structures/bbox/transforms.py:110
↓ 44 callersMethodload_state_dict
Loads the loss_scaler state dict. Args: state_dict (dict): scaler state.
mmcv/mmcv/runner/fp16_utils.py:443
↓ 44 callersMethodsample
Sample positive and negative bboxes. This is a simple implementation of bbox sampling given candidates, assigning results and ground
sapiens/det/mmdet/models/task_modules/samplers/base_sampler.py:77
↓ 43 callersMethodset_data
Set or change key-value pairs in ``data_field`` by parameter ``data``. Args: data (dict): A dict contains annotations of
sapiens/engine/mmengine/structures/base_data_element.py:263
↓ 42 callersMethodmerge_from_dict
Merge list into cfg_dict. Merge the dict parsed by MultipleKVAction into this cfg. Args: options (dict): dict of configs
sapiens/engine/mmengine/config/config.py:1598
↓ 41 callersFunctionbuild_from_cfg
Build a module from config dict when it is a class configuration, or call a function from config dict when it is a function configuration. Ex
mmcv/mmcv/utils/registry.py:33
↓ 41 callersFunctiondump
Dump data to json/yaml/pickle strings or files. This method provides a unified api for dumping data as strings or to files, and also supports
sapiens/engine/mmengine/fileio/io.py:892
↓ 41 callersMethodseed
int: A number to set random modules.
sapiens/engine/mmengine/runner/runner.py:584
↓ 40 callersMethoddump
Dumps config into a file or returns a string representation of the config. If a file argument is given, saves the config to that file
mmcv/mmcv/utils/config.py:586
↓ 39 callersMethoddump
Dump keys and items to an npz file. Args: npz_path (str): Path to a dumped npz file. overwrite (bool,
mmhuman3d/data/data_structures/human_data.py:266
↓ 39 callersFunctionmkdir_or_exist
(dir_name, mode=0o777)
sapiens/engine/mmengine/utils/path.py:54
↓ 38 callersMethodpush_back
sapiens/cv/mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h:147
↓ 38 callersFunctionrot6d_to_rotmat
Convert 6D rotation representation to 3x3 rotation matrix. Based on Zhou et al., "On the Continuity of Rotation Representations in Neural Net
aios_repo/detrsmpl/utils/geometry.py:76
↓ 38 callersMethodsqueeze
mmcv/mmcv/ops/csrc/common/utils/spconv/tensorview/tensorview.h:359
↓ 38 callersMethodwrite
(self, image_array: np.ndarray)
mmhuman3d/utils/ffmpeg_utils.py:98
↓ 37 callersFunction_build_demo_runner
(runner_type='EpochBasedRunner', max_epochs=1, max_iters=None,
mmcv/tests/test_runner/test_hooks.py:1673
↓ 37 callersMethod_transform_inputs
Transform inputs for decoder. Args: inputs (list[Tensor]): List of multi-level img features. Returns: Tensor
sapiens/seg/mmseg/models/decode_heads/decode_head.py:237
↓ 37 callersMethodisfile
Check whether a file path is a file. Args: filepath (str or Path): Path to be checked whether it is a file. Returns:
mmcv/mmcv/fileio/file_client.py:1112
↓ 37 callersFunctionjoin_path
r"""Concatenate all file paths. Join one or more filepath components intelligently. The return value is the concatenation of filepath and any
sapiens/engine/mmengine/fileio/io.py:368
← previousnext →101–200 of 19,095, ranked by callers