diff --git a/slim/sensitive/sensitive.py b/slim/sensitive/sensitive.py index a8e95abf0ada3009f11c40a74daf0f2b1bdd58e7..72c81fe418100dc29cc166c1a3e99985412b22fa 100644 --- a/slim/sensitive/sensitive.py +++ b/slim/sensitive/sensitive.py @@ -82,7 +82,6 @@ def main(): feed_vars, eval_loader = model.build_inputs(**inputs_def) fetches = model.eval(feed_vars) eval_prog = eval_prog.clone(True) - if FLAGS.print_params: print( "-------------------------All parameters in current graph----------------------" @@ -104,7 +103,7 @@ def main(): if cfg.metric == 'COCO': extra_keys = ['im_info', 'im_id', 'im_shape'] if cfg.metric == 'VOC': - extra_keys = ['gt_box', 'gt_label', 'is_difficult'] + extra_keys = ['gt_bbox', 'gt_class', 'is_difficult'] if cfg.metric == 'WIDERFACE': extra_keys = ['im_id', 'im_shape', 'gt_box'] eval_keys, eval_values, eval_cls = parse_fetches(fetches, eval_prog,