记录一下matplotlib画图不显示解决方法。
原代码
import matplotlib.pyplot as plt
plt.switch_backend('agg')
img = Image.open('./fold1/data_train/1.png')
img_rotate = flip_and_rotate(img, flip = False,rotate = True)#调用自己编写函数进行图片旋转
img_flip = flip_and_rotate(img, flip = True,rotate = False)
# 画图
plt.subplot(1, 2, 1)
plt.imshow(img_rotate)
plt.axis("off")
plt.subplot(1, 2, 2)
plt.imshow(img_flip)
plt.show()
plt.close()
重点就是第二行,改为:
plt.switch_backend('Tkagg')
可以正常显示了