Functions — github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch

CodeHub MCPcopy Create free account

hub / github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch / functions

Functions72 in github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch

⨍ Functions 72 ◇ Types & classes 23

↓ 23 callersFunctionconv

(in_channels, out_channels, kernel_size=3, padding=1, bn=True, dilation=1, stride=1, relu=True, bias=True)

modules/conv.py:4

↓ 11 callersFunctionconv_dw

(in_channels, out_channels, kernel_size=3, padding=1, stride=1, dilation=1)

modules/conv.py:13

↓ 7 callersMethod__init__

(self, in_channels, out_channels)

models/with_mobilenet.py:8

↓ 3 callersFunctionconv_dw_no_bn

(in_channels, out_channels, kernel_size=3, padding=1, stride=1, dilation=1)

modules/conv.py:25

↓ 3 callersFunctionload_state

(net, checkpoint)

modules/load_state.py:4

↓ 2 callersFunctionget_alpha

(rate=30, cutoff=1)

modules/one_euro_filter.py:4

↓ 2 callersMethodplot

(self, img, vertices, edges)

modules/draw.py:36

↓ 2 callersFunctionresizeFeatureMaps

pose_extractor/src/extract_poses.cpp:7

↓ 2 callersFunctionwrap_feature_maps

pose_extractor/wrapper.cpp:15

↓ 1 callersMethod_draw_axes

modules/draw.py:44

↓ 1 callersMethod_get_rotation

(self, theta, phi)

modules/draw.py:58

↓ 1 callersMethod_normalize

(img, img_mean, img_scale)

modules/inference_engine_pytorch.py:43

↓ 1 callersMethod_plot_edges

(self, img, vertices, edges, R)

modules/draw.py:51

↓ 1 callersFunctionconvert_to_onnx

(net, output_name)

scripts/convert_to_onnx.py:9

↓ 1 callersFunctionconvert_to_trt

(net, output_name, height, width)

scripts/convert_to_trt.py:10

↓ 1 callersFunctiondraw_poses

(img, poses_2d)

modules/draw.py:91

↓ 1 callersFunctionextractPoses

pose_extractor/src/extract_poses.cpp:34

↓ 1 callersFunctionextract_keypoints

(heatmap, all_keypoints, total_keypoint_num)

modules/legacy_pose_extractor.py:18

↓ 1 callersFunctionextract_poses

(heatmaps, pafs, upsample_ratio)

modules/legacy_pose_extractor.py:202

↓ 1 callersMethodfilter

(self, translation)

modules/pose.py:43

↓ 1 callersFunctionfindPeaks

pose_extractor/src/peak.cpp:29

↓ 1 callersFunctionget_root_relative_poses

(inference_results)

modules/parse_poses.py:23

↓ 1 callersFunctionget_similarity

(a, b, threshold=0.5)

modules/pose.py:50

↓ 1 callersFunctiongroupPeaksToPoses

pose_extractor/src/peak.cpp:108

↓ 1 callersFunctiongroup_keypoints

(all_keypoints_by_type, pafs, pose_entry_size=20, min_paf_score=0.05)

modules/legacy_pose_extractor.py:53

↓ 1 callersMethodinfer

modules/inference_engine_pytorch.py:33

↓ 1 callersFunctionlinspace2d

(start, stop, n=10)

modules/legacy_pose_extractor.py:13

↓ 1 callersFunctionparse_poses

(inference_results, input_scale, stride, fx, is_video=False)

modules/parse_poses.py:80

↓ 1 callersFunctionpropagate_ids

Propagate poses ids from previous frame results. Id is propagated, if there are at least `threshold` similar keypoints between pose from previous

modules/pose.py:62

↓ 1 callersFunctionrotate_poses

(poses_3d, R, t)

↓ 1 callersMethodupdate_id

(self, id=None)

modules/pose.py:37

MethodFindPeaksBody

pose_extractor/src/extract_poses.cpp:16

MethodHumanPose

pose_extractor/src/human_pose.cpp:10

MethodHumanPoseByPeaksIndices

pose_extractor/src/peak.cpp:17

pose_extractor/src/peak.cpp:12

FunctionPyInit_pose_extractor

pose_extractor/wrapper.cpp:81

MethodTwoJointsConnection

pose_extractor/src/peak.cpp:22

(self, x, alpha=0.5)

modules/one_euro_filter.py:14

modules/one_euro_filter.py:34

(self, name, cmake_lists_dir=PACKAGE_NAME, **kwargs)

(self, checkpoint_path, device, img_mean=np.array([128, 128, 128], dtype=np.float32),

modules/inference_engine_pytorch.py:6

(self, canvas_size, origin=(0.5, 0.5), scale=1)

modules/draw.py:18

(self, keypoints, confidence)

modules/pose.py:20

(self, file_names)

modules/input_reader.py:5

(self, file_name)

modules/input_reader.py:24

modules/one_euro_filter.py:11

(self, freq=15, mincutoff=1, beta=1, dcutoff=1)

modules/one_euro_filter.py:24

(self, net_model_xml_path, device)

modules/inference_engine_openvino.py:8

(self, num_channels, num_heatmaps, num_pafs)

models/with_mobilenet.py:25

(self, in_channels, out_channels)

models/with_mobilenet.py:49

(self, in_channels, out_channels, num_heatmaps, num_pafs)

models/with_mobilenet.py:64

(self, in_channels, mid_channels, out_channels)

models/with_mobilenet.py:90

(self, in_channels, out_channels, ratio, should_align=False)

models/with_mobilenet.py:108

(self, in_channels, num_2d_heatmaps, ratio=2, out_channels=57)

models/with_mobilenet.py:127

(self, num_refinement_stages=1, num_channels=128, num_heatmaps=19, num_pafs=38, is_convertibl

models/with_mobilenet.py:146

modules/input_reader.py:9

modules/input_reader.py:31

modules/input_reader.py:13

modules/input_reader.py:39

Methodbuild_extensions

Functionextract_poses

pose_extractor/wrapper.cpp:30

models/with_mobilenet.py:18

models/with_mobilenet.py:41

models/with_mobilenet.py:57

models/with_mobilenet.py:82

models/with_mobilenet.py:101

models/with_mobilenet.py:119

(self, x, feature_maps_2d)

models/with_mobilenet.py:139

models/with_mobilenet.py:179

modules/inference_engine_openvino.py:23

Methodmouse_callback

(event, x, y, flags, params)

modules/draw.py:67

Methodoperator()

pose_extractor/src/extract_poses.cpp:22