yolov8: fix destroy order

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

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());
delete builder;
delete config;
delete serialized_engine;
delete config;
delete builder;
}

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());
delete builder;
delete config;
delete serialized_engine;
delete config;
delete builder;
}
void deserialize_engine(std::string &engine_name, IRuntime **runtime, ICudaEngine **engine, IExecutionContext **context)