未验证 提交 bc5bd154 编写于 作者: J juncaipeng 提交者: GitHub

Fix yolo_box bug (#2704)

* fix yolov3 bug when run several times, test=develop
上级 bb1cf7ff
...@@ -32,6 +32,8 @@ void YoloBoxCompute::Run() { ...@@ -32,6 +32,8 @@ void YoloBoxCompute::Run() {
int class_num = param.class_num; int class_num = param.class_num;
float conf_thresh = param.conf_thresh; float conf_thresh = param.conf_thresh;
int downsample_ratio = param.downsample_ratio; int downsample_ratio = param.downsample_ratio;
Boxes->clear();
Scores->clear();
lite::arm::math::yolobox(X, lite::arm::math::yolobox(X,
ImgSize, ImgSize,
Boxes, Boxes,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册