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/yoloPlugins.h
+++ b/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
@@ -85,9 +85,7 @@ public:

    void terminate () noexcept override {}

-    size_t getWorkspaceSize (int maxBatchSize) const noexcept override {
-        return maxBatchSize * sizeof(int);
-    }
+    size_t getWorkspaceSize (int maxBatchSize) const noexcept override { return 0; }

    int32_t enqueue (
        int batchSize, void const* const* inputs, void* const* outputs, void* workspace, cudaStream_t stream)