Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JubSteven/POEM-v2
/ types & classes
Types & classes
149 in github.com/JubSteven/POEM-v2
⨍
Functions
1,165
◇
Types & classes
149
↓ 44 callers
Class
CN
lib/utils/config.py:8
↓ 21 callers
Class
ConvBlock
lib/models/bricks/conv.py:4
↓ 16 callers
Class
MeanEPE
lib/metrics/mean_epe.py:11
↓ 13 callers
Class
ConvBlock
lib/external/cmr/net.py:92
↓ 12 callers
Class
Compose
lib/utils/transform.py:345
↓ 12 callers
Class
Registry
A registry to map strings to classes. Registered object could be built from registry. Example: >>> MODELS = Registry('models')
lib/utils/builder.py:50
↓ 7 callers
Class
AverageMeter
Computes and stores the average and current value
lib/metrics/basic_metric.py:33
↓ 6 callers
Class
LossMetric
lib/metrics/basic_metric.py:63
↓ 5 callers
Class
HO3DOfficialTestEvalUtil
Util class for evaluation networks.
lib/datasets/ho3d_official_test.py:319
↓ 5 callers
Class
PAEval
lib/metrics/pa_eval.py:15
↓ 5 callers
Class
ResNet
lib/models/backbones/resnet.py:147
↓ 5 callers
Class
SpiralConv
lib/external/cmr/net.py:22
↓ 4 callers
Class
DDPSummaryWriter
lib/utils/summary_writer.py:6
↓ 4 callers
Class
Recorder
lib/utils/recorder.py:26
↓ 3 callers
Class
DexYCB
lib/datasets/dexycb.py:29
↓ 3 callers
Class
DexYCBMultiView
lib/datasets/dexycb.py:255
↓ 3 callers
Class
InterHand
lib/datasets/interhand.py:27
↓ 3 callers
Class
ptTransformerBlock
lib/models/bricks/point_transformers.py:47
↓ 2 callers
Class
AUCCallback
lib/utils/testing.py:32
↓ 2 callers
Class
Arctic
lib/datasets/arctic.py:31
↓ 2 callers
Class
CMRDataConverter
lib/external/cmr/data_adaptor.py:63
↓ 2 callers
Class
DrawingHandCallback
lib/utils/testing.py:303
↓ 2 callers
Class
HO3D
lib/datasets/ho3d.py:30
↓ 2 callers
Class
HO3DOfficialTestMultiView
lib/datasets/ho3d_official_test.py:60
↓ 2 callers
Class
IdleCallback
lib/utils/testing.py:17
↓ 2 callers
Class
ParallelDeblock
ISM in the paper. Note that "indices[:, :indices.size(1)//3]" is approximate and not-strict k-disk. In this way, a from-small-to-large recept
lib/external/cmr/net.py:58
↓ 2 callers
Class
SinePositionalEncoding3D
Position encoding with sine and cosine functions. See `End-to-End Object Detection with Transformers <https://arxiv.org/pdf/2005.12872>`_ for
lib/models/layers/petr_transformer.py:400
↓ 2 callers
Class
_Sequential
A wrapper to allow nn.Sequential to accept multiple inputs
lib/models/layers/ptEmb_transformer.py:110
↓ 2 callers
Class
ptTransformerBlock_CrossAttn
lib/models/bricks/point_transformers.py:99
↓ 1 callers
Class
ArcticMultiView
lib/datasets/arctic.py:194
↓ 1 callers
Class
BertLayerNorm
lib/models/bricks/metro_transformer.py:9
↓ 1 callers
Class
BertLayerNorm
lib/external/metro/base_model.py:20
↓ 1 callers
Class
EncodeMesh
lib/external/cmr/model.py:48
↓ 1 callers
Class
EncodeUV
lib/external/cmr/model.py:25
↓ 1 callers
Class
ExpLogger
lib/utils/logger.py:57
↓ 1 callers
Class
FFMPEGFrameLoader
video_tool/ffmpeg_util.py:32
↓ 1 callers
Class
FileFormatter
lib/utils/logger.py:46
↓ 1 callers
Class
HO3DMultiView
lib/datasets/ho3d.py:500
↓ 1 callers
Class
HO3DOfficialEvalCallback
lib/utils/testing.py:102
↓ 1 callers
Class
HO3DOfficialTest
lib/datasets/ho3d_official_test.py:30
↓ 1 callers
Class
HighResolutionModule
lib/models/backbones/hrnet.py:108
↓ 1 callers
Class
HighResolutionModule
lib/external/metro/hrnet/hrnet.py:101
↓ 1 callers
Class
HighResolutionNet
lib/external/metro/hrnet/hrnet.py:233
↓ 1 callers
Class
IntegralDeconvHead
lib/models/integal_pose.py:255
↓ 1 callers
Class
InterHandMultiView
lib/datasets/interhand.py:213
↓ 1 callers
Class
Joint3DPCK
lib/metrics/pck.py:150
↓ 1 callers
Class
JointsMSELoss
lib/models/darkpose.py:48
↓ 1 callers
Class
MANO
lib/models/layers/mano_wrapper.py:17
↓ 1 callers
Class
METRO_Encoder
Modified the original METRO encoder to provide an interface for key value and query to be different.
lib/models/bricks/metro_transformer.py:26
↓ 1 callers
Class
METRO_Encoder
lib/external/metro/base_model.py:37
↓ 1 callers
Class
MeshSampler
Mesh object that is used for handling certain graph operations.
lib/external/metro/base_model.py:346
↓ 1 callers
Class
MetroTR
lib/models/layers/ptEmb_transformer.py:124
↓ 1 callers
Class
MixDataset
lib/datasets/mix_dataset.py:12
↓ 1 callers
Class
MixWebDataset
lib/datasets/mix_dataset.py:79
↓ 1 callers
Class
MultiviewSilhouetteLoss
lib/fit/silhouette_loss.py:9
↓ 1 callers
Class
MultiviewWebDataset
lib/data_wds/multiview_wds.py:28
↓ 1 callers
Class
MvPDecoder
lib/models/layers/mvp_decoder.py:450
↓ 1 callers
Class
MvPDecoderLayer
lib/models/layers/mvp_decoder.py:243
↓ 1 callers
Class
OakInk
lib/datasets/oakink.py:25
↓ 1 callers
Class
OakInk2_Dev
lib/datasets/oakink2_dev.py:28
↓ 1 callers
Class
OakInkMultiView
lib/datasets/oakink.py:386
↓ 1 callers
Class
OpenDRRenderer
lib/viztools/opendr_renderer.py:11
↓ 1 callers
Class
ProjAttn
lib/models/layers/mvp_decoder.py:97
↓ 1 callers
Class
PtEmbedTRv2
lib/models/layers/ptEmb_transformer.py:23
↓ 1 callers
Class
Pytorch3dRenderer
lib/fit/pytorch3d_renderer.py:17
↓ 1 callers
Class
RandomOcclusion
Add random occlusion based on occlusion probability. Args: occlusion_prob (float): probability of the image having occlusion. Def
lib/utils/transform.py:21
↓ 1 callers
Class
SelfAttention
lib/external/cmr/net.py:166
↓ 1 callers
Class
SteamFormatter
lib/utils/logger.py:25
↓ 1 callers
Class
Vert3DPCK
lib/metrics/pck.py:176
↓ 1 callers
Class
point_METRO_layer
lib/models/bricks/pt_metro_transformer.py:43
↓ 1 callers
Class
pointer_layer
lib/models/bricks/pt_metro_transformer.py:13
Class
BasePointEmbedHead
lib/models/heads/ptEmb_head.py:55
Class
BaseTransformerLayer
Base `TransformerLayer` for vision transformer. It can be built from `mmcv.ConfigDict` and support more flexible customization, for example,
lib/models/bricks/transformer.py:239
Class
BasicBlock
lib/models/backbones/hrnet.py:38
Class
BasicBlock
lib/models/backbones/resnet.py:77
Class
BasicBlock
lib/external/metro/hrnet/hrnet.py:31
Class
Bottleneck
lib/models/backbones/hrnet.py:70
Class
Bottleneck
lib/models/backbones/resnet.py:109
Class
Bottleneck
lib/external/metro/hrnet/hrnet.py:63
Class
BottleneckX2
lib/models/backbones/hourglass.py:6
Class
CMR_G
Implementation of CMR_SG. :param spiral_indices: pre-defined spiral sample :param up_transform: pre-defined upsample matrix :param re
lib/external/cmr/model.py:77
Class
CONST
lib/utils/misc.py:66
Class
ColorMode
lib/viztools/utils.py:15
Class
ConvTBlock
lib/external/cmr/net.py:125
Class
DarkPose_ResNet
lib/models/darkpose.py:70
Class
DeformFunction
lib/models/layers/mvp_decoder.py:66
Class
DexYCBMultiView_CMR
lib/external/cmr/data_adaptor.py:156
Class
DexYCBMultiView_Video
lib/datasets/dexycb.py:521
Class
DexYCB_CMR
lib/external/cmr/data_adaptor.py:141
Class
FFN
Implements feed-forward networks (FFNs) with identity connection. Args: embed_dims (int): The feature dimension. Same as `Mul
lib/models/bricks/transformer.py:175
Class
Formatter
Logging Formatter to add colors and count warning / errors
lib/utils/logger.py:11
Class
FreiHAND
Mano version and FreiHAND v2 The following paragraph is only relevant if you downloaded the dataset before 22. reference: https://github.com/l
lib/datasets/freihand.py:144
Class
FreiHAND_v2_Extra
lib/datasets/freihand.py:437
Class
FrozenBatchNorm2d
https://github.com/facebookresearch/detectron2/blob/dfe8d368c8b7cc2be42c5c3faf9bdcc3c08257b1/detectron2/layers/batch_norm.py It contains non-
lib/models/backbones/resnet.py:31
Class
HDataset
lib/datasets/hdata.py:31
Class
HO3DV3
lib/datasets/ho3d.py:462
Class
HRNet
lib/models/backbones/hrnet.py:444
Class
HRNetw60
lib/models/backbones/hrnet.py:457
Class
HandLoss
lib/fit/hand_loss.py:10
Class
HighResolutionNet
lib/models/backbones/hrnet.py:240
next →
1–100 of 149, ranked by callers