我一直在关注pytorch传输学习教程，并按照我自己的数据集教程来到这里
model_conv = train_model(model_conv, criterion, optimizer_conv,
exp_lr_scheduler, num_epochs=25)
Epoch 1/1
......
......
......
Epoch 24/24
train Loss: 0.8674 Acc: 0.5784
val Loss: 1.2930 Acc: 0.4583
Training complete in 43m 24s
Best val Acc: 0.486111
本教程在此结束，但我不能确定如何获得新图像的预测？
任何建议都会非常有用。 提前谢谢。
答案 0 :(得分：2)
只需将新图像（格式与训练数据集中的图像相同）输入模型：
labels = model_conv(new_images)