Add support to YOLOv5 v4.0 and v5.0

2022-04-08 17:13:05 -03:00
parent 2aa52a8e8c
commit 7bcc9b62fa
6 changed files with 102 additions and 63 deletions
--- a/nvdsinfer_custom_impl_Yolo/Makefile
+++ b/nvdsinfer_custom_impl_Yolo/Makefile
@@ -61,7 +61,7 @@ SRCFILES:= nvdsinfer_yolo_engine.cpp \
           layers/upsample_layer.cpp \
           layers/maxpool_layer.cpp \
           layers/activation_layer.cpp \
-           layers/reorg_r_layer.cpp \
+           layers/reorgv5_layer.cpp \
           utils.cpp \
           yolo.cpp \
           yoloForward.cu \