Merge pull request jolibrain#558 from fantes/ncnn_imgconn_fix

apply mean only if it exists in parameters
anazh · Apr 5, 2019 · 0ecc5a8 · 0ecc5a8
2 parents 8e435a6 + f4f444e
commit 0ecc5a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/backends/ncnn/ncnninputconns.h b/src/backends/ncnn/ncnninputconns.h
@@ -91,7 +91,8 @@ namespace dd
             _width = bgr.cols;
 
             _in = ncnn::Mat::from_pixels(bgr.data, ncnn::Mat::PIXEL_BGR, bgr.cols, bgr.rows);
-            _in.substract_mean_normalize(&_mean[0], 0);
+            if (_has_mean_scalar)
+              _in.substract_mean_normalize(&_mean[0], 0);
 	    _ids.push_back(this->_uris.at(0));
         }