Fix YOLOv5 accuracy

2022-02-09 12:55:24 -03:00
parent d9c3cbf41d
commit 5b9b1f77c8
3 changed files with 7 additions and 13 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp
@@ -156,7 +156,7 @@ nvinfer1::ILayer* convolutionalLayer(
            }
            for (int i = 0; i < filters; ++i)
            {
-                bnRunningVar.push_back(sqrt(weights[weightPtr] + 1.0e-5));
+                bnRunningVar.push_back(sqrt(weights[weightPtr] + 1.0e-3));
                weightPtr++;
            }
            trtWeights.push_back(convWt);