未验证 提交 27235321 编写于 作者: S shangliang Xu 提交者: GitHub

[dev] fix postprocess bug in deploy benchmark bug (#7028)

上级 f0ae1b64
...@@ -152,7 +152,10 @@ class Detector(object): ...@@ -152,7 +152,10 @@ class Detector(object):
def postprocess(self, inputs, result): def postprocess(self, inputs, result):
# postprocess output of predictor # postprocess output of predictor
np_boxes_num = result['boxes_num'] np_boxes_num = result['boxes_num']
if sum(np_boxes_num) <= 0: assert isinstance(np_boxes_num, np.ndarray), \
'`np_boxes_num` should be a `numpy.ndarray`'
if np_boxes_num.sum() <= 0:
print('[WARNNING] No object detected.') print('[WARNNING] No object detected.')
result = {'boxes': np.zeros([0, 6]), 'boxes_num': np_boxes_num} result = {'boxes': np.zeros([0, 6]), 'boxes_num': np_boxes_num}
result = {k: v for k, v in result.items() if v is not None} result = {k: v for k, v in result.items() if v is not None}
...@@ -188,7 +191,7 @@ class Detector(object): ...@@ -188,7 +191,7 @@ class Detector(object):
shape: [N, im_h, im_w] shape: [N, im_h, im_w]
''' '''
# model prediction # model prediction
np_boxes, np_masks = None, None np_boxes_num, np_boxes, np_masks = np.array([0]), None, None
for i in range(repeats): for i in range(repeats):
self.predictor.run() self.predictor.run()
output_names = self.predictor.get_output_names() output_names = self.predictor.get_output_names()
...@@ -428,7 +431,7 @@ class Detector(object): ...@@ -428,7 +431,7 @@ class Detector(object):
if not os.path.exists(self.output_dir): if not os.path.exists(self.output_dir):
os.makedirs(self.output_dir) os.makedirs(self.output_dir)
out_path = os.path.join(self.output_dir, video_out_name) out_path = os.path.join(self.output_dir, video_out_name)
fourcc = cv2.VideoWriter_fourcc(* 'mp4v') fourcc = cv2.VideoWriter_fourcc(*'mp4v')
writer = cv2.VideoWriter(out_path, fourcc, fps, (width, height)) writer = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
index = 1 index = 1
while (1): while (1):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册