From 5b9b1f77c8f75169b4718d11f1017505b0d03d9b Mon Sep 17 00:00:00 2001 From: unknown Date: Wed, 9 Feb 2022 12:55:24 -0300 Subject: [PATCH] Fix YOLOv5 accuracy --- deepstream_app_config.txt | 2 +- .../layers/convolutional_layer.cpp | 2 +- utils/gen_wts_yoloV5.py | 16 +++++----------- 3 files changed, 7 insertions(+), 13 deletions(-) diff --git a/deepstream_app_config.txt b/deepstream_app_config.txt index ad4c95a..8c6822f 100644 --- a/deepstream_app_config.txt +++ b/deepstream_app_config.txt @@ -29,7 +29,7 @@ nvbuf-memory-type=0 [osd] enable=1 gpu-id=0 -border-width=1 +border-width=5 text-size=15 text-color=1;1;1;1; text-bg-color=0.3;0.3;0.3;1 diff --git a/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp b/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp index 08bd57e..a6fecac 100644 --- 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); diff --git a/utils/gen_wts_yoloV5.py b/utils/gen_wts_yoloV5.py index f648ee4..1f6ea3c 100644 --- a/utils/gen_wts_yoloV5.py +++ b/utils/gen_wts_yoloV5.py @@ -240,7 +240,7 @@ with open(cfg_file, "w") as c: layer += "\n# SPPF\n" layer += "\n[convolutional]\n" layer += "batch_normalize=1\n" - layer += "filters=%d\n" % get_width(v[3][0] / 2, width_multiple) + layer += "filters=%d\n" % (get_width(v[3][0], width_multiple) / 2) layer += "size=1\n" layer += "stride=1\n" layer += "pad=1\n" @@ -250,22 +250,16 @@ with open(cfg_file, "w") as c: layer += "stride=1\n" layer += "size=%d\n" % v[3][1] blocks += 1 - layer += "\n[route]\n" - layer += "layers=-2\n" + layer += "\n[maxpool]\n" + layer += "stride=1\n" + layer += "size=%d\n" % v[3][1] blocks += 1 layer += "\n[maxpool]\n" layer += "stride=1\n" layer += "size=%d\n" % v[3][1] blocks += 1 layer += "\n[route]\n" - layer += "layers=-2\n" - blocks += 1 - layer += "\n[maxpool]\n" - layer += "stride=1\n" - layer += "size=%d\n" % v[3][1] - blocks += 1 - layer += "\n[route]\n" - layer += "layers=-1, -3, -5, -6\n" + layer += "layers=-4, -3, -2, -1\n" blocks += 1 layer += "\n[convolutional]\n" layer += "batch_normalize=1\n"