From a723e1c62a27aeb39aaf7fcdeb3beb4e89fba32d Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Wed, 15 Aug 2018 20:52:09 +0000
Subject: [PATCH] Merge pull request #766 from HotChick91/AlexeyAB-mask

---
 src/detector.c |    2 ++
 1 files changed, 2 insertions(+), 0 deletions(-)

diff --git a/src/detector.c b/src/detector.c
index 244b4c3..a816c74 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -568,6 +568,7 @@
     }
     //set_batch_network(&net, 1);
     fuse_conv_batchnorm(net);
+    calculate_binary_weights(net);
     srand(time(0));
 
     list *plist = get_paths(valid_images);
@@ -1094,6 +1095,7 @@
     }
     //set_batch_network(&net, 1);
     fuse_conv_batchnorm(net);
+    calculate_binary_weights(net);
     if (net.layers[net.n - 1].classes != names_size) {
         printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n",
             name_list, names_size, net.layers[net.n - 1].classes, cfgfile);

--
Gitblit v1.10.0