Fix memory leak

fix trtModelStream release

csrc/detect/end2end/include/yolov8.hpp

@@ -66,7 +66,7 @@ YOLOv8::YOLOv8(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);

csrc/detect/normal/include/yolov8.hpp

@@ -72,7 +72,7 @@ YOLOv8::YOLOv8(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);

csrc/jetson/detect/include/yolov8.hpp

@@ -66,7 +66,7 @@ YOLOv8::YOLOv8(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);

csrc/jetson/pose/include/yolov8-pose.hpp

@@ -80,7 +80,7 @@ YOLOv8_pose::YOLOv8_pose(const std::string &engine_file_path) {
 
     this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
     assert(this->engine != nullptr);
-
+    delete[] trtModelStream;
     this->context = this->engine->createExecutionContext();
 
     assert(this->context != nullptr);

csrc/jetson/segment/include/yolov8-seg.hpp

@@ -76,7 +76,7 @@ YOLOv8_seg::YOLOv8_seg(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);

csrc/pose/normal/include/yolov8-pose.hpp

@@ -80,7 +80,7 @@ YOLOv8_pose::YOLOv8_pose(const std::string &engine_file_path) {
 
     this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
     assert(this->engine != nullptr);
-
+    delete[] trtModelStream;
     this->context = this->engine->createExecutionContext();
 
     assert(this->context != nullptr);

csrc/segment/normal/include/yolov8-seg.hpp

@@ -76,7 +76,7 @@ YOLOv8_seg::YOLOv8_seg(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);

csrc/segment/simple/include/yolov8-seg.hpp

@@ -76,7 +76,7 @@ YOLOv8_seg::YOLOv8_seg(const std::string& engine_file_path)
 
 	this->engine = this->runtime->deserializeCudaEngine(trtModelStream, size);
 	assert(this->engine != nullptr);
-
+	delete[] trtModelStream;
 	this->context = this->engine->createExecutionContext();
 
 	assert(this->context != nullptr);