Z,J Hu
commited on
Commit
·
29da076
1
Parent(s):
7976207
fix(demo): release blob in TensorRT(#403)
Browse files
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();
|