From 4dae28be209c22dace596ab06895de7914f86d69 Mon Sep 17 00:00:00 2001 From: spacewalk01 Date: Wed, 28 Feb 2024 17:48:11 +0900 Subject: [PATCH] fix the problem of destroying a builder object (#1443) --- retinaface/retina_mnet.cpp | 1 + retinaface/retina_r50.cpp | 1 + 2 files changed, 2 insertions(+) diff --git a/retinaface/retina_mnet.cpp b/retinaface/retina_mnet.cpp index bef3f39..213cbfd 100644 --- a/retinaface/retina_mnet.cpp +++ b/retinaface/retina_mnet.cpp @@ -236,6 +236,7 @@ void APIToModel(unsigned int maxBatchSize, IHostMemory** modelStream) { // Close everything down engine->destroy(); + config->destroy(); builder->destroy(); } diff --git a/retinaface/retina_r50.cpp b/retinaface/retina_r50.cpp index f5c4413..c751dea 100644 --- a/retinaface/retina_r50.cpp +++ b/retinaface/retina_r50.cpp @@ -255,6 +255,7 @@ void APIToModel(unsigned int maxBatchSize, IHostMemory** modelStream) { // Close everything down engine->destroy(); + config->destroy(); builder->destroy(); }