How can we import a tensorflow 2.x model to Java?

Question

Since Tensorflow's Java API is still using 1.15, can we import a model made with Python and version 2.x to Java. If so, how?

Answer 1

TensorFlow 2.x support for Java is actually in progress and available only through snapshots on Sonatype OSS.

If you add this to your POM, you should be able to load and use them. Note that this only works for Linux and MacOSx at the moment since there is a bug with the Windows build. Also, the API varies a bit from the 1.15 version so you might need to adapt some pieces of code.

 <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>