Tong

Tong

0个粉丝

7

问答

0

专栏

0

资料

Tong  发布于  2025-07-15 17:09:00
采纳率 15%
7个问答
437

ss928 yolov8模型推理失败

     

最近终于找到了可以转换模型的方法,但是这个yolov8.om跑出来以后数据不太对,是
output dims[0]—name:,dims:1
output dims[1]—name:,dims:1
output dims[2]—name:,dims:1
output dims[3]—name:,dims:80
这样的,我的模型输入在转换的时候设定为yvu420sp了,自带的例子是rgb,是这个模型无法跑yvu的视频吗?有没有朋友懂的,求教一下
处理函数是这样的
static td_s32 svp_npu_process_yolo(const ot_video_frame_info yolo_frame, td_float crop, td_u32 crop_size)
{
td_s32 ret;
td_void virt_addr = TD_NULL;

ot_svp_dst_img yolo_img;
td_u64 start_curr_time;


td_u32 size = (td_u32)(yolo_frame->video_frame.height * yolo_frame->video_frame.stride[0] *
    SAMPLE_SVP_NPU_IMG_THREE_CHN / SAMPLE_SVP_NPU_DOUBLE);
virt_addr = g_svp_npu_vb_virt_addr_yolo + (yolo_frame->video_frame.phys_addr[0] -
    g_svp_npu_vb_pool_info_yolo.pool_phy_addr);

//   yuv_to_rgb_convert(&yolo_img,yolo_frame);

ret = sample_common_svp_npu_update_input_data_buffer_info(virt_addr, size,
    yolo_frame->video_frame.stride[0], SVP_NPU_YOLO_IMAGE_IN,
    &g_svp_npu_task[0]);

//  ret = sample_common_svp_npu_update_input_data_buffer_info(yolo_img.virt_addr, size*2,
//     yolo_img.stride[0]/3, SVP_NPU_YOLO_IMAGE_IN,
//    &g_svp_npu_task[0]);
sample_svp_check_exps_return(ret != TD_SUCCESS, ret, SAMPLE_SVP_ERR_LEVEL_ERROR, "update data buffer failed!\n");

start_curr_time = sample_common_svp_npu_get_timestamp();


ret = sample_common_svp_npu_model_execute(&g_svp_npu_task[0]);
sample_svp_check_exps_return(ret != TD_SUCCESS, ret, SAMPLE_SVP_ERR_LEVEL_ERROR, "model execute failed!\n");

printf("[TIME] %llu ms\n", sample_common_svp_npu_get_timestamp() - start_curr_time);

// ret = svp_npu_post_process_yolo(svp_npu_get_task_info(YOLO_TASK), crop, crop_size);
ret = svp_npu_post_process_yolo(&g_svp_npu_task[0], crop, crop_size);
sample_svp_check_exps_return(ret != TD_SUCCESS, ret, SAMPLE_SVP_ERR_LEVEL_ERROR, “Preprocess Yolo failed!\n”);
return ret;
}

我来回答
回答1个
时间排序
认可量排序

Tong

0个粉丝

7

问答

0

专栏

0

资料

Tong 2025-07-29 17:12:11
认可0

已解决,这里得到的是输出的第一个buff,记录的是有效框数,具体的坐标需要进入到输出的第二个buff

或将文件直接拖到这里
悬赏:
E币
网盘
* 网盘链接:
* 提取码:
悬赏:
E币

Markdown 语法

  • 加粗**内容**
  • 斜体*内容*
  • 删除线~~内容~~
  • 引用> 引用内容
  • 代码`代码`
  • 代码块```编程语言↵代码```
  • 链接[链接标题](url)
  • 无序列表- 内容
  • 有序列表1. 内容
  • 缩进内容
  • 图片![alt](url)
+ 添加网盘链接/附件

Markdown 语法

  • 加粗**内容**
  • 斜体*内容*
  • 删除线~~内容~~
  • 引用> 引用内容
  • 代码`代码`
  • 代码块```编程语言↵代码```
  • 链接[链接标题](url)
  • 无序列表- 内容
  • 有序列表1. 内容
  • 缩进内容
  • 图片![alt](url)
相关问答
无更多相似问答 去提问
举报反馈

举报类型

  • 内容涉黄/赌/毒
  • 内容侵权/抄袭
  • 政治相关
  • 涉嫌广告
  • 侮辱谩骂
  • 其他

详细说明

易百纳技术社区