pytorch学习笔记------迁移学习④

用训练好的模型对未知的图像进行预测

'''
加载训练模型
需要对图像进行预处理
严格按照网络需要的格式
'''
import numpy as np
train_on_gpu=True
dataiter=iter(dataloaders['valid'])
images,labels=dataiter.next()
model_ft.eval()
if train_on_gpu:
    output=model_ft(images.cuda())
else:
    output=model_ft(images)
_,preds_tensor=torch.max(output,1)
preds=np.squeeze(preds_tensor.numpy()) if not train_on_gpu else np.squeeze(preds_tensor.cpu.numpy())