Function plot_graph

pytracking/utils/plotting.py:32–47

Plot graph. Data is a 1D tensor. args: fig_num: Figure number. title: Title of figure.

(a: torch.Tensor, fig_num = None, title = None)

def plot_graph(a: torch.Tensor, fig_num = None, title = None):
    """Plot graph. Data is a 1D tensor.
    args:
        fig_num: Figure number.
        title: Title of figure.
    """
    a_np = a.squeeze().cpu().clone().detach().numpy()
    if a_np.ndim > 1:
        raise ValueError
    fig = plt.figure(fig_num)
    # plt.tight_layout()
    plt.cla()
    plt.plot(a_np)
    if title is not None:
        plt.title(title)
    draw_figure(fig)