fastai show_batch 和 show_results 不执行任何操作

Yun*_*dik 3 machine-learning deep-learning fast-ai image-classification

我在 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()
Run Code Online (Sandbox Code Playgroud)

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

小智 10

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

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

因此,就像在 show_results() 调用后键入“plt.show()”一样简单!

将 matplotlib.pyplot 导入为 plt

...

学习.show_results()

plt.show()

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