From 3bdafe5f8b7fadedf56a85d77e7ce566b422247c Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Thu, 31 Aug 2023 15:29:34 -0300
Subject: [PATCH] Fix

---
 nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp b/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp
index 5fc548d..3a093c6 100644
--- a/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp
+++ b/nvdsinfer_custom_impl_Yolo/nvdsparsebbox_Yolo.cpp
@@ -196,3 +196,4 @@ NvDsInferParseYoloE(std::vector<NvDsInferLayerInfo> const& outputLayersInfo, NvD
 }
 
 CHECK_CUSTOM_PARSE_FUNC_PROTOTYPE(NvDsInferParseYolo);
+CHECK_CUSTOM_PARSE_FUNC_PROTOTYPE(NvDsInferParseYoloE);