garmentiq.landmark.detection

 1# garmentiq/landmark/detection/__init__.py
 2"""Landmark detection model and its pre- and post-processing.
 3
 4Holds the HRNet-based pose model used to predict garment landmarks, its loader, and
 5the coordinate transforms that map between the original image and the model's input
 6resolution, including affine cropping and heatmap decoding.
 7"""
 8from .load_model import load_model
 9from .model_definition import PoseHighResolutionNet
10from .utils import (
11    get_max_preds,
12    get_final_preds,
13    flip_back,
14    fliplr_joints,
15    transform_preds,
16    get_affine_transform,
17    affine_transform,
18    get_3rd_point,
19    get_dir,
20    crop,
21    input_image_transform,
22)