Add YOLOR-P6 support

2022-02-14 15:38:12 -03:00
parent 9d118801be
commit a82f1b8662
7 changed files with 117 additions and 21 deletions
--- a/nvdsinfer_custom_impl_Yolo/Makefile
+++ b/nvdsinfer_custom_impl_Yolo/Makefile
@@ -61,6 +61,7 @@ SRCFILES:= nvdsinfer_yolo_engine.cpp \
           layers/upsample_layer.cpp \
           layers/maxpool_layer.cpp \
           layers/activation_layer.cpp \
+           layers/reorg_r_layer.cpp \
           utils.cpp \
           yolo.cpp \
           yoloForward.cu \