thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix YOLOv7 eps

Browse Source
This commit is contained in:
Marcos Luciano
2022-08-13 17:54:05 -03:00
parent 80d08990a0
commit 8ad99d3f8d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nvdsinfer_custom_impl_Yolo/yolo.cpp
View File

@@ -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;