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/layers/implicit_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/implicit_layer.cpp
@@ -3,7 +3,6 @@
 * https://www.github.com/marcoslucianops
 */

-#include <math.h>
 #include "implicit_layer.h"

 nvinfer1::ILayer* implicitLayer(