From 8ad99d3f8dba6be9dba6472be2a4f43d80199265 Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Sat, 13 Aug 2022 17:54:05 -0300
Subject: [PATCH] Fix YOLOv7 eps

---
 nvdsinfer_custom_impl_Yolo/yolo.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nvdsinfer_custom_impl_Yolo/yolo.cpp b/nvdsinfer_custom_impl_Yolo/yolo.cpp
index c784250..74b219c 100644
--- a/nvdsinfer_custom_impl_Yolo/yolo.cpp
+++ b/nvdsinfer_custom_impl_Yolo/yolo.cpp
@@ -166,7 +166,7 @@ NvDsInferStatus Yolo::buildYoloNetwork(std::vector<float>& weights, nvinfer1::IN
         weightsType = "wts";
 
     float eps = 1.0e-5;
-    if (m_NetworkType.find("yolov5") != std::string::npos)
+    if (m_NetworkType.find("yolov5") != std::string::npos || m_NetworkType.find("yolov7") != std::string::npos)
         eps = 1.0e-3;
     else if (m_NetworkType.find("yolor") != std::string::npos)
         eps = 1.0e-4;