multimodal/plot_log.py

import matplotlib.pyplot as plt

visual = []
previsual = []
with open("slurm-878067.out") as f:
    lines = f.readlines()
    for line in lines:
        if "previsual_iou" in line:
            line = line.strip().split("//")
            pre_iou = float(line[-6].strip().split(" ")[-1])
            iou = float(line[-5].strip().split(" ")[-1])
            visual.append(iou)
            previsual.append(pre_iou)

plt.plot(visual, label="visual")
plt.plot(previsual, label="previsual")
plt.legend()
plt.ylim(0, 4)
plt.savefig("save.png")