python - 如何使用我的类型进行预测在我的一组图像上？Python / Torchvision / PyTorch

Question

我已经在我的训练和验证集上训练了我的图像，但现在我希望将我的模型应用到我的测试集（我也存储了分类）。我见过的甚至处理测试集的唯一方法是通过与验证集完全相同的机制，但这使它成为验证集而不是测试集的扩展？没有预测。

model_ft, hist = train_model(model_ft, dataloaders_dict, criterion, optimizer_ft, num_epochs=num_epochs)

标签： pythonpytorchtorchvision

Answer 1

试试下面的东西

def predict(model, dataloader):  
  # Set model to evaluate mode
  model.eval()   
  predictions = []

  with torch.no_grad()
    # Iterate over data.
    for inputs, _ in dataloader:
      outputs = model(inputs)
      _, preds = torch.max(outputs, 1)
      predictions.extend(preds.numpy())

  return predictions

print (predict(model_ft, your_test_dataloader))