提交 ad26080e 编写于 作者: L LielinJiang 提交者: whs

fix score shape bug and some typo (#2906)

上级 bead65be
...@@ -47,7 +47,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe ...@@ -47,7 +47,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe
boxes = fluid.layers.data( boxes = fluid.layers.data(
name='boxes', shape=[-1, -1, 4], dtype='float32') name='boxes', shape=[-1, -1, 4], dtype='float32')
scores = fluid.layers.data( scores = fluid.layers.data(
name='scores', shape=[-1, -1, num_classes], dtype='float32') name='scores', shape=[-1, num_classes, -1], dtype='float32')
pred_result = fluid.layers.multiclass_nms( pred_result = fluid.layers.multiclass_nms(
bboxes=boxes, bboxes=boxes,
scores=scores, scores=scores,
...@@ -60,7 +60,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe ...@@ -60,7 +60,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe
executor.run(fluid.default_startup_program()) executor.run(fluid.default_startup_program())
for batch_id, data in enumerate(test_reader()): for batch_id, data in enumerate(test_reader()):
boxes_np, socres_np = exe.run(program=infer_program, boxes_np, scores_np = exe.run(program=infer_program,
feed={feeded_var_names[0]: feeder.feed(data)['image']}, feed={feeded_var_names[0]: feeder.feed(data)['image']},
fetch_list=target_var) fetch_list=target_var)
...@@ -68,7 +68,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe ...@@ -68,7 +68,7 @@ def use_coco_api_compute_mAP(data_args, test_list, num_classes, test_reader, exe
program=test_program, program=test_program,
feed={ feed={
'boxes': boxes_np, 'boxes': boxes_np,
'scores': socres_np 'scores': scores_np
}, },
fetch_list=[pred_result], return_numpy=False) fetch_list=[pred_result], return_numpy=False)
if batch_id % 20 == 0: if batch_id % 20 == 0:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册