DeepStream 6.1 update

2022-05-26 08:05:06 -03:00
parent 4192a16751
commit ed387abf5d
3 changed files with 259 additions and 79 deletions
--- a/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
+++ b/nvdsinfer_custom_impl_Yolo/yoloPlugins.h
@@ -81,8 +81,8 @@ public:
    int initialize () noexcept override { return 0; }
    void terminate () noexcept override {}
    size_t getWorkspaceSize (int maxBatchSize) const noexcept override { return 0; }
-    int enqueue (
-        int batchSize, void const* const* inputs, void* const* outputs,
+    int32_t enqueue (
+        int32_t batchSize, void const* const* inputs, void* const* outputs,
        void* workspace, cudaStream_t stream) noexcept override;
    size_t getSerializationSize() const noexcept override;
    void serialize (void* buffer) const noexcept override;