From 5336af0551ec6389e2c160418b64f9df4b3151ec Mon Sep 17 00:00:00 2001 From: wang-xinyu Date: Tue, 23 Jan 2024 10:19:39 +0800 Subject: [PATCH] yolov8: fix destroy order --- yolov8/yolov8_det.cpp | 4 ++-- yolov8/yolov8_seg.cpp | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/yolov8/yolov8_det.cpp b/yolov8/yolov8_det.cpp index 0c327a6..21e6f25 100644 --- a/yolov8/yolov8_det.cpp +++ b/yolov8/yolov8_det.cpp @@ -28,9 +28,9 @@ void serialize_engine(std::string &wts_name, std::string &engine_name, std::stri } p.write(reinterpret_cast(serialized_engine->data()), serialized_engine->size()); - delete builder; - delete config; delete serialized_engine; + delete config; + delete builder; } diff --git a/yolov8/yolov8_seg.cpp b/yolov8/yolov8_seg.cpp index 143d532..8ba7142 100644 --- a/yolov8/yolov8_seg.cpp +++ b/yolov8/yolov8_seg.cpp @@ -75,9 +75,9 @@ void serialize_engine(std::string &wts_name, std::string &engine_name, std::stri } p.write(reinterpret_cast(serialized_engine->data()), serialized_engine->size()); - delete builder; - delete config; delete serialized_engine; + delete config; + delete builder; } void deserialize_engine(std::string &engine_name, IRuntime **runtime, ICudaEngine **engine, IExecutionContext **context)