handle batchnorm scale_factor zero, fix Tencent#302

XiaoguangHu01 · Apr 27, 2018 · b560252 · b560252
1 parent 8f56e00
commit b560252
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/caffe/caffe2ncnn.cpp b/tools/caffe/caffe2ncnn.cpp
@@ -438,7 +438,7 @@ int main(int argc, char** argv)
             }
             else
             {
-                float scale_factor = 1 / binlayer.blobs(2).data().data()[0];
+                float scale_factor = binlayer.blobs(2).data().data()[0] == 0 ? 0 : 1 / binlayer.blobs(2).data().data()[0];
                 // premultiply scale_factor to mean and variance
                 float tmp;
                 for (int j=0; j<mean_blob.data_size(); j++)