背景:我是 Tensorflow 和 AI 的新手,希望在基于 Java 的环境中试用 Tensorflow。发现 Tensorflow 有一个 Java API 并尝试了它,但结果似乎与我的 Python 实现并不相似。我将 Maven 用于我的 Java 实现。
我按照Tensorflow for Poets教程使用 Python 中的 Tensorflow 重新训练了一个数据集。
在基于 Python 的 Tensorflow 环境中,数据集输出的值或多或少是正确的。例如,如果我有一张汽车的图像,我的分类选项是汽车、虫子或建筑物;它会输出汽车。
但是在我尝试使用 Tensorflow Java API 和Image label demo 的Java 等价物中;我之前生成的数据集是可读的,但输出与 Python 实现完全不同。与 Python 示例不同,如果我扔一辆车,它会将其输出为错误或建筑物。
我曾在几台装有不同代处理器和 gpu 的不同机器上尝试输出数据,看看它是否会影响输出;但它似乎是语言/API 独有的。
关于为什么使用相同数据集的两个实现会产生完全不同的输出的任何想法?
有什么办法可以让我的 Java 版本输出与我的 Python 示例类似的结果?
PIPIONE
相关分类