airockchip/rknn-toolkit2

TFLite conversion failed

Opened this issue · 0 comments

Having this TFListe model which take as an input shape (None, 4, 224, 244, 3) (4 frames)

model.zip

    rknn.config(
        float_dtype='tfloat32',
        mean_values=[170, 170, 170, 170],
        std_values=[170, 170, 170, 170],
        target_platform='rv1106',
        dynamic_input=[[[1,4, 224, 224, 3]]])
    ret = rknn.load_tflite(model='model.tflite', input_is_nchw=False)

I am getting the error


ValueError: Traceback (most recent call last):
  File "rknn/api/rknn_log.py", line 309, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper
  File "rknn/api/rknn_base.py", line 1945, in rknn.api.rknn_base.RKNNBase.build
  File "rknn/api/rknn_base.py", line 176, in rknn.api.rknn_base.RKNNBase._quantize
  File "rknn/api/quantizer.py", line 1397, in rknn.api.quantizer.Quantizer.run
  File "rknn/api/quantizer.py", line 899, in rknn.api.quantizer.Quantizer._get_layer_range
  File "rknn/api/rknn_utils.py", line 274, in rknn.api.rknn_utils.get_input_img
  File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e
ValueError: The height_width of r_shape [224, 224, 3] is invalid!

Please help