Added YOLOR native support

YOLOR-CSP YOLOR-CSP* YOLOR-CSP-X YOLOR-CSP-X*
2021-12-12 00:47:32 -03:00
parent 7761ca7a6b
commit e2257a81c0
12 changed files with 336 additions and 6 deletions
--- a/nvdsinfer_custom_impl_Yolo/yolo.h
+++ b/nvdsinfer_custom_impl_Yolo/yolo.h
@@ -27,6 +27,8 @@
 #define _YOLO_H_

 #include "layers/convolutional_layer.h"
+#include "layers/implicit_layer.h"
+#include "layers/channels_layer.h"
 #include "layers/dropout_layer.h"
 #include "layers/shortcut_layer.h"
 #include "layers/route_layer.h"