Function plot_points

evaluation/evaluate.py:66–109

(points, image, visible=None, correct=None)

def plot_points(points, image, visible=None, correct=None):
    colors = get_n_colors(len(points), pastel_factor=0.2)
    if correct is None:
        correct = [1] * len(points)
    for i, (coord, color, visible_, correct_) in enumerate(
        zip(points, colors, visible, correct)
    ):
        if visible_ == 1:
            color = [255 * c for c in color]
            if correct_:
                image = cv2.circle(
                    np.ascontiguousarray(image),
                    tuple(coord.astype("int32")),
                    4,
                    color,
                    2,
                )
            else:
                # plot x
                image = cv2.line(
                    np.ascontiguousarray(image),
                    tuple(coord.astype("int32") - 4),
                    tuple(coord.astype("int32") + 4),
                    color,
                    2,
                )
                image = cv2.line(
                    np.ascontiguousarray(image),
                    tuple(coord.astype("int32") + np.array([-4, 4])),
                    tuple(coord.astype("int32") - np.array([-4, 4])),
                    color,
                    2,
                )
            # plot index next to point
            image = cv2.putText(
                np.ascontiguousarray(image),
                str(i),
                tuple(coord.astype("int32") + np.array([4, 4])),
                cv2.FONT_HERSHEY_SIMPLEX,
                0.5,
                color,
                1,
            )
    return image


def visualize_vertices(verts, img, cmap="hot"):