TFLite conversion failed
Opened this issue · 0 comments
nassimus26 commented
Having this TFListe model which take as an input shape (None, 4, 224, 244, 3) (4 frames)
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