@@ -15,6 +15,7 @@ limitations under the License.
15
15
******************************************************************************/
16
16
17
17
using NumSharp ;
18
+ using Tensorflow ;
18
19
using Tensorflow . Keras . Engine ;
19
20
using Tensorflow . Keras . Utils ;
20
21
using static Tensorflow . Binding ;
@@ -56,16 +57,16 @@ public override void BuildModel()
56
57
var x = layers . Conv2D ( 32 , 3 , activation : "relu" ) . Apply ( inputs ) ;
57
58
x = layers . Conv2D ( 64 , 3 , activation : "relu" ) . Apply ( x ) ;
58
59
59
- x = layers . BatchNormalization ( ) . Apply ( x ) ;
60
+ // x = layers.BatchNormalization().Apply(x);
60
61
var block_1_output = layers . MaxPooling2D ( 3 ) . Apply ( x ) ;
61
62
62
63
x = layers . Conv2D ( 64 , 3 , activation : "relu" , padding : "same" ) . Apply ( block_1_output ) ;
63
64
x = layers . Conv2D ( 64 , 3 , activation : "relu" , padding : "same" ) . Apply ( x ) ;
64
- var block_2_output = layers . add ( x , block_1_output ) ;
65
+ var block_2_output = layers . Add ( ) . Apply ( new Tensors ( x , block_1_output ) ) ;
65
66
66
67
x = layers . Conv2D ( 64 , 3 , activation : "relu" , padding : "same" ) . Apply ( block_2_output ) ;
67
68
x = layers . Conv2D ( 64 , 3 , activation : "relu" , padding : "same" ) . Apply ( x ) ;
68
- var block_3_output = layers . add ( x , block_2_output ) ;
69
+ var block_3_output = layers . Add ( ) . Apply ( new Tensors ( x , block_2_output ) ) ;
69
70
70
71
x = layers . Conv2D ( 64 , 3 , activation : "relu" ) . Apply ( block_3_output ) ;
71
72
x = layers . GlobalAveragePooling2D ( ) . Apply ( x ) ;
0 commit comments