Browse Source

added object detection to test suite

tags/v0.9
Meinrad Recheis 6 years ago
parent
commit
d86bce0fad
2 changed files with 8 additions and 1 deletions
  1. +2
    -1
      test/TensorFlowNET.Examples/NeuralNetXor.cs
  2. +6
    -0
      test/TensorFlowNET.UnitTest/ExamplesTests/ExamplesTest.cs

+ 2
- 1
test/TensorFlowNET.Examples/NeuralNetXor.cs View File

@@ -25,13 +25,14 @@ namespace TensorFlowNET.Examples
private (Operation, Tensor, Tensor) make_graph(Tensor features,Tensor labels, int num_hidden = 8) private (Operation, Tensor, Tensor) make_graph(Tensor features,Tensor labels, int num_hidden = 8)
{ {
var stddev = 1 / Math.Sqrt(2); var stddev = 1 / Math.Sqrt(2);
var hidden_weights = tf.Variable(tf.truncated_normal(new int []{2, num_hidden}, stddev: (float) stddev ));
var hidden_weights = tf.Variable(tf.truncated_normal(new int []{2, num_hidden}, seed:1, stddev: (float) stddev ));
// Shape [4, num_hidden] // Shape [4, num_hidden]
var hidden_activations = tf.nn.relu(tf.matmul(features, hidden_weights)); var hidden_activations = tf.nn.relu(tf.matmul(features, hidden_weights));
var output_weights = tf.Variable(tf.truncated_normal( var output_weights = tf.Variable(tf.truncated_normal(
new[] {num_hidden, 1}, new[] {num_hidden, 1},
seed: 17,
stddev: (float) (1 / Math.Sqrt(num_hidden)) stddev: (float) (1 / Math.Sqrt(num_hidden))
)); ));


+ 6
- 0
test/TensorFlowNET.UnitTest/ExamplesTests/ExamplesTest.cs View File

@@ -123,5 +123,11 @@ namespace TensorFlowNET.ExamplesTests
} }
[TestMethod]
public void ObjectDetection()
{
tf.Graph().as_default();
Assert.IsTrue(new ObjectDetection() { Enabled = true, ImportGraph = true }.Run());
}
} }
} }

Loading…
Cancel
Save