您知道,在某些情况下,通过在推理之前调用ResizeInputTensor,TensorFlow Lite可以很好地处理多批推理。但是我的模型不能很好地工作。
要让多个批处理在TensorFlow Lite中正常工作,需要满足哪些条件?
我怀疑您需要将batch维度设置为-1才能进行整形。
我想知道确切的情况。
谢谢。
发布于 2021-06-30 13:04:35
为了能够具有动态批处理,原始模型必须具有动态批处理,然后当您转换为tflite时,导出的模型将支持调整批处理大小。
https://stackoverflow.com/questions/68148631
复制相似问题