提交 93997483 编写于 作者: H HydrogenSulfate

fix bug in DetPredictor.predict

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