首页 > 解决方案 > 如何通过 Tensorflow 上的训练模型预测一些图像

问题描述

我是超级初学者。我检查了官方网站Transfer learning with a pretrained ConvNet。我想通过该站点的训练模型进行预测。下面的代码有错吗?我想知道“图像”和“类名”。嗯..我该怎么办?...请给我一些建议...</p>

predictions = model.predict(test_batches)[1]
print(predictions)
#[-0.11642772]

标签: pythonimagetensorflowmachine-learningprediction

解决方案


首先,TensorFlow 很糟糕。非常容易出错并且难以安装和使用。我建议改用 Pytorch 教程。其次,您发布的链接有一个 Colab 按钮。你试过点击那个按钮吗?Colab 使安装更容易,因为它是在线的,而不是在您的计算机上。

迁移学习也不是初学者的话题。如果适合您,也许可以尝试一个更简单的笔记本:https ://github.com/sgrvinod/Deep-Tutorials-for-PyTorch/blob/master/README.md


推荐阅读