Chiebot-Mirror
/
YOLOv8-TensorRT
mirror of https://github.com/triple-Mu/YOLOv8-TensorRT.git
9
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix memory leak

Browse Source 
fix trtModelStream release
pull/111/head
triple Mu 2 years ago committed by GitHub
parent d9687b12c1 4f52dd501a
commit 0f99469af6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 8 additions and 8 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      csrc/detect/end2end/include/yolov8.hpp
  2. 2
      csrc/detect/normal/include/yolov8.hpp
  3. 2
      csrc/jetson/detect/include/yolov8.hpp
  4. 2
      csrc/jetson/pose/include/yolov8-pose.hpp
  5. 2
      csrc/jetson/segment/include/yolov8-seg.hpp
  6. 2
      csrc/pose/normal/include/yolov8-pose.hpp
  7. 2
      csrc/segment/normal/include/yolov8-seg.hpp
  8. 2
      csrc/segment/simple/include/yolov8-seg.hpp

2
csrc/detect/end2end/include/yolov8.hpp
Unescape View File

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

2
csrc/detect/normal/include/yolov8.hpp
Unescape View File

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

2
csrc/jetson/detect/include/yolov8.hpp
Unescape View File

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

2
csrc/jetson/pose/include/yolov8-pose.hpp
Unescape View File

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

2
csrc/jetson/segment/include/yolov8-seg.hpp
Unescape View File

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

2
csrc/pose/normal/include/yolov8-pose.hpp
Unescape View File

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

2
csrc/segment/normal/include/yolov8-seg.hpp
Unescape View File

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

2
csrc/segment/simple/include/yolov8-seg.hpp
Unescape View File

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

Write Preview
Loading…
Cancel
Save