garmentiq.landmark.utils

 1import numpy as np
 2
 3
 4def find_instruction_landmark_index(instruction_landmarks: dict, predefined: bool):
 5    """
 6    Finds the indices of predefined or non-predefined landmarks from an instruction landmarks dictionary.
 7
 8    Args:
 9        instruction_landmarks (dict): A dictionary of landmark information from an instruction schema.
10        predefined (bool): If True, returns indices of predefined landmarks.
11                           If False, returns indices of non-predefined (derived) landmarks.
12
13    Returns:
14        list: A list of integer indices (0-based) for the requested type of landmarks.
15    """
16    if predefined:
17        return [
18            int(k) - 1
19            for k, v in instruction_landmarks.items()
20            if v.get("predefined") is True
21        ]
22    else:
23        return [
24            int(k) - 1
25            for k, v in instruction_landmarks.items()
26            if v.get("predefined") is False
27        ]
28
29
30def fill_instruction_landmark_coordinate(
31    instruction_landmarks: dict, index: list, fill_in_value: np.array
32):
33    """
34    Fills the 'x' and 'y' coordinates for specified landmarks in the instruction dictionary
35    based on provided coordinate values.
36
37    Args:
38        instruction_landmarks (dict): The dictionary of landmark instructions to be updated.
39        index (list): A list of integer indices (0-based) corresponding to the landmarks
40                      in `instruction_landmarks` that should be updated.
41        fill_in_value (np.array): A NumPy array containing the new coordinates.
42                                  Expected shape: (1, N, 2), where N is the number of landmarks to fill.
43
44    Returns:
45        dict: The updated `instruction_landmarks` dictionary with filled coordinates.
46    """
47    for k in instruction_landmarks:
48        idx = int(k) - 1
49        if idx in index:
50            preds_idx = index.index(idx)
51            instruction_landmarks[k]["x"] = float(fill_in_value[0, preds_idx, 0])
52            instruction_landmarks[k]["y"] = float(fill_in_value[0, preds_idx, 1])
53    return instruction_landmarks