Z,J Hu commited on
Commit
29da076
·
1 Parent(s): 7976207

fix(demo): release blob in TensorRT(#403)

Browse files
Files changed (1) hide show
  1. demo/TensorRT/cpp/yolox.cpp +2 -1
demo/TensorRT/cpp/yolox.cpp CHANGED
@@ -519,7 +519,8 @@ int main(int argc, char** argv) {
519
  std::vector<Object> objects;
520
  decode_outputs(prob, objects, scale, img_w, img_h);
521
  draw_objects(img, objects, input_image_path);
522
-
 
523
  // destroy the engine
524
  context->destroy();
525
  engine->destroy();
 
519
  std::vector<Object> objects;
520
  decode_outputs(prob, objects, scale, img_w, img_h);
521
  draw_objects(img, objects, input_image_path);
522
+ // delete the pointer to the float
523
+ delete blob;
524
  // destroy the engine
525
  context->destroy();
526
  engine->destroy();