From d56f25df65459365204cb930b4a94b84264fa165 Mon Sep 17 00:00:00 2001
From: juncaipeng <52520497+juncaipeng@users.noreply.github.com>
Date: Tue, 31 Dec 2019 15:15:10 +0800
Subject: [PATCH] Fix yolo_box bug  (#2704) (#2707)

* fix yolov3 bug when run several times, test=develop
---
 lite/kernels/arm/yolo_box_compute.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lite/kernels/arm/yolo_box_compute.cc b/lite/kernels/arm/yolo_box_compute.cc
index ad8a630b8c..38443bf277 100644
--- a/lite/kernels/arm/yolo_box_compute.cc
+++ b/lite/kernels/arm/yolo_box_compute.cc
@@ -32,6 +32,8 @@ void YoloBoxCompute::Run() {
   int class_num = param.class_num;
   float conf_thresh = param.conf_thresh;
   int downsample_ratio = param.downsample_ratio;
+  Boxes->clear();
+  Scores->clear();
   lite::arm::math::yolobox(X,
                            ImgSize,
                            Boxes,
-- 
GitLab