pytorch如何对单个图片进行预测

1、pytorch如何对单个图片进行预测？

2、首先使用与训练网络时相同的transform

3、然后输入图片的地址

4、导入模型和权重

5、开始预测

6、预测结果如下所示