第一步:准备测试图片和测试模型
准备好一张需要测试的图片和训练好的测试模型,放到测试文件夹中
第二步:测试代码
1 | from PIL import Image |
注:
- 测试图片的尺寸需要和训练尺寸相同,并且也需要进行张量变换(ToTensor)和归一化(Normalize)。
- 由于pytorch要求的输入的维度为[batch_size, channels, width, height],而一个样本的维度为[channels, width, height],此时因为我们需要测试一张图片,所以用unsqueeze()增加一个维度变为[1, channels, width, height]。
- index只能给你一个数字,它其实代表的是第几个类别,如果想知道具体类别名称,可以打开训练集的文件夹,那么0就是第一个类别,1就是第二个类别…….以此类推。比如以下是我的训练集。
训练集文件夹
那么mn就是第0个类别,mp就是第一个类别。
money money money~ money money~
- 本文链接:http://yoursite.com/2020/07/28/DL_P4/
- 版权声明:本博客所有文章除特别声明外,均默认采用 许可协议。
若没有本文 Issue,您可以使用 Comment 模版新建。
GitHub Issues