Skip to content

Commit 0368dc6

Browse files
authored
Fix Faster RCNN TF examples (#1437)
Signed-off-by: zehao-intel <zehao.huang@intel.com>
1 parent cf847c1 commit 0368dc6

File tree

2 files changed

+6
-6
lines changed
  • examples/tensorflow/object_detection/tensorflow_models
    • faster_rcnn_inception_resnet_v2/quantization/ptq
    • faster_rcnn_resnet101/quantization/ptq

2 files changed

+6
-6
lines changed

examples/tensorflow/object_detection/tensorflow_models/faster_rcnn_inception_resnet_v2/quantization/ptq/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ def evaluate(model):
5858
iteration = -1
5959
if args.benchmark and args.mode == 'performance':
6060
iteration = args.iters
61-
input_tensor_names = ["image_tensor:0"]
62-
output_tensor_names = ["num_detections:0", "detection_boxes:0", \
63-
"detection_scores:0", "detection_classes:0"]
61+
input_tensor_names = ["image_tensor"]
62+
output_tensor_names = ["num_detections", "detection_boxes", \
63+
"detection_scores", "detection_classes"]
6464
metric = COCOmAPv2(output_index_mapping={'num_detections':0, 'boxes':1, 'scores':2, 'classes':3})
6565

6666
if isinstance(model, AutoTrackable):

examples/tensorflow/object_detection/tensorflow_models/faster_rcnn_resnet101/quantization/ptq/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ def evaluate(model):
5858
iteration = -1
5959
if args.benchmark and args.mode == 'performance':
6060
iteration = args.iters
61-
input_tensor_names = ["image_tensor:0"]
62-
output_tensor_names = ["num_detections:0", "detection_boxes:0", \
63-
"detection_scores:0", "detection_classes:0"]
61+
input_tensor_names = ["image_tensor"]
62+
output_tensor_names = ["num_detections", "detection_boxes", \
63+
"detection_scores", "detection_classes"]
6464
metric = COCOmAPv2(output_index_mapping={'num_detections':0, 'boxes':1, 'scores':2, 'classes':3})
6565

6666
if isinstance(model, AutoTrackable):

0 commit comments

Comments
 (0)