Segmentation fault yolov5s.onn customized

[titopuertolara]

Hi: Currently i'm facing issues with onnx->rknn conversion. With python simulator everything works fine. but when i convert onnx to rkkn i'm getting segmentation fault error. Here's a couple of screenshots . First one is yolov5s example, then we trained a yolov5s custom model with two classes using yolov5 ultralytics API. Finally . pt model was converted to onnx

[zen-xingle]

Hello,

The problem is raised by the mismatch of class number in C demo. You can correct class number and I think it will work.

[titopuertolara]

Thanks Zen!. That gives me another doubt and its related with rknn c API because i tried to change those indexes with no success.

[zhuzhenxxx]

tensor不一样 rk的yolo经过了裁剪