Skip to content

Commit 92fb8b2

Browse files
authored
fix deploy when bs=2 (PaddlePaddle#6915)
1 parent 9231377 commit 92fb8b2

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

deploy/python/infer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def postprocess(self, inputs, result):
154154
np_boxes_num = result['boxes_num']
155155
if sum(np_boxes_num) <= 0:
156156
print('[WARNNING] No object detected.')
157-
result = {'boxes': np.zeros([0, 6]), 'boxes_num': [0]}
157+
result = {'boxes': np.zeros([0, 6]), 'boxes_num': np_boxes_num}
158158
result = {k: v for k, v in result.items() if v is not None}
159159
return result
160160

@@ -402,10 +402,8 @@ def predict_image(self,
402402
self.pred_config.labels,
403403
output_dir=self.output_dir,
404404
threshold=self.threshold)
405-
406405
results.append(result)
407406
print('Test iter {}'.format(i))
408-
409407
results = self.merge_batch_result(results)
410408
if save_results:
411409
Path(self.output_dir).mkdir(exist_ok=True)

0 commit comments

Comments
 (0)