Я пытаюсь использовать модель tflite для вывода из пакета. Я использую следующий код:
interpreter = tf.lite.Interpreter(model_path=model_path)
input_details = interpreter.get_input_details()
interpreter.resize_tensor_input(input_details[0]["index"], [batch_size, 513, 513, 3])
interpreter.allocate_tensors()
Код дает сбой и выдает следующую ошибку:
RuntimeError: tensorflow/lite/kernels/reshape.cc:58 num_input_elements != num_output_elements (1579014 != 789507)Node number 0 (RESHAPE) failed to prepare.
При просмотре деталей вывода он по-прежнему имеет форму [1, 513, 513, output_channels], а не[batch_size, 513, 513, output_channels] как я надеюсь.
Есть идеи?