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)