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

Fix YOLOv8 accuracy

Browse Source
This commit is contained in:
Marcos Luciano
2023-02-09 00:41:35 -03:00
parent 940d244ad4
commit 2d6b92eb64
9 changed files with 43 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nvdsinfer_custom_impl_Yolo/layers/c2f_layer.cpp
View File

@@ -58,7 +58,7 @@ c2fLayer(int layerIdx, std::map<std::string, std::string>& block, std::vector<fl
assert(cv2M != nullptr);
if (shortcut) {
nvinfer1::IElementWiseLayer* ew = network->addElementWise(*rb, *cv2M, nvinfer1::ElementWiseOperation::kSUM);
nvinfer1::IElementWiseLayer* ew = network->addElementWise(*output, *cv2M, nvinfer1::ElementWiseOperation::kSUM);
assert(ew != nullptr);
std::string ewLayerName = "shortcut_c2f_" + std::to_string(i + 1) + "_" + std::to_string(layerIdx);
ew->setName(ewLayerName.c_str());