未验证 提交 cc84b7ab 编写于 作者: W whs 提交者: GitHub

Fix sensitivy in slim demo. (#365)

上级 1455801e
...@@ -82,7 +82,6 @@ def main(): ...@@ -82,7 +82,6 @@ def main():
feed_vars, eval_loader = model.build_inputs(**inputs_def) feed_vars, eval_loader = model.build_inputs(**inputs_def)
fetches = model.eval(feed_vars) fetches = model.eval(feed_vars)
eval_prog = eval_prog.clone(True) eval_prog = eval_prog.clone(True)
if FLAGS.print_params: if FLAGS.print_params:
print( print(
"-------------------------All parameters in current graph----------------------" "-------------------------All parameters in current graph----------------------"
...@@ -104,7 +103,7 @@ def main(): ...@@ -104,7 +103,7 @@ def main():
if cfg.metric == 'COCO': if cfg.metric == 'COCO':
extra_keys = ['im_info', 'im_id', 'im_shape'] extra_keys = ['im_info', 'im_id', 'im_shape']
if cfg.metric == 'VOC': if cfg.metric == 'VOC':
extra_keys = ['gt_box', 'gt_label', 'is_difficult'] extra_keys = ['gt_bbox', 'gt_class', 'is_difficult']
if cfg.metric == 'WIDERFACE': if cfg.metric == 'WIDERFACE':
extra_keys = ['im_id', 'im_shape', 'gt_box'] extra_keys = ['im_id', 'im_shape', 'gt_box']
eval_keys, eval_values, eval_cls = parse_fetches(fetches, eval_prog, eval_keys, eval_values, eval_cls = parse_fetches(fetches, eval_prog,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册