diff --git a/TensorFlow/built-in/cv/image_classification/InceptionV3_ID0504_for_TensorFlow/infer_from_pb_cv.py b/TensorFlow/built-in/cv/image_classification/InceptionV3_ID0504_for_TensorFlow/infer_from_pb_cv.py index 358f8b3c80d11b13a35d3ed9529d66af277fa634..5061e0cf65b8147804454d4a39e0c2b29ee6d507 100644 --- a/TensorFlow/built-in/cv/image_classification/InceptionV3_ID0504_for_TensorFlow/infer_from_pb_cv.py +++ b/TensorFlow/built-in/cv/image_classification/InceptionV3_ID0504_for_TensorFlow/infer_from_pb_cv.py @@ -36,7 +36,7 @@ def parse_args(): help="""label file""") parser.add_argument('--input_tensor_name', default = 'input:0', help = """input_tensor_name""") - parser.add_argument('--output_tensor_name', default='InceptionV4/Logits/Logits/BiasAdd:0', + parser.add_argument('--output_tensor_name', default='InceptionV4/Logits/SpatialSqueeze:0', help="""output_tensor_name""") args, unknown_args = parser.parse_known_args() if len(unknown_args) > 0: