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/upsample_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/upsample_layer.h
@@ -7,12 +7,11 @@
 #define __UPSAMPLE_LAYER_H__

 #include <map>
-#include <vector>
 #include <cassert>

 #include "NvInfer.h"

-nvinfer1::ILayer* upsampleLayer(
+nvinfer1::ITensor* upsampleLayer(
    int layerIdx,
    std::map<std::string, std::string>& block,
    nvinfer1::ITensor* input,