Add PP-YOLOE support

2022-07-24 18:00:47 -03:00
parent d09879d557
commit a3782ed65e
51 changed files with 1812 additions and 600 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
@@ -6,18 +6,14 @@
 #ifndef __ACTIVATION_LAYER_H__
 #define __ACTIVATION_LAYER_H__

-#include <string>
 #include <cassert>
 #include <iostream>

 #include "NvInfer.h"

-#include "activation_layer.h"
-
-nvinfer1::ILayer* activationLayer(
+nvinfer1::ITensor* activationLayer(
    int layerIdx,
    std::string activation,
-    nvinfer1::ILayer* output,
    nvinfer1::ITensor* input,
    nvinfer1::INetworkDefinition* network);