java - 我们如何将 TensorFlow 2.x 模型导入到 Java 中？

Question

由于Tensorflow的Java API仍然使用1.15，我们是否可以将Python和2.x版本制作的模型导入到Java中。如果是这样，怎么办？

Answer 1

TensorFlow 2.x 对 Java 的支持实际上正在进行中，并且只能通过 Sonatype OSS 上的快照提供。

如果将其添加到 POM，您应该能够加载和使用它们。请注意，目前这仅适用于 Linux 和 MacOSx，因为 Windows 版本存在错误。此外，API 与 1.15 版本略有不同，因此您可能需要调整一些代码。

 <repositories>
    <repository>
        <id>tensorflow-snapshots</id>
        <url>https://oss.sonatype.org/content/repositories/snapshots/</url>
        <snapshots>
            <enabled>true</enabled>
        </snapshots>
    </repository>
 </repositories>
 <dependencies>
    <dependency>
        <groupId>org.tensorflow</groupId>
        <artifactId>tensorflow-core-platform</artifactId>
        <version>0.1.0-SNAPSHOT</version>
    </dependency>
 </dependencies>