New YOLOv5 conversion and support (>= v2.0)

2022-07-14 11:50:55 -03:00
parent 095696a296
commit 058db92ad1
15 changed files with 763 additions and 502 deletions
--- a/nvdsinfer_custom_impl_Yolo/Makefile
+++ b/nvdsinfer_custom_impl_Yolo/Makefile
@@ -53,6 +53,7 @@ SRCFILES:= nvdsinfer_yolo_engine.cpp \
           nvdsparsebbox_Yolo.cpp \
           yoloPlugins.cpp \
           layers/convolutional_layer.cpp \
+           layers/batchnorm_layer.cpp \
           layers/implicit_layer.cpp \
           layers/channels_layer.cpp \
           layers/shortcut_layer.cpp \