[jira] [Created] (IGNITE-10202) ML: Create a POC of TensorFlow model inference in Java

classic Classic list List threaded Threaded
1 message Options
Reply | Threaded
Open this post in threaded view
|

[jira] [Created] (IGNITE-10202) ML: Create a POC of TensorFlow model inference in Java

Anton Vinogradov (Jira)
Anton Dmitriev created IGNITE-10202:
---------------------------------------

             Summary: ML: Create a POC of TensorFlow model inference in Java
                 Key: IGNITE-10202
                 URL: https://issues.apache.org/jira/browse/IGNITE-10202
             Project: Ignite
          Issue Type: Sub-task
          Components: ml
    Affects Versions: 2.8
            Reporter: Anton Dmitriev
            Assignee: Anton Dmitriev
             Fix For: 2.8


TensorFlow models are represented internally as computation graphs that are saved as a Protobuf objects with some kind of metadata. TensorFlow provides a good API for Python that allows to work with them, but Java API is very poor.

TensorFlow allows to import computation graph in Java API and manipulate it using low-level API and it looks like a one way to make model inference in Java. Another way is based on the fact that the most popular way to build a TensorFlow model is using of the Keras API. It allows to save model in _.h5_ format that can be imported into Java using Deeplearning4J.

So, the goal of this task is to *prepare a POC that demonstrates*:
 * How to import TensorFlow model into Java and make inference using TensorFlow Java API.
 * How to import TensorFlow model into Java and make inference using Deeplearning4J.



--
This message was sent by Atlassian JIRA
(v7.6.3#76005)