From 18a3b86f29a5a2ba9ed34b062706d8e077c6d9ea Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Tue, 24 Mar 2020 11:36:38 +0800
Subject: [PATCH] Fix sensitivy in slim demo. (#390)

---
 slim/sensitive/sensitive.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/slim/sensitive/sensitive.py b/slim/sensitive/sensitive.py
index a8e95abf0..72c81fe41 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,
-- 
GitLab