search

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

c++版本运行我改过模型的yolov7无法运行，python版本可以 #30

Open youbaotang opened 10 months ago

youbaotang commented 10 months ago

Loading model... sdk version: 1.5.2 (c6b7b351a@2023-08-23T15:28:22) driver version: 0.8.2 model input num: 1, output num: 3 model is NHWC input fmt model input height=640, width=640, channel=3 rgaapi version 1.9.1[4] loadLabelName ./model/coco_80_labels_list.txt Segmentation fault (core dumped)

报错如上所述，同样的rknn模型文件，Python版本可以运行，c++加报错，请问作者有什么头绪吗，困扰我半个月了，感谢

leafqycc commented 10 months ago

Segmentation fault

Segmentation fault，或许是内存非法访问导致的？ 您可以使用netron （https://netron.app/）看看模型的结构，确认yolov7和demo中的yolov5s模型输入输出是否一致。 python版本目前是基于rknn 1.4.0版本，而c++版本是1.5.2版本，所以在后处理等方面有些许差别，在模型转换等过程中需要注意一下