Fixed yolov3-tiny_xnor.cfg

AlexeyAB · Aug 24, 2018 · 18d5e4f · 18d5e4f
1 parent 08285db
commit 18d5e4f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/build/darknet/x64/cfg/yolov3-tiny_xnor.cfg b/build/darknet/x64/cfg/yolov3-tiny_xnor.cfg
@@ -120,7 +120,6 @@ pad=1
 activation=leaky
 
 [convolutional]
-xnor=1
 batch_normalize=1
 filters=512
 size=3

diff --git a/cfg/yolov3-tiny_xnor.cfg b/cfg/yolov3-tiny_xnor.cfg
@@ -120,7 +120,6 @@ pad=1
 activation=leaky
 
 [convolutional]
-xnor=1
 batch_normalize=1
 filters=512
 size=3

diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
@@ -404,7 +404,9 @@ convolutional_layer make_convolutional_layer(int batch, int h, int w, int c, int
 
     //fprintf(stderr, "conv  %5d %2d x%2d /%2d  %4d x%4d x%4d   ->  %4d x%4d x%4d\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c);
     l.bflops = (2.0 * l.n * l.size*l.size*l.c * l.out_h*l.out_w) / 1000000000.;
-    fprintf(stderr, "conv  %5d %2d x%2d /%2d  %4d x%4d x%4d   ->  %4d x%4d x%4d %5.3f BF\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c, l.bflops);
+    if (l.xnor) fprintf(stderr, "convX ");
+    else  fprintf(stderr, "conv  ");
+    fprintf(stderr, "%5d %2d x%2d /%2d  %4d x%4d x%4d   ->  %4d x%4d x%4d %5.3f BF\n", n, size, size, stride, w, h, c, l.out_w, l.out_h, l.out_c, l.bflops);
 
     return l;
 }