haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

yolov8: fix destroy order

Browse Source
This commit is contained in:
wang-xinyu 2024-01-23 10:19:39 +08:00
parent ad2c7a14d0
commit 5336af0551
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yolov8/yolov8_det.cpp
View File

@ -28,9 +28,9 @@ void serialize_engine(std::string &wts_name, std::string &engine_name, std::stri
} }
p.write(reinterpret_cast<const char *>(serialized_engine->data()), serialized_engine->size()); p.write(reinterpret_cast<const char *>(serialized_engine->data()), serialized_engine->size());
delete builder;
delete config;
delete serialized_engine; delete serialized_engine;
delete config;
delete builder;
} }

4
yolov8/yolov8_seg.cpp
View File

@ -75,9 +75,9 @@ void serialize_engine(std::string &wts_name, std::string &engine_name, std::stri
} }
p.write(reinterpret_cast<const char *>(serialized_engine->data()), serialized_engine->size()); p.write(reinterpret_cast<const char *>(serialized_engine->data()), serialized_engine->size());
delete builder;
delete config;
delete serialized_engine; delete serialized_engine;
delete config;
delete builder;
} }
void deserialize_engine(std::string &engine_name, IRuntime **runtime, ICudaEngine **engine, IExecutionContext **context) void deserialize_engine(std::string &engine_name, IRuntime **runtime, ICudaEngine **engine, IExecutionContext **context)