wandb.log matplotlib pcolormesh

Question

我正在尝试生成一个 pcolormesh 对象。然后使用 wandb.log 来记录它，但我看到 wandb 默认情况下会先将此图转换为plotly。这就是为什么我最终收到此警告并且我的情节没有显示任何内容：

Dang! That path collection is out of this world. I totally don't know what to do with it yet! Plotly can only import path collections linked to 'data' coordinates

这是我在代码中所做的事情：

    fig = plt.figure()
    plt.pcolormesh(out_abs.cpu())
    plt.ylabel('Frequency (Hz)')
    plt.xlabel('Time')

    return fig

看来解决方案是要么从一开始就用plotly制作这个无花果，要么将其转换为plotly。

谢谢！

Answer 1

我现在的一个解决方法是通过

wandb.Image(fig)

将 matplotlib 图转换为 wandb 中的图像对象