MCPcopy Create free account

hub / github.com/NationalGAILab/HoT / functions

Functions327 in github.com/NationalGAILab/HoT

↓ 13 callersMethodupdate
(self, val, n=1)
common/utils.py:112
↓ 6 callersMethod__init__
(self, cfgfile)
demo/lib/yolov3/darknet.py:267
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mhformer/module/trans_hypothesis_cluster.py:63
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mhformer/module/trans_hypothesis.py:9
↓ 4 callersFunctionLoad_model
(args, model)
common/utils.py:191
↓ 4 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/motionbert/hot_motionbert.py:63
↓ 4 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mixste/hot_mixste.py:62
↓ 4 callersFunctiondefine_actions
( action )
common/utils.py:176
↓ 4 callersFunctiondefine_error_list
(actions)
common/utils.py:164
↓ 4 callersFunctionnormalize_screen_coordinates
(X, w, h)
common/camera.py:6
↓ 4 callersFunctionparse_args
()
common/arguments.py:10
↓ 4 callersFunctionprint_error
(data_type, action_error_sum, is_train)
common/utils.py:119
↓ 4 callersFunctionsave_model
(args, epoch, mpjpe, model, model_name)
common/utils.py:151
↓ 4 callersFunctiontest_calculation
(predicted, target, action, error_sum, data_type, subject)
common/utils.py:12
↓ 4 callersMethodupdate
Params: dets - a numpy array of detections in the format [[x1,y1,x2,y2,score],[x1,y1,x2,y2,score],...] Requires: this metho
demo/lib/sort/sort.py:177
↓ 3 callersMethod__init__
(self, args, n_layers=12, dim_in=3, dim_feat=64, dim_rep=512, dim_out=3, mlp_ratio=4, act_layer=nn.GELU, attn_
model/motionagformer/hot_motionagformer.py:261
↓ 3 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mhformer/module/trans.py:9
↓ 3 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/motionbert/motionbert.py:12
↓ 3 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mixste/mixste.py:11
↓ 3 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/mixste/tpc_mixste.py:65
↓ 3 callersMethod__init__
(self, inplanes, planes, stride=1, downsample=None)
demo/lib/hrnet/lib/models/pose_hrnet.py:31
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
demo/lib/hrnet/lib/models/pose_hrnet.py:393
↓ 3 callersMethod_make_transition_layer
( self, num_channels_pre_layer, num_channels_cur_layer)
demo/lib/hrnet/lib/models/pose_hrnet.py:333
↓ 3 callersFunctionbn_init
(bn, scale)
model/motionagformer/modules/ctrgc.py:42
↓ 3 callersMethodcameras
(self)
common/mocap_dataset.py:29
↓ 3 callersFunctionedge2mat
(link, num_node)
model/motionagformer/modules/ctrgc.py:9
↓ 3 callersFunctionget_affine_transform
( center, scale, rot, output_size, shift=np.array([0, 0], dtype=np.float32), inv=0 )
demo/lib/hrnet/lib/utils/transforms.py:58
↓ 3 callersFunctionwrap
(func, *args, unsqueeze=False)
common/camera.py:41
↓ 2 callersMethod__init__
(self, args, n_layers=12, dim_in=3, dim_feat=64, dim_rep=512, dim_out=3, mlp_ratio=4, act_layer=nn.GELU, attn_
model/motionagformer/motionagformer.py:177
↓ 2 callersMethod_change_shared_attn_device
(self, channel_attn, shared_attn)
model/motionagformer/modules/ctr_attention.py:117
↓ 2 callersMethod_compute_metadata
(self)
common/skeleton.py:68
↓ 2 callersFunctionarg_parse
Parse arguements to the detect module
demo/lib/yolov3/human_detector.py:55
↓ 2 callersMethodchange_adj_device_to_cuda
(self, adj)
model/motionagformer/modules/graph.py:89
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
demo/lib/hrnet/lib/models/pose_hrnet.py:22
↓ 2 callersFunctionconv_init
(conv)
model/motionagformer/modules/ctrgc.py:35
↓ 2 callersFunctionconvert_bbox_to_z
Takes a bounding box in the form [x1,y1,x2,y2] and returns z in the form [x,y,s,r] where x,y is the centre of the box and s is the scale/ar
demo/lib/sort/sort.py:33
↓ 2 callersFunctionconvert_x_to_bbox
Takes a bounding box in the centre form [x,y,s,r] and returns it in the form [x1,y1,x2,y2] where x1,y1 is the top left and x2,y2 is the bot
demo/lib/sort/sort.py:48
↓ 2 callersMethodfetch
(self, dataset, subjects, subset=1, parse_3d_poses=True)
common/load_data_hm36.py:120
↓ 2 callersFunctionget_3rd_point
(a, b)
demo/lib/hrnet/lib/utils/transforms.py:99
↓ 2 callersMethodget_batch
(self, seq_i, start_3d, end_3d, flip, reverse)
common/generator.py:112
↓ 2 callersMethodmixer_forward
(self, x)
model/motionagformer/modules/metaformer.py:67
↓ 2 callersFunctionnormalize_digraph
(A)
model/motionagformer/modules/ctrgc.py:16
↓ 2 callersMethodnum_frames
(self)
common/generator.py:89
↓ 2 callersFunctionpredict_transform
(prediction, inp_dim, anchors, num_classes, CUDA = True)
demo/lib/yolov3/util.py:34
↓ 2 callersMethodprepare_data
(self, dataset, folder_list)
common/load_data_hm36.py:61
↓ 2 callersFunctionshowimage
(ax, img)
demo/vis.py:136
↓ 2 callersMethodskeleton
(self)
common/mocap_dataset.py:26
↓ 1 callersFunctionPreProcess
(image, bboxs, cfg, num_pos=2)
demo/lib/hrnet/lib/utils/utilitys.py:139
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, dilation=1)
model/motionagformer/modules/tcn.py:6
↓ 1 callersMethod__init__
(self, in_channels, out_channels, rel_reduction=8)
model/motionagformer/modules/ctrgc.py:48
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
demo/lib/hrnet/lib/models/pose_hrnet.py:119
↓ 1 callersMethod_init_A
()
model/motionagformer/modules/ctrgc.py:112
↓ 1 callersMethod_init_fusion
(self)
model/motionagformer/motionagformer.py:118
↓ 1 callersMethod_init_fusion
(self)
model/motionagformer/hot_motionagformer.py:202
↓ 1 callersMethod_init_gcn
(self)
model/motionagformer/modules/graph.py:46
↓ 1 callersMethod_init_spatial_adj
(self)
model/motionagformer/modules/graph.py:52
↓ 1 callersMethod_init_spatial_shared_attn
(self)
model/motionagformer/modules/ctr_attention.py:38
↓ 1 callersMethod_init_temporal_adj
Connects each joint to itself and the same joint withing next `connection_length` frames.
model/motionagformer/modules/graph.py:62
↓ 1 callersMethod_init_temporal_shared_attn
(self)
model/motionagformer/modules/ctr_attention.py:53
↓ 1 callersMethod_make_branches
(self, num_branches, block, num_blocks, num_channels)
demo/lib/hrnet/lib/models/pose_hrnet.py:177
↓ 1 callersMethod_make_fuse_layers
(self)
demo/lib/hrnet/lib/models/pose_hrnet.py:187
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
demo/lib/hrnet/lib/models/pose_hrnet.py:374
↓ 1 callersMethod_make_one_branch
(self, branch_index, block, num_blocks, num_channels, stride=1)
demo/lib/hrnet/lib/models/pose_hrnet.py:139
↓ 1 callersFunctionaffine_transform
(pt, t)
demo/lib/hrnet/lib/utils/transforms.py:93
↓ 1 callersFunctionassociate_detections_to_trackers
Assigns detections to tracked object (both represented as bounding boxes) Returns 3 lists of matches, unmatched_detections and unmatched_tra
demo/lib/sort/sort.py:125
↓ 1 callersFunctionbbox_iou
Returns the IoU of two bounding boxes
demo/lib/yolov3/bbox.py:51
↓ 1 callersFunctionbox_to_center_scale
convert a box to center,scale information required for pose transformation Parameters ---------- box : (x1, y1, x2, y2) model_image_wi
demo/lib/hrnet/lib/utils/utilitys.py:102
↓ 1 callersFunctioncamera_to_world
(X, R, t)
common/camera.py:16
↓ 1 callersFunctioncluster_dpc_knn
(x, cluster_num, k, token_mask=None)
model/motionagformer/hot_motionagformer.py:28
↓ 1 callersFunctioncluster_dpc_knn
(x, cluster_num, k, token_mask=None)
model/motionbert/hot_motionbert.py:25
↓ 1 callersFunctioncluster_dpc_knn
(x, cluster_num, k, token_mask=None)
model/mixste/hot_mixste.py:23
↓ 1 callersFunctioncluster_dpc_knn_center
(x, cluster_num, k, center, token_mask=None)
model/mhformer/module/trans_hypothesis_cluster.py:21
↓ 1 callersFunctioncluster_dpc_knn_center
(x, cluster_num, k, center, token_mask=None)
model/mixste/tpc_mixste.py:23
↓ 1 callersFunctioncoco_h36m
(keypoints)
demo/lib/preprocess.py:10
↓ 1 callersFunctioncoco_h36m
(keypoints)
demo/lib/hrnet/lib/utils/coco_h36m.py:9
↓ 1 callersFunctioncreate_layers
(dim, n_layers, mlp_ratio=4., act_layer=nn.GELU, attn_drop=0., drop_rate=0., drop_path_rate=0.,
model/motionagformer/motionagformer.py:146
↓ 1 callersFunctioncreate_layers
(dim, n_layers, mlp_ratio=4., act_layer=nn.GELU, attn_drop=0., drop_rate=0., drop_path_rate=0.,
model/motionagformer/hot_motionagformer.py:230
↓ 1 callersFunctioncreate_modules
(blocks)
demo/lib/yolov3/darknet.py:138
↓ 1 callersFunctiondeterministic_random
(min_value, max_value, data)
common/utils.py:170
↓ 1 callersMethodforward
(self, x)
model/motionagformer/modules/tcn.py:19
↓ 1 callersMethodforward_spatial
(self, q, k, v)
model/motionagformer/modules/attention.py:40
↓ 1 callersMethodforward_spatial
(self, q, k, v, xx)
model/motionagformer/modules/ctr_attention.py:97
↓ 1 callersMethodforward_spatial
(self, q, k, v)
model/motionbert/hot_motionbert.py:110
↓ 1 callersMethodforward_spatial
(self, q, k, v)
model/motionbert/motionbert.py:59
↓ 1 callersMethodforward_temporal
(self, q, k, v)
model/motionagformer/modules/attention.py:50
↓ 1 callersMethodforward_temporal
(self, q, k, v, xx)
model/motionagformer/modules/ctr_attention.py:123
↓ 1 callersMethodforward_temporal
(self, q, k, v, seqlen=8)
model/motionbert/hot_motionbert.py:120
↓ 1 callersMethodforward_temporal
(self, q, k, v, seqlen=8)
model/motionbert/motionbert.py:69
↓ 1 callersFunctionget_dir
(src_point, rot_rad)
demo/lib/hrnet/lib/utils/transforms.py:104
↓ 1 callersFunctionget_final_preds
(config, batch_heatmaps, center, scale)
demo/lib/hrnet/lib/utils/inference.py:52
↓ 1 callersFunctionget_max_preds
get predictions from score maps heatmaps: numpy.ndarray([batch_size, num_joints, height, width])
demo/lib/hrnet/lib/utils/inference.py:21
↓ 1 callersMethodget_num_inchannels
(self)
demo/lib/hrnet/lib/models/pose_hrnet.py:244
↓ 1 callersFunctionget_pose2D
(video_path, output_dir)
demo/vis.py:97
↓ 1 callersFunctionget_pose3D
(video_path, output_dir, fix_z)
demo/vis.py:143
↓ 1 callersFunctionget_spatial_graph
(self_link, inward, outward, num_node)
model/motionagformer/modules/ctrgc.py:27
↓ 1 callersMethodget_state
Returns the current bounding box estimate.
demo/lib/sort/sort.py:118
↓ 1 callersFunctionh36m_coco_format
(keypoints, scores)
demo/lib/preprocess.py:40
↓ 1 callersFunctionimg2video
(video_path, output_dir)
demo/vis.py:117
↓ 1 callersFunctionindex_points
(points, idx)
model/motionagformer/hot_motionagformer.py:15
↓ 1 callersFunctionindex_points
(points, idx)
model/mhformer/module/trans_hypothesis_cluster.py:8
next →1–100 of 327, ranked by callers