开始使用免费开始使用

分析模型输出

已经拿到预训练模型的预测结果后,您现在可以更深入地分析它。prediction 已为您准备好。

您会发现它是一个长度为 1 的列表(因为您只对一张图像进行了预测),其中包含一个具有 4 个键的字典。labels 是表示预测类别的整数,scores 是对应的类别概率。您可以通过 class_names 访问与 labels 对应的类别名称列表。

前两个掩膜的预测类别分别是什么?它们对应的分数是多少?

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

动手互动练习

通过我们的互动练习之一,将理论转化为实践

开始练习