Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

YOLO V8/V9 TFlite Conversion Error to RKNN #457

Open
aayushb-95 opened this issue Jun 25, 2024 · 0 comments
Open

YOLO V8/V9 TFlite Conversion Error to RKNN #457

aayushb-95 opened this issue Jun 25, 2024 · 0 comments

Comments

@aayushb-95
Copy link

I am trying to convert the TFLite model to RKNN model and I am getting the following error.

W rknn-toolkit version: 1.7.5
W Channels(512) of input node: input_0 > 128, mean/std values will be set to default 0/1.
W Please do pre-processing manually before inference.
--> Config model
--> Loading model
TFLite Model Loaded /home/aayush/yolov9s.tflite
--> Building model
E get error shape: [None, None]
E Catch exception when loading tflite model:  /home/aayush/yolov9s.tflite!
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 492, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/api/rknn_log.py", line 323, in rknn.api.rknn_log.RKNNLog.e
E ValueError: get error shape: [None, None]
E Please feedback the detailed log file <log_feedback_to_the_rknn_toolkit_dev_team.log> to the RKNN Toolkit development team.
E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues
Traceback (most recent call last):
  File " /home/aayush/yolo_create_rknn.py", line 43, in <module>
    ret = rknn.build(do_quantization=True, dataset='images_512.txt', pre_compile=False, rknn_batch_size=1)
  File "/home/aayush/miniconda3/envs/rknn/lib/python3.8/site-packages/rknn/api/rknn.py", line 272, in build
    inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'

Any help on this will be appreciated.
Thanks

@aayushb-95 aayushb-95 changed the title TFlite Conversion Error TFlite Conversion Error to RKNN Jun 25, 2024
@aayushb-95 aayushb-95 changed the title TFlite Conversion Error to RKNN YOLO V8/V9 TFlite Conversion Error to RKNN Jun 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant