pocketpose.models.decoder.posenet
#
Module Contents#
Classes#
- class pocketpose.models.decoder.posenet.PoseNetDecoder(output_stride=32, local_maximum_radius=1, threshold=0.5)#
- decode_multi_pose(heatmaps, offsets, displacement_fwd, displacement_bwd)#
- build_part_with_score_queue(scores)#
- score_is_maximum_in_local_window(keypointId, score, heatmapY, heatmapX, scores)#
- traverse_to_target_keypoint(keypoints, displacements, direction, scores, offsets)#
- get_edge_keypoints(edge_id, direction)#
- estimate_target_keypoint_position(edge_id, source_keypoint_id, target_keypoint_id, keypoints, displacements, scores, offsets)#
- get_displacement(edge_id, keypoint, displacements)#
- get_strided_index_near_point(point, output_stride, displacements)#
- get_instance_score(keypoints)#
- sigmoid(x)#