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

一个奇怪的BUG

UserWarning: FigureCanvasAgg is non-interactive, and thus cannot be shown_第1张图片

这个代码会报下面的错:

在这里插入图片描述

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

UserWarning: FigureCanvasAgg is non-interactive, and thus cannot be shown_第2张图片

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

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()

你可能感兴趣的:(python)