plot_cpu.py

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.ticker as ticker

# Data organization
configs = [
    "Ti, Pyr.", "S, Pyr.", "M, Pyr.", "B, Pyr.",
    "Ti, Iso.", "S, Iso.", "B, Iso."
]
N_values = [256, 512, 1024, 2048, 4096]

data = np.array([
    [0.950849, 0.987486, 0.989155, 0.990131, 0.997716],    # Ti, Pyr.
    [0.941655, 0.970678, 0.986227, 0.983855, 0.99496],     # S, Pyr.
    [0.90779,  0.964782, 0.976409, 0.981987, 0.993324],    # M, Pyr.
    [0.899912, 0.94694,  0.95224,  0.966231, 0.992633],    # B, Pyr.
    [0.922873, 0.956576, 0.974857, 0.972628, 0.990712],    # Ti, Iso.
    [0.822407, 0.841416, 0.942377, 0.962056, 0.984504],    # S, Iso.
    [0.883495, 0.886366, 0.891059, 0.92211,  0.947521]     # B, Iso.
])

# Plot configuration
plt.figure(figsize=(14, 8))
ax = plt.gca()

# Style definitions
colors = {'Ti': '#E41A1C', 'S': '#4DAF4A', 'M': '#377EB8', 'B': '#984EA3'}
markers = {'Pyr.': 'o', 'Iso.': 's'}
linestyles = {'Pyr.': '-', 'Iso.': '--'}

# Plot each configuration
for i, config in enumerate(configs):
    size, pyramid_type = map(str.strip, config.split(','))
    y = data[i]
    mask = ~np.isnan(y)
    x = np.array(N_values)[mask]
    y = y[mask]
    ax.plot(
        x, y,
        color=colors[size],
        marker=markers[pyramid_type],
        linestyle=linestyles[pyramid_type],
        linewidth=2.5,
        markersize=10,
        label=config
    )

# Axis formatting
ax.set_xscale('log', base=2)
ax.set_xticks(N_values)
ax.get_xaxis().set_major_formatter(ticker.ScalarFormatter())
ax.tick_params(axis='both', which='major', labelsize=17)
ax.grid(True, linestyle='--', alpha=0.7)

# Labels and title
# Labels and title
ax.set_title("Graph Construction Runtime Proportion for ViG on CPU", fontsize=25, pad=15)
ax.set_xlabel("Image Resolution (Height=Width)", fontsize=20, labelpad=10)
ax.set_ylabel("Runtime Proportion", fontsize=20, labelpad=10)


# Legend
handles, labels = ax.get_legend_handles_labels()
ax.legend(
    handles, labels,
    fontsize=15,
    title="Variants",
    title_fontsize=17,
    loc='upper left',
    bbox_to_anchor=(1, 1)
)

plt.tight_layout(rect=[0, 0, 0.85, 1])  # Make space for legend
plt.savefig('cpu_vig_profile.pdf', dpi=300, bbox_inches='tight')
plt.show()