提交 93997483 编写于 作者: H HydrogenSulfate

fix bug in DetPredictor.predict

上级 9e53098c
......@@ -128,13 +128,11 @@ class DetPredictor(Predictor):
results = []
if reduce(lambda x, y: x * y, np_boxes.shape) < 6:
print('[WARNNING] No object detected.')
results = np.array([])
results = []
else:
results = np_boxes
results = self.parse_det_results(results,
self.config["Global"]["threshold"],
self.config["Global"]["label_list"])
results = self.parse_det_results(
np_boxes, self.config["Global"]["threshold"],
self.config["Global"]["label_list"])
return results
......
......@@ -14,6 +14,8 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import division
from __future__ import print_function
import platform
import paddle
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册