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();
|