Skip to content

Commit 3273cbc

Browse files
author
“Wanglongzhi2001”
committed
fix: fix ci error
1 parent b0ce73c commit 3273cbc

File tree

1 file changed

+21
-10
lines changed

1 file changed

+21
-10
lines changed

test/TensorFlowNET.Graph.UnitTest/ImageTest.cs

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,14 @@ public void adjust_contrast()
2828
{
2929
var input = np.array(0f, 1f, 2f, 3f, 4f, 5f, 6f, 7f, 8f);
3030
var image = tf.reshape(input, new int[] { 3, 3, 1 });
31-
var img = tf.image.adjust_contrast(image, 2.0f);
31+
32+
var init = tf.global_variables_initializer();
33+
var sess = tf.Session();
34+
sess.run(init);
35+
var adjust_contrast = tf.image.adjust_contrast(image, 2.0f);
36+
var result = sess.run(adjust_contrast);
3237
var res = np.array(-4f, -2f, 0f, 2f, 4f, 6f, 8f, 10f, 12f).reshape((3,3,1));
33-
Assert.AreEqual(img.numpy(), res);
38+
Assert.AreEqual(result.numpy(), res);
3439
}
3540

3641
[Ignore]
@@ -48,25 +53,31 @@ public void adjust_hue()
4853
[TestMethod]
4954
public void combined_non_max_suppression()
5055
{
51-
var boxesX = tf.constant(new float[,] { { 200, 100, 150, 100 }, { 220, 120, 150, 100 }, { 190, 110, 150, 100 },{ 210, 112, 150, 100 } });
56+
var boxesX = tf.constant(new float[,] { { 200, 100, 150, 100 }, { 220, 120, 150, 100 }, { 190, 110, 150, 100 }, { 210, 112, 150, 100 } });
5257
var boxes1 = tf.reshape(boxesX, (1, 4, 1, 4));
53-
var scoresX = tf.constant(new float[,] { { 0.2f, 0.7f, 0.1f },{ 0.1f, 0.8f, 0.1f },{ 0.3f, 0.6f, 0.1f },{ 0.05f, 0.9f, 0.05f } });
58+
var scoresX = tf.constant(new float[,] { { 0.2f, 0.7f, 0.1f }, { 0.1f, 0.8f, 0.1f }, { 0.3f, 0.6f, 0.1f }, { 0.05f, 0.9f, 0.05f } });
5459
var scores1 = tf.reshape(scoresX, (1, 4, 3));
55-
var (boxes, scores, classes, valid_detections) = tf.image.combined_non_max_suppression(boxes1, scores1, 10, 10, 0.5f, 0.2f, clip_boxes:false);
60+
61+
var init = tf.global_variables_initializer();
62+
var sess = tf.Session();
63+
sess.run(init);
64+
65+
var (boxes, scores, classes, valid_detections) = tf.image.combined_non_max_suppression(boxes1, scores1, 10, 10, 0.5f, 0.2f, clip_boxes: false);
66+
var result = sess.run((boxes, scores, classes, valid_detections));
5667

5768
var boxes_gt = tf.constant(new float[,] { { 210f, 112f, 150f, 100f }, { 200f, 100f, 150f, 100f }, { 190f, 110f, 150f, 100f },
5869
{ 0f, 0f, 0f, 0f},{ 0f, 0f, 0f, 0f},{ 0f, 0f, 0f, 0f},{ 0f, 0f, 0f , 0f},{ 0f, 0f, 0f, 0f},{ 0f , 0f, 0f, 0f},{ 0f, 0f, 0f, 0f} });
59-
boxes_gt = tf.reshape(boxes_gt,(1, 10, 4));
60-
Assert.AreEqual(boxes.numpy(), boxes_gt.numpy());
70+
boxes_gt = tf.reshape(boxes_gt, (1, 10, 4));
71+
Assert.AreEqual(result.Item1.numpy(), boxes_gt.numpy());
6172
var scores_gt = tf.constant(new float[,] { { 0.9f, 0.7f, 0.3f, 0f, 0f, 0f, 0f, 0f, 0f, 0f } });
6273
scores_gt = tf.reshape(scores_gt, (1, 10));
63-
Assert.AreEqual(scores.numpy(), scores_gt.numpy());
74+
Assert.AreEqual(result.Item2.numpy(), scores_gt.numpy());
6475
var classes_gt = tf.constant(new float[,] { { 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f } });
6576
classes_gt = tf.reshape(classes_gt, (1, 10));
66-
Assert.AreEqual(classes.numpy(), classes_gt.numpy());
77+
Assert.AreEqual(result.Item3.numpy(), classes_gt.numpy());
6778
var valid_detections_gt = tf.constant(new int[,] { { 3 } });
6879
valid_detections_gt = tf.reshape(valid_detections_gt, (1));
69-
Assert.AreEqual(valid_detections.numpy(), valid_detections_gt.numpy());
80+
Assert.AreEqual(result.Item4.numpy(), valid_detections_gt.numpy());
7081
}
7182

7283
[TestMethod]

0 commit comments

Comments
 (0)