@@ -28,9 +28,14 @@ public void adjust_contrast()
28
28
{
29
29
var input = np . array ( 0f , 1f , 2f , 3f , 4f , 5f , 6f , 7f , 8f ) ;
30
30
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 ) ;
32
37
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 ) ;
34
39
}
35
40
36
41
[ Ignore ]
@@ -48,25 +53,31 @@ public void adjust_hue()
48
53
[ TestMethod ]
49
54
public void combined_non_max_suppression ( )
50
55
{
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 } } ) ;
52
57
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 } } ) ;
54
59
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 ) ) ;
56
67
57
68
var boxes_gt = tf . constant ( new float [ , ] { { 210f , 112f , 150f , 100f } , { 200f , 100f , 150f , 100f } , { 190f , 110f , 150f , 100f } ,
58
69
{ 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 ( ) ) ;
61
72
var scores_gt = tf . constant ( new float [ , ] { { 0.9f , 0.7f , 0.3f , 0f , 0f , 0f , 0f , 0f , 0f , 0f } } ) ;
62
73
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 ( ) ) ;
64
75
var classes_gt = tf . constant ( new float [ , ] { { 1f , 1f , 0f , 0f , 0f , 0f , 0f , 0f , 0f , 0f } } ) ;
65
76
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 ( ) ) ;
67
78
var valid_detections_gt = tf . constant ( new int [ , ] { { 3 } } ) ;
68
79
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 ( ) ) ;
70
81
}
71
82
72
83
[ TestMethod ]
0 commit comments