Fix YOLO kernels

- Fix YOLO kernels - Update deprecated functions
2021-12-12 09:58:23 -03:00
parent ce35e17334
commit 9565254551
11 changed files with 316 additions and 153 deletions
--- a/nvdsinfer_custom_impl_Yolo/Makefile
+++ b/nvdsinfer_custom_impl_Yolo/Makefile
@@ -63,7 +63,10 @@ SRCFILES:= nvdsinfer_yolo_engine.cpp \
           layers/activation_layer.cpp \
           utils.cpp \
           yolo.cpp \
-           yoloForward.cu
+           yoloForward.cu \
+           yoloForward_v2.cu \
+           yoloForward_nc.cu \
+           yoloForward_r.cu

 ifeq ($(OPENCV), 1)
 SRCFILES+= calibrator.cpp