当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.lite.Interpreter.resize_tensor_input用法及代码示例


用法

resize_tensor_input(
    input_index, tensor_size, strict=False
)

参数

  • input_index 要设置的输入的张量索引。该值可以从get_input_details 中的'index' 字段中获取。
  • tensor_size 调整输入大小的tensor_shape。
  • strict strict 为 True 时,只能调整未知尺寸的大小。未知维度在给定张量的shape_signature 属性中表示为-1。 (默认为假)

抛出

  • ValueError 如果解释器无法调整输入张量的大小。

调整输入张量的大小。

用法:

interpreter = Interpreter(model_content=tflite_model)
interpreter.resize_tensor_input(0, [num_test_images, 224, 224, 3])
interpreter.allocate_tensors()
interpreter.set_tensor(0, test_images)
interpreter.invoke()

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.lite.Interpreter.resize_tensor_input。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。