added object detection to test suite

henon · henon · commit d86bce0fad2c · 2019-04-28T19:56:21.000+02:00
diff --git a/test/TensorFlowNET.Examples/NeuralNetXor.cs b/test/TensorFlowNET.Examples/NeuralNetXor.cs
@@ -25,13 +25,14 @@ public class NeuralNetXor : Python, IExample
         private (Operation, Tensor, Tensor) make_graph(Tensor features,Tensor labels, int num_hidden = 8)
         {
             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]
             var hidden_activations = tf.nn.relu(tf.matmul(features, hidden_weights));
 
             var output_weights = tf.Variable(tf.truncated_normal(
                 new[] {num_hidden, 1},
+                seed: 17,
                 stddev: (float) (1 / Math.Sqrt(num_hidden))
             ));
 
diff --git a/test/TensorFlowNET.UnitTest/ExamplesTests/ExamplesTest.cs b/test/TensorFlowNET.UnitTest/ExamplesTests/ExamplesTest.cs
@@ -123,5 +123,11 @@ public void NeuralNetXor_ImportedGraph()
         }
 
 
+        [TestMethod]
+        public void ObjectDetection()
+        {
+            tf.Graph().as_default();
+            Assert.IsTrue(new ObjectDetection() { Enabled = true, ImportGraph = true }.Run());
+        }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -123,5 +123,11 @@ public void NeuralNetXor_ImportedGraph()`
`123`	`123`	`}`
`124`	`124`
`125`	`125`
	`126`	`+ [TestMethod]`
	`127`	`+ public void ObjectDetection()`
	`128`	`+ {`
	`129`	`+ tf.Graph().as_default();`
	`130`	`+ Assert.IsTrue(new ObjectDetection() { Enabled = true, ImportGraph = true }.Run());`
	`131`	`+ }`
`126`	`132`	`}`
`127`	`133`	`}`