UserWarning: FigureCanvasAgg is non-interactive, and thus cannot be shown

六点零六2023-12-02 0:34

一个奇怪的BUG

这个代码会报下面的错：

但是把模型导入部分注释掉之后就没有这个错误了（第六行）

解决办法：在模型加载后面加入一行代码 matplotlib.use( TkAgg')，这个bug的问题就是模型加载改变了matplotlib使用的终端，但为什么会改变，还是不知道啊

bash 复制代码

import torch
import thop
import torch.nn.utils .prune as prune
import matplotlib
import matplotlib.pyplot as plt
model = torch.load("runs/train/exp6/weights/last.pt")
matplotlib.use( TkAgg')
x = [i for i in range(3)]
y = [i for i in range(3)]
plt.plot(y,x)
plt.show()