leafqycc / rknn-cpp-Multithreading

A simple demo of yolov5s running on rk3588/3588s using c++ (about 142 frames). / 一个使用c++在rk3588/3588s上运行的yolov5s简单demo(142帧/s)。
Apache License 2.0
472 stars 87 forks source link

Segmentation Fault When Loading Custom RKNN #10

Open mdurrani808 opened 1 year ago

mdurrani808 commented 1 year ago

When running ./build-linux_RK3588 on an Orange Pi 5 running Ubuntu 22.04, I get a segmentation fault on line 26 (the line that runs rknn_yolov5_demo. This is a model converted from onnx using rknn-toolkit-2 if that helps. Any tips to get this working? Regular model provided works fine w/o segmentation faults. 2023.zip

Seems like it has the segfault on line 52 of rknnPool.hpp

leafqycc commented 1 year ago

When running ./build-linux_RK3588 on an Orange Pi 5 running Ubuntu 22.04, I get a segmentation fault on line 26 (the line that runs rknn_yolov5_demo. This is a model converted from onnx using rknn-toolkit-2 if that helps. Any tips to get this working? Regular model provided works fine w/o segmentation faults. 2023.zip

Seems like it has the segfault on line 52 of rknnPool.hpp

I guess you didn't specify the platform during the conversion process,

rknn.config(mean_values=[[0, 0, 0]], std_values=[[255, 255, 255]])

rknn.config(mean_values=[[0, 0, 0]], std_values=[[255, 255, 255]],target_platform='rk3588')

and you can refer to this repository for model training https://github.com/airockchip/yolov5