本文整理汇总了Java中org.tensorflow.TensorFlow类的典型用法代码示例。如果您正苦于以下问题:Java TensorFlow类的具体用法?Java TensorFlow怎么用?Java TensorFlow使用的例子?那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
TensorFlow类属于org.tensorflow包,在下文中一共展示了TensorFlow类的7个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于我们的系统推荐出更棒的Java代码示例。
示例1: loadGraph
import org.tensorflow.TensorFlow; //导入依赖的package包/类
private void loadGraph(InputStream is, Graph g) throws IOException {
final long startMs = System.currentTimeMillis();
Trace.beginSection("initializeTensorFlow");
Trace.beginSection("readGraphDef");
// TODO(ashankar): Can we somehow mmap the contents instead of copying them?
byte[] graphDef = new byte[is.available()];
final int numBytesRead = is.read(graphDef);
if (numBytesRead != graphDef.length) {
throw new IOException(
"read error: read only "
+ numBytesRead
+ " of the graph, expected to read "
+ graphDef.length);
}
Trace.endSection();
Trace.beginSection("importGraphDef");
try {
g.importGraphDef(graphDef);
} catch (IllegalArgumentException e) {
throw new IOException("Not a valid TensorFlow Graph serialization: " + e.getMessage());
}
Trace.endSection();
Trace.endSection(); // initializeTensorFlow.
final long endMs = System.currentTimeMillis();
Log.i(
TAG,
"Model load took " + (endMs - startMs) + "ms, TensorFlow version: " + TensorFlow.version());
}
开发者ID:Jamjomjara,项目名称:snu-artoon,代码行数:34,代码来源:TensorFlowInferenceInterface.java
示例2: main
import org.tensorflow.TensorFlow; //导入依赖的package包/类
public static void main(String[] args) throws Exception {
try (Graph g = new Graph()) {
final String value = "Hello from " + TensorFlow.version();
// Construct the computation graph with a single operation, a constant
// named "MyConst" with a value "value".
try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
// The Java API doesn't yet include convenience functions for adding operations.
g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
}
// Execute the "MyConst" operation in a Session.
try (Session s = new Session(g);
Tensor output = s.runner().fetch("MyConst").run().get(0)) {
System.out.println(new String(output.bytesValue(), "UTF-8"));
}
}
}
开发者ID:neo4j-contrib,项目名称:neo4j-ml-procedures,代码行数:19,代码来源:HelloTF.java
示例3: testTf
import org.tensorflow.TensorFlow; //导入依赖的package包/类
@Test
public void testTf() throws UnsupportedEncodingException {
try (Graph g = new Graph()) {
final String value = "Hello from " + TensorFlow.version();
// Construct the computation graph with a single operation, a constant
// named "MyConst" with a value "value".
try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
// The Java API doesn't yet include convenience functions for adding operations.
g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
}
// Execute the "MyConst" operation in a Session.
try (Session s = new Session(g); Tensor output = s.runner().fetch("MyConst").run().get(0)) {
logger.info(new String(output.bytesValue(), "UTF-8"));
logger.info(output.toString());
}
}
}
开发者ID:softelnet,项目名称:sponge,代码行数:20,代码来源:TensorflowTest.java
示例4: main
import org.tensorflow.TensorFlow; //导入依赖的package包/类
public static void main(String[] args) throws Exception {
try (Graph g = new Graph()) {
final String value = "Hello from " + TensorFlow.version();
// Construct the computation graph with a single operation, a constant
// named "MyConst" with a value "value".
try (Tensor t = Tensor.create(value.getBytes("UTF-8"))) {
// The Java API doesn't yet include convenience functions for adding operations.
g.opBuilder("Const", "MyConst").setAttr("dtype", t.dataType()).setAttr("value", t).build();
}
// Execute the "MyConst" operation in a Session.
try (Session s = new Session(g);
Tensor output = s.runner().fetch("MyConst").run().get(0)) {
System.out.println(new String(output.bytesValue(), "UTF-8"));
}
}
}
开发者ID:SensorsINI,项目名称:jaer,代码行数:19,代码来源:HelloTF.java
示例5: printUsage
import org.tensorflow.TensorFlow; //导入依赖的package包/类
private static void printUsage(PrintStream s) {
final String url
= "https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip";
s.println(
"Java program that uses a pre-trained Inception model (http://arxiv.org/abs/1512.00567)");
s.println("to label JPEG images.");
s.println("TensorFlow version: " + TensorFlow.version());
s.println();
s.println("Usage: label_image <model dir> <image file>");
s.println();
s.println("Where:");
s.println("<model dir> is a directory containing the unzipped contents of the inception model");
s.println(" (from " + url + ")");
s.println("<image file> is the path to a JPEG image file");
}
开发者ID:SensorsINI,项目名称:jaer,代码行数:16,代码来源:LabelImage.java
示例6: printUsage
import org.tensorflow.TensorFlow; //导入依赖的package包/类
private static void printUsage(PrintStream s) {
final String url =
"https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip";
s.println(
"Java program that uses a pre-trained Inception model (http://arxiv.org/abs/1512.00567)");
s.println("to label JPEG images.");
s.println("TensorFlow version: " + TensorFlow.version());
s.println();
s.println("Usage: label_image <model dir> <image file>");
s.println();
s.println("Where:");
s.println("<model dir> is a directory containing the unzipped contents of the inception model");
s.println(" (from " + url + ")");
s.println("<image file> is the path to a JPEG image file");
}
开发者ID:loretoparisi,项目名称:tensorflow-java,代码行数:16,代码来源:LabelImage.java
示例7: main
import org.tensorflow.TensorFlow; //导入依赖的package包/类
public static void main(String[] args) {
System.out.println("TensorFlowExample using TensorFlow version: " + TensorFlow.version());
}
开发者ID:loretoparisi,项目名称:tensorflow-java,代码行数:4,代码来源:TensorFlowExample.java
注:本文中的org.tensorflow.TensorFlow类示例整理自Github/MSDocs等源码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。 |
请发表评论