Spaces:
Running on Zero
Running on Zero
File size: 1,369 Bytes
bff20b3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | # Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
from .gaussian_heatmap import (
generate_gaussian_heatmaps,
generate_udp_gaussian_heatmaps,
generate_unbiased_gaussian_heatmaps,
)
from .instance_property import (
get_diagonal_lengths,
get_instance_bbox,
get_instance_root,
)
from .offset_heatmap import generate_displacement_heatmap, generate_offset_heatmap
from .post_processing import (
batch_heatmap_nms,
gaussian_blur,
gaussian_blur1d,
get_heatmap_maximum,
get_simcc_maximum,
get_simcc_normalized,
)
from .refinement import (
refine_keypoints,
refine_keypoints_dark,
refine_keypoints_dark_udp,
refine_simcc_dark,
)
__all__ = [
"generate_gaussian_heatmaps",
"generate_udp_gaussian_heatmaps",
"generate_unbiased_gaussian_heatmaps",
"gaussian_blur",
"get_heatmap_maximum",
"get_simcc_maximum",
"generate_offset_heatmap",
"batch_heatmap_nms",
"refine_keypoints",
"refine_keypoints_dark",
"refine_keypoints_dark_udp",
"generate_displacement_heatmap",
"refine_simcc_dark",
"gaussian_blur1d",
"get_diagonal_lengths",
"get_instance_root",
"get_instance_bbox",
"get_simcc_normalized",
]
|