자바에서 훈련 및 예측을 위해 Python으로 작성된 tensorflow 모델을 실행하십시오.

Question

나는 내 자신의 데이터 세트에 대한 시작 모델을 재교육했습니다. Tho 모델은 파이썬으로 만들어졌으며 저장된 그래프는 .pb 파일로 저장되고 레이블 파일은 .txt로 저장됩니다. 이제 Java를 통해이 모델을 이미지에 사용하여 예측해야합니다. 누구든지 도와주세요

Answer 1

TensorFlow 팀이 Java 인터페이스를 개발하고 있지만 아직 안정적이지 않습니다. 기존 코드는 https://github.com/tensorflow/tensorflow/tree/master/tensorflow/java이고 여기에서 개발 된 업데이트는 https://github.com/tensorflow/tensorflow/issues/5입니다. 이미 사용 된 방법을 보려면 GraphTest.java, SessionTest.java 및 TensorTest.java을 살펴볼 수 있습니다 (비록 설명이 있지만 나중에 변경 될 수 있음). 기본적으로 Graph 개체에 이진 저장된 그래프를로드하고 Session을 생성 한 다음 적절한 값 (Tensor 초)으로 실행하여 출력이 List<Tensor>이되도록해야합니다. 소스의 예제에서 함께 넣어 :

import java.nio.file.Files; 
import java.nio.file.Paths; 
import org.tensorflow.Graph; 
import org.tensorflow.Session; 
import org.tensorflow.Tensor; 

try (Graph graph = new Graph()) { 
    graph.importGraphDef(Files.readAllBytes(Paths.get("saved_model.pb")); 
    try (Session sess = new Session(graph)) { 
     try (Tensor x = Tensor.create(1.0f); 
      Tensor y = s.runner().feed("x", x).fetch("y").run().get(0)) { 
      System.out.println(y.floatValue()); 
     } 
    } 
} 

Answer 2

나는 그 .pb로 끝나는하는 protobuf 파일을 읽는 작업에 사용되는 코드입니다.

try (SavedModelBundle b = SavedModelBundle.load("/tmp/model", "serve")) { 
    Session sess = b.session(); 
    ... 
    float[][]matrix = sess.runner() 
     .feed("x", input) 
     .feed("keep_prob", keep_prob) 
     .fetch("y_conv") 
     .run() 
     .get(0) 
     .copyTo(new float[1][10]); 
    ... 
} 

내가 그것을 저장하는 데 사용되는 파이썬 코드이었다 :

signature = tf.saved_model.signature_def_utils.build_signature_def(
    inputs = {'x': tf.saved_model.utils.build_tensor_info(x)}, 
    outputs = {'y_conv': tf.saved_model.utils.build_tensor_info(y_conv)}, 
) 
    builder = tf.saved_model.builder.SavedModelBuilder("/tmp/model") 
    builder.add_meta_graph_and_variables(sess, 
     [tf.saved_model.tag_constants.SERVING], 
     signature_def_map={tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: signature} 
    ) 
    builder.save()