fastai show_batch 和 show_results 什么都不做

问题描述

我在 fastai v2.3.1 上建立了一个模型。当我尝试调用函数 show_batch 和 show_results 时不显示任何内容。我找不到我做错了什么。这是有问题的代码

from fastai.vision.all import *
from fastai.data.all import *
import fastai.vision
import zipfile as zf
import random
import timeit



fields = DataBlock(blocks=(ImageBlock,CategoryBlock),get_items=get_image_files,get_y=yer,splitter=RandomSplitter(valid_pct=0.2,seed=random.randint(0,10)),item_tfms=RandomresizedCrop(224,min_scale=0.5),batch_tfms=aug_transforms()
                   )


dls = fields.DataLoaders(os.path.join(Path(os.getcwd()),"train"),num_workers=0,bs=32)

dls.show_batch()

learn = cnn_learner(dls,resnet18,metrics=error_rate)

learn.fine_tune(2)

learn.show_results()

我可以使用模型但是这些功能

解决方法

我遇到了同样的事情,在这里找到了答案: https://www.debuggingtissue.com/latest-articles/how-to-save-fastai-plots-when-using-an-external-server-or-terminal-environment

基本上 PyPlot 正在创建一个图形对象但不显示它,因此您需要立即告诉 plt 保存/显示缓冲区。

所以就像在 show_results() 调用后输入“plt.show()”一样简单!

将 matplotlib.pyplot 导入为 plt

...

learn.show_results()

plt.show()

(花了我很长时间才发现,希望这会有所帮助!)