MCPcopy Create free account

hub / github.com/aigc3d/LHM / functions

Functions3,669 in github.com/aigc3d/LHM

↓ 440 callersFunctionprint
(*args, **kwargs)
engine/pose_estimation/dinov2/dinov2/distributed/__init__.py:82
↓ 324 callersMethodget
(self, propId)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:335
↓ 217 callersMethoditems
(self)
LHM/models/rendering/smplx/smplx/utils.py:52
↓ 201 callersMethodcpu
(self)
engine/SegmentAPI/base.py:24
↓ 167 callersMethodfloat
(self)
engine/SegmentAPI/base.py:28
↓ 153 callersMethodto
move all tensor attributes to device. Args: device (torch.device): target device. Returns: Mesh: self.
LHM/models/rendering/mesh_utils.py:690
↓ 148 callersMethodevaluate
(self, results, res_folder=None, logger=None)
engine/pose_estimation/third-party/ViTPose/tests/test_backward_compatibility/test_eval_hook_compatibility.py:25
↓ 109 callersMethodinfo
Returns a dictionary of info about the object.
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:82
↓ 107 callersMethodregister_buffer
(self, name, maxsize=0)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/buffer.py:72
↓ 106 callersMethodmax
(self)
engine/pose_estimation/dinov2/dinov2/logging/helpers.py:180
↓ 90 callersMethodload
load mesh from path. Args: path (str): path to mesh file, supports ply, obj, glb. clean (bool, optional): perform mes
LHM/models/rendering/mesh_utils.py:111
↓ 86 callersMethodpop
(self, *args, **kwargs)
engine/ouputs.py:74
↓ 83 callersMethodto
(self, device)
engine/SegmentAPI/base.py:32
↓ 78 callersMethodeval
(self)
engine/SegmentAPI/base.py:36
↓ 78 callersMethodkeys
(self)
LHM/models/rendering/smplx/smplx/utils.py:44
↓ 77 callersMethodget
(self, key, default=None)
LHM/models/rendering/smplx/smplx/utils.py:38
↓ 65 callersFunctionbuild_norm_layer
( dim, norm_layer, in_format="channels_last", out_format="channels_last", eps=1e-6 )
engine/BiRefNet/models/modules/utils.py:15
↓ 63 callersMethodcuda
(self)
engine/SegmentAPI/base.py:20
↓ 61 callersMethodload
(self, *args, **kwargs)
engine/pose_estimation/dinov2/dinov2/fsdp/__init__.py:114
↓ 59 callersFunctionconvert_db_to_output
(db, batch_size=2, keys=None, is_3d=False)
engine/pose_estimation/third-party/ViTPose/tests/utils/data_utils.py:5
↓ 58 callersMethodresize
(self, img)
tools/metrics/compute_metric_ssim_lipis.py:109
↓ 58 callersMethodto
Change the device of the data inplace. Example: >>> self = SamplingResult.random() >>> print(f'self = {self.to(None)}
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:58
↓ 55 callersMethodnorm
(self)
engine/pose_estimation/dinov2/dinov2/hub/depth/decode_heads.py:453
↓ 55 callersMethodupdate
Update a new data sample.
engine/pose_estimation/third-party/ViTPose/mmpose/utils/timer.py:21
↓ 54 callersFunctionbuild_loss
Build loss.
engine/pose_estimation/third-party/ViTPose/mmpose/models/builder.py:32
↓ 51 callersMethodclone
(self)
LHM/models/rendering/gs_renderer.py:338
↓ 47 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to Non
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/vit.py:286
↓ 43 callersMethodinfo
(self, txt)
engine/BiRefNet/utils.py:56
↓ 40 callersMethodtrain
Convert the model into training mode.
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/vit.py:338
↓ 39 callersMethodsave
Dump model and checkpointables to a file. Args: name (str): name of the file. kwargs (dict): extra arbitrary
engine/pose_estimation/dinov2/dinov2/fsdp/__init__.py:86
↓ 38 callersMethodwrite
write the mesh to a path. Args: path (str): path to write, supports ply, obj and glb.
LHM/models/rendering/mesh_utils.py:706
↓ 31 callersMethodread
(self)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:329
↓ 30 callersFunctioninit_pose_model
Initialize a pose model from config file. Args: config (str or :obj:`mmcv.Config`): Config file path or the config object.
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference.py:22
↓ 29 callersMethod_xywh2cs
This encodes bbox(x,y,w,h) into (center, scale) Args: x, y, w, h (float): left, top, width and height padding (float)
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/base/kpt_2d_sview_rgb_vid_top_down_dataset.py:125
↓ 29 callersMethoddevice
(self)
LHM/runners/infer/base_inferrer.py:46
↓ 28 callersMethodvalues
(self)
LHM/models/rendering/smplx/smplx/utils.py:48
↓ 23 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hrnet.py:385
↓ 23 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hrnet.py:390
↓ 23 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
LHM/models/rendering/smplx/smplx/utils.py:99
↓ 21 callersMethodnorm
(self)
LHM/models/encoders/dinov2/hub/depth/decode_heads.py:453
↓ 21 callersMethodupdate
(self, **kwargs)
engine/pose_estimation/dinov2/dinov2/logging/helpers.py:26
↓ 20 callersMethodbackward
(ctx, g)
LHM/models/rendering/utils/utils.py:106
↓ 20 callersMethodcheckpoint
(self)
engine/pose_estimation/dinov2/dinov2/run/eval/knn.py:28
↓ 19 callersMethoddecode
Decode hand keypoint, relative root depth and hand type. Args: img_metas (list(dict)): Information about data augmentation
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/interhand_3d_head.py:443
↓ 19 callersMethodget_image
Get the frame image. Returns: array: The frame image.
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/message.py:103
↓ 19 callersFunctioninference_top_down_pose_model
Inference a single image with a list of person bounding boxes. Note: - num_people: P - num_keypoints: K - bbox height: H
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference.py:423
↓ 18 callersMethod_sort_and_unique_bboxes
sort kpts and remove the repeated ones.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/animal/animal_pose_dataset.py:350
↓ 18 callersMethod_write_keypoint_results
Write results into a json file.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/mesh/mesh_h36m_dataset.py:55
↓ 18 callersFunctionbuild_dataset
Build a dataset from config dict. Args: cfg (dict): Config dict. It should at least contain the key "type". default_args (dict, o
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/builder.py:62
↓ 18 callersMethodexport
(self, *, overwrite: bool)
engine/pose_estimation/dinov2/dinov2/distributed/__init__.py:219
↓ 18 callersFunctionget_expansion
Get the expansion of a residual block. The block expansion will be obtained by the following order: 1. If ``expansion`` is given, just retur
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:275
↓ 17 callersMethodcriterion
Criterion of wingloss. Note: - batch_size: N - num_keypoints: K - dimension of keypoints: D (D=2 or D=3)
engine/pose_estimation/third-party/ViTPose/mmpose/models/losses/regression_loss.py:79
↓ 16 callersFunction_demo_inputs
Create a superset of inputs needed to run backbone. Args: input_shape (tuple): input batch dimensions. Default: (1, 3, 64, 64
engine/pose_estimation/third-party/ViTPose/tests/test_models/test_top_down_head.py:507
↓ 16 callersMethodinference_model
Inference function. Returns: output (list[np.ndarray]): list of output hand keypoint heatmaps, relative root depth an
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/interhand_3d_head.py:394
↓ 16 callersMethodrandom
Args: rng (None | int | numpy.random.RandomState): seed or state. kwargs (keyword arguments): - num_p
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:95
↓ 16 callersMethodstep
(self, pred: np.ndarray, gt: np.ndarray)
engine/BiRefNet/evaluation/metrics.py:151
↓ 15 callersFunctionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
LHM/utils/rot6d.py:238
↓ 15 callersMethodrun
Method representing the Node's activity. This method override the standard run() method of Thread. Users should not override this met
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/nodes/node.py:300
↓ 15 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
LHM/models/rendering/smplx_gsavatar/utils.py:101
↓ 14 callersMethodget_loss
Calculate bottom-up masked mse loss. Note: - batch_size: N - num_channels: C - heatmaps height: H
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/deconv_head.py:248
↓ 13 callersFunctionget_track_id
Get track id for each person instance on the current frame. Args: results (list[dict]): The bbox & pose results of the current frame
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference_tracking.py:167
↓ 13 callersFunctionvis_pose_result
Visualize the detection results on the image. Args: model (nn.Module): The loaded detector. img (str | np.ndarray): Image filenam
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference.py:674
↓ 12 callersMethodget_pose_results
Get pose estimation results of the message. Args: tag (str, optional): If given, only the results with the tag wi
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/message.py:161
↓ 12 callersMethodinit_weights
(self, pretrained=None)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/shufflenet_v1.py:260
↓ 12 callersMethodinit_weights
(self, pretrained=None)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/mobilenet_v2.py:232
↓ 12 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:638
↓ 12 callersFunctionload
(module, prefix='')
engine/pose_estimation/third-party/ViTPose/mmcv_custom/checkpoint.py:73
↓ 12 callersFunctionload_checkpoint
Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/utils/utils.py:7
↓ 11 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_laye
engine/BiRefNet/models/backbones/pvt_v2.py:14
↓ 11 callersFunction_make_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
LHM/models/encoders/dinov2/hub/backbones.py:18
↓ 11 callersFunction_make_dinov2_model
( *, arch_name: str = "vit_large", img_size: int = 518, patch_size: int = 14, init_values:
engine/pose_estimation/dinov2/dinov2/hub/backbones.py:18
↓ 11 callersMethod_report_metric
Keypoint evaluation. Report mean per joint position error (MPJPE) and mean per joint position error after rigid alignment (MPJPE-PA)
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/mesh/mesh_h36m_dataset.py:61
↓ 11 callersFunctioneval
(input_folder, target_folder, front_view_idx, device="cuda")
tools/metrics/compute_facesimilarity.py:95
↓ 11 callersMethodinit_weights
Initialize model weights.
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/ae_higher_resolution_head.py:240
↓ 11 callersMethodto
(self, device)
engine/pose_estimation/pose_estimator.py:57
↓ 11 callersMethodtrain
(self, mode=True)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/shufflenet_v1.py:323
↓ 11 callersMethodtrain
(self, mode=True)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/mobilenet_v2.py:269
↓ 10 callersFunction_demo_inputs
Create a superset of inputs needed to run backbone. Args: input_shape (tuple): input batch dimensions. Default: (1, 3, 64, 64
engine/pose_estimation/third-party/ViTPose/tests/test_models/test_bottom_up_head.py:472
↓ 10 callersFunctionbuild_posenet
Build posenet.
engine/pose_estimation/third-party/ViTPose/mmpose/models/builder.py:37
↓ 10 callersMethodconvert
Multiple with alpha and add beta with clip.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/pipelines/shared_transform.py:334
↓ 10 callersMethodevaluate
Evaluate coco keypoint results. The pose prediction results will be saved in ``${res_folder}/result_keypoints.json``. Note:
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/bottom_up/bottom_up_coco.py:143
↓ 10 callersMethodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
engine/pose_estimation/third-party/ViTPose/mmpose/models/detectors/top_down.py:91
↓ 10 callersMethodforward
Calls either forward_train or forward_test depending on whether return_loss=True. Note this setting will change the expected inputs. W
engine/pose_estimation/third-party/ViTPose/mmpose/models/detectors/posewarper.py:62
↓ 10 callersMethodinit_weights
(self, pretrained=None)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/shufflenet_v2.py:262
↓ 10 callersFunctionkeypoint_pck_accuracy
Calculate the pose accuracy of PCK for each individual keypoint and the averaged accuracy across all keypoints for coordinates. Note:
engine/pose_estimation/third-party/ViTPose/mmpose/core/evaluation/top_down_eval.py:179
↓ 10 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB The tenso
LHM/models/rendering/smplx/smplx/lbs.py:152
↓ 10 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
LHM/models/rendering/smplx_gsavatar/lbs.py:152
↓ 10 callersFunctionload_image_from_disk_or_url
Load an image file, from disk or url. Args: filename (str): file name on the disk or url link. readFlag (int): readFlag for imdec
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:45
↓ 10 callersFunctionresize_image_keepaspect_np
similar to ImageOps.contain(img_pil, (img_size, img_size)) # keep the same aspect ratio
LHM/runners/infer/utils.py:101
↓ 10 callersMethodupdate
(self, *args, **kwargs)
LHM/outputs/base.py:79
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the file where the mode
LHM/models/rendering/smplx/smplx/body_models.py:68
↓ 9 callersMethod__init__
SMPL model constructor Parameters ---------- model_path: str The path to the folder or to the fi
LHM/models/rendering/smplx_gsavatar/body_models.py:49
↓ 9 callersFunction_demo_mm_inputs
Create a superset of inputs needed to run test or train batches. Args: input_shape (tuple): input batch dimensions nu
engine/pose_estimation/third-party/ViTPose/tests/test_models/test_top_down_forward.py:465
↓ 9 callersFunctioncast_tensor_type
Recursively convert Tensor in inputs from src_type to dst_type. Args: inputs: Inputs that to be casted. src_type (torch.dtype): S
engine/pose_estimation/third-party/ViTPose/mmpose/core/fp16/utils.py:8
↓ 9 callersMethodevaluate
Evaluate PCKh for MPII dataset. Adapted from https://github.com/leoxiaobin/deep-high-resolution-net.pytorch Copyright (c) Microsoft, u
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/top_down/topdown_mpii_dataset.py:142
↓ 9 callersMethodforward
(self, x)
engine/pose_estimation/third-party/ViTPose/tests/test_onnx.py:17
↓ 9 callersFunctionget_root_logger
Use `get_logger` method in mmcv to get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHan
engine/pose_estimation/third-party/ViTPose/mmpose/utils/logger.py:7
↓ 9 callersMethodinit_weights
(self, pretrained=None)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/mobilenet_v3.py:144
↓ 9 callersMethodinit_weights
(self, pretrained=None)
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/vgg.py:148
↓ 9 callersFunctionnms
Greedily select boxes with high confidence and overlap <= thr. Args: dets: [[x1, y1, x2, y2, score]]. thr: Retain overlap < thr.
engine/pose_estimation/third-party/ViTPose/mmpose/core/post_processing/nms.py:9
next →1–100 of 3,669, ranked by callers