cqu20160901 / yolov8_rknn_Cplusplus

yolov8 瑞芯微 rknn 板端 C++部署。
BSD 3-Clause "New" or "Revised" License
89 stars 17 forks source link

rk3588 C++部署出现问题 #5

Open leepinkz opened 11 months ago

leepinkz commented 11 months ago

博主你好,我在yolov8n中加入了一个p2检测头,用您的代码运行不正确,我修改了postprocess.h中的headNum = 4,int strides[4] = {4, 8, 16, 32};int mapSize[4][2] = {{160, 160}, {80, 80}, {40, 40}, {20, 20}};运行出现段错误,我还需要修改哪些地方呢?

leepinkz commented 11 months ago

Read ./20009174.jpg ... img width = 1360, img height = 1020 sdk version: 1.5.2 (c6b7b351a@2023-08-23T15:28:22) driver version: 0.8.2 model input num: 1, output num: 5 index=0, name=images, n_dims=4, dims=[1, 640, 640, 3], n_elems=1228800, size=2457600, fmt=NHWC, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000

index=0, name=output0, n_dims=4, dims=[1, 24, 34000, 1], n_elems=816000, size=1632000, fmt=NCHW, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000 index=1, name=430, n_dims=4, dims=[1, 84, 160, 160], n_elems=2150400, size=4300800, fmt=NCHW, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000 index=2, name=439, n_dims=4, dims=[1, 84, 80, 80], n_elems=537600, size=1075200, fmt=NCHW, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000 index=3, name=448, n_dims=4, dims=[1, 84, 40, 40], n_elems=134400, size=268800, fmt=NCHW, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000 index=4, name=457, n_dims=4, dims=[1, 84, 20, 20], n_elems=33600, size=67200, fmt=NCHW, type=FP16, qnt_type=AFFINE, zp=0, scale=1.000000 model is NHWC input fmt model input height=640, width=640, channel=3 resize with RGA! rgaapi version 1.8.0[0] once run use 261.843000 ms === yolov8 Meshgrid Generate success! Segmentation fault

cqu20160901 commented 11 months ago

本仓库中的代码只适用按照本仓库给的参考导出onnx的方式,其他方式导出的onnx模型自行写后处理。导出onnx的参考链接 https://blog.csdn.net/zhangqian_1/article/details/128918268

Cigarliu commented 3 months ago

博主你好,我在yolov8n中加入了一个p2检测头,用您的代码运行不正确,我修改了postprocess.h中的headNum = 4,int strides[4] = {4, 8, 16, 32};int mapSize[4][2] = {{160, 160}, {80, 80}, {40, 40}, {20, 20}};运行出现段错误,我还需要修改哪些地方呢?

好兄弟 你的问题解决了吗,我也是加了p2检测头